u

code/allapp/allapp-go-v3/pkg/db/select.go

@@ -0,0 +1,82 @@
+package db
+
+import (
+	"context"
+	"fmt"
+	"strings"
+)
+
+func (c *Client) LoadData(
+	ctx context.Context,
+	viewName string,
+	searchCondition string,
+	orderBy string,
+	searchColumns []string,
+	args []any,
+) ([][]any, error) {
+
+	var where string
+
+	// WHERE 构造
+	if searchCondition != "" && len(searchColumns) > 0 {
+		var conditions []string
+
+		for _, col := range searchColumns {
+			conditions = append(conditions, fmt.Sprintf("%s LIKE ?", col))
+		}
+
+		where = " WHERE (" + strings.Join(conditions, " OR ") + ")"
+
+		// 自动加 %
+		for i := range args {
+			if s, ok := args[i].(string); ok {
+				args[i] = "%" + s + "%"
+			}
+		}
+	}
+
+	// ORDER BY
+	var order string
+	if orderBy != "" {
+		order = " ORDER BY " + orderBy
+	}
+
+	sql := fmt.Sprintf("SELECT * FROM %s%s%s", viewName, where, order)
+
+	return c.query(ctx, sql, args...)
+}
+
+func (c *Client) LoadDataBySQL(
+	ctx context.Context,
+	sql string,
+	args []any,
+) ([][]any, error) {
+	return c.query(ctx, sql, args...)
+}
+
+// 统一查询方法（核心优化）
+func (c *Client) query(ctx context.Context, sql string, args ...any) ([][]any, error) {
+	sql = convertPlaceholder(sql)
+
+	rows, err := c.exec().Query(ctx, sql, args...)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	result := make([][]any, 0, 16)
+
+	for rows.Next() {
+		values, err := rows.Values()
+		if err != nil {
+			return nil, err
+		}
+
+		row := make([]any, len(values))
+		copy(row, values)
+
+		result = append(result, row)
+	}
+
+	return result, rows.Err()
+}