Method Query

internal/store/postgres/postgres.go:154–203

Query 在只读事务中执行 SQL，最多返回 500 行，超过 5 秒自动取消。 使用只读事务隔离，防止 DDL/DML 误操作。

(sqlStr string)

// Query 在只读事务中执行 SQL，最多返回 500 行，超过 5 秒自动取消。
// 使用只读事务隔离，防止 DDL/DML 误操作。
func (db *DB) Query(sqlStr string) (*QueryResult, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	start := time.Now()

	tx, err := db.conn.BeginTx(ctx, pgx.TxOptions{AccessMode: pgx.ReadOnly})
	if err != nil {
		return nil, err
	}
	defer tx.Rollback(ctx) //nolint:errcheck

	rows, err := tx.Query(ctx, sqlStr)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	fds := rows.FieldDescriptions()
	cols := make([]string, len(fds))
	for i, fd := range fds {
		cols[i] = fd.Name
	}

	result := &QueryResult{Columns: cols}
	for rows.Next() {
		if len(result.Rows) >= 500 {
			break
		}
		vals := make([]any, len(cols))
		ptrs := make([]any, len(cols))
		for i := range vals {
			ptrs[i] = &vals[i]
		}
		if err := rows.Scan(ptrs...); err != nil {
			return nil, err
		}
		// 将 []byte 转为 string，方便 JSON 序列化
		row := make([]any, len(cols))
		for i, v := range vals {
			if b, ok := v.([]byte); ok {
				row[i] = string(b)
			} else {
				row[i] = v
			}
		}
		result.Rows = append(result.Rows, row)
	}
	result.Elapsed = float64(time.Since(start).Microseconds()) / 1000.0
	return result, rows.Err()
}

// UserGroupStore 返回用户组 Store 实例。
func (db *DB) UserGroupStore() *UserGroupStore {