go-db/myhandle/queryHandlerBytes.go

package myhandle

import (
	"encoding/json"
	"net/http"

	"git.x2erp.com/qdy/go-base/types"
	"git.x2erp.com/qdy/go-db/factory/database"
)

// 最简单的 queryHandler，只处理 []byte 返回
func QueryHandlerBytes(
	w http.ResponseWriter,
	r *http.Request,
	dbFactory *database.DBFactory,
	handlerFunc func(*database.DBFactory, types.QueryRequest) []byte,
) {
	// 解析请求参数
	var req types.QueryRequest
	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
		// 返回 CSV 格式的错误信息
		errorCSV := "error,Invalid request body\n"
		w.Header().Set("Content-Type", "text/csv")
		w.WriteHeader(http.StatusBadRequest)
		w.Write([]byte(errorCSV))
		return
	}

	// 调用业务逻辑函数
	csvData := handlerFunc(dbFactory, req)

	// 直接返回 CSV 数据（包含错误信息时也会被正确处理）
	w.Header().Set("Content-Type", "text/csv")
	w.Header().Set("Content-Disposition", "attachment; filename=query_result.csv")
	w.Write(csvData)
}