增加参数。csv是否包含表头

factory/db_factory.go

@@ -131,21 +131,21 @@ func (f *DBFactory) QueryParamsNameToJSON(sql string, params map[string]interfac
 }
 
 // QueryToCSV 快捷查询，直接返回 CSV 字符串（包含表头）
-func (f *DBFactory) QueryToCSV(sql string) ([]byte, error) {
-	return functions.QueryToCSV(f.db, sql)
+func (f *DBFactory) QueryToCSV(sql string, writerHeader bool) ([]byte, error) {
+	return functions.QueryToCSV(f.db, sql, writerHeader)
 }
 
 // QueryParamsToCSV 位置参数查询并返回 CSV 字节数据
-func (f *DBFactory) QueryPositionalToCSV(sql string, params []interface{}) ([]byte, error) {
+func (f *DBFactory) QueryPositionalToCSV(sql string, writerHeader bool, params []interface{}) ([]byte, error) {
 
-	return functions.QueryPositionalToCSV(f.db, sql, params)
+	return functions.QueryPositionalToCSV(f.db, sql, writerHeader, params)
 }
 
 // QueryParamsNameToCSV 命名参数查询并返回 CSV 字节数据
 // params 可以是 map[string]interface{} 或结构体
-func (f *DBFactory) QueryParamsNameToCSV(sql string, params map[string]interface{}) ([]byte, error) {
+func (f *DBFactory) QueryParamsNameToCSV(sql string, writerHeader bool, params map[string]interface{}) ([]byte, error) {
 
-	return functions.QueryParamsNameToCSV(f.db, sql, params)
+	return functions.QueryParamsNameToCSV(f.db, sql, writerHeader, params)
 }
 
 // ExecuteDDL 快捷执行DDL语句

functions/query_csv.go

@@ -10,7 +10,7 @@ import (
 )
 
 // QueryToCSV 无参数查询并返回 CSV 字节数据
-func QueryToCSV(db *sqlx.DB, sql string) ([]byte, error) {
+func QueryToCSV(db *sqlx.DB, sql string, writerHeader bool) ([]byte, error) {
 	if sql == "" {
 		return nil, fmt.Errorf("SQL query cannot be empty")
 	}
@@ -20,11 +20,11 @@ func QueryToCSV(db *sqlx.DB, sql string) ([]byte, error) {
 		return nil, fmt.Errorf("query execution failed: %v", err)
 	}
 
-	return rowsToCSV(rows)
+	return rowsToCSV(rows, writerHeader)
 }
 
 // QueryParamsToCSV 位置参数查询并返回 CSV 字节数据
-func QueryPositionalToCSV(db *sqlx.DB, sql string, params []interface{}) ([]byte, error) {
+func QueryPositionalToCSV(db *sqlx.DB, sql string, writerHeader bool, params []interface{}) ([]byte, error) {
 	if sql == "" {
 		return nil, fmt.Errorf("SQL query cannot be empty")
 	}
@@ -34,12 +34,12 @@ func QueryPositionalToCSV(db *sqlx.DB, sql string, params []interface{}) ([]byte
 		return nil, fmt.Errorf("query execution failed: %v", err)
 	}
 
-	return rowsToCSV(rows)
+	return rowsToCSV(rows, writerHeader)
 }
 
 // QueryParamsNameToCSV 命名参数查询并返回 CSV 字节数据
 // params 可以是 map[string]interface{} 或结构体
-func QueryParamsNameToCSV(db *sqlx.DB, sql string, params map[string]interface{}) ([]byte, error) {
+func QueryParamsNameToCSV(db *sqlx.DB, sql string, writerHeader bool, params map[string]interface{}) ([]byte, error) {
 	if sql == "" {
 		return nil, fmt.Errorf("SQL query cannot be empty")
 	}
@@ -55,11 +55,11 @@ func QueryParamsNameToCSV(db *sqlx.DB, sql string, params map[string]interface{}
 		return nil, fmt.Errorf("query execution failed: %v", err)
 	}
 
-	return rowsToCSV(rows)
+	return rowsToCSV(rows, writerHeader)
 }
 
-// / rowsToCSV 公共方法：将查询结果转换为 CSV 字节数据
-func rowsToCSV(rows *sql.Rows) ([]byte, error) {
+// rowsToCSV 公共方法：将查询结果转换为 CSV 字节数据
+func rowsToCSV(rows *sql.Rows, writerHeader bool) ([]byte, error) {
 	defer rows.Close()
 
 	columns, err := rows.Columns()
@@ -70,9 +70,11 @@ func rowsToCSV(rows *sql.Rows) ([]byte, error) {
 	var builder strings.Builder
 	writer := csv.NewWriter(&builder)
 
-	// 写入表头
-	if err := writer.Write(columns); err != nil {
-		return nil, fmt.Errorf("failed to write CSV header: %v", err)
+	// 根据参数决定是否写入表头
+	if writerHeader {
+		if err := writer.Write(columns); err != nil {
+			return nil, fmt.Errorf("failed to write CSV header: %v", err)
+		}
 	}
 
 	for rows.Next() {

functions/query_json.go

@@ -40,8 +40,8 @@ func QueryParamsNameToJSON(db *sqlx.DB, sql string, params map[string]interface{
 func QueryPositionalToJSON(db *sqlx.DB, sql string, params []interface{}) *types.QueryResult {
 	startTime := time.Now()
 
-	fmt.Printf("positionalParams: %s", params)
-	fmt.Printf("sql: %s", sql)
+	//fmt.Printf("positionalParams: %s", params)
+	//fmt.Printf("sql: %s", sql)
 	if sql == "" {
 		return createErrorResult("SQL query cannot be empty", startTime)
 	}