go-service-agent/test/my0_test.go

package main

import (
	"fmt"
	"log"
	"strings"
	"sync"
	"testing"
	"time"

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

// // QueryRequest 查询请求结构体
// type QueryRequest struct {
// 	SQL              string                 `json:"sql" binding:"required"`
// 	Params           map[string]interface{} `json:"params,omitempty"`           // 名称参数
// 	PositionalParams []interface{}          `json:"positionalParams,omitempty"` // 位置参数
// }

// 分页配置
type PageConfig1 struct {
	TotalRows  int // 总共需要查询多少行
	PageSize   int // 每次返回多少行
	MaxWorkers int // 几条线程同时工作
	StartRow   int //从哪行开始查询
}

// 查询任务
type QueryTask1 struct {
	Page           int
	StartRow       int
	EndRow         int
	QuerySQL       string
	QueryParams    []interface{}
	CSVData        string
	Error          error
	QueryTime      time.Duration
	SaveTime       time.Duration
	LastClothingID string // 上一页最后一条记录的CLOTHING_ID
}

func TestQueryAndInsertToDoris1(t *testing.T) {
	// 记录总开始时间
	totalStartTime := time.Now()

	// 配置分页参数
	pageConfig := PageConfig{
		TotalRows:  100, // 总共需要查询多少行
		PageSize:   50,  // 每次查询2000条
		MaxWorkers: 2,   // 几条线程同时工作
		StartRow:   0,   //从哪行开始查询
	}

	fmt.Printf("开始执行分页查询，总共%d行，每页%d条，%d线程工作\n", pageConfig.TotalRows, pageConfig.PageSize, pageConfig.MaxWorkers)

	// 1. 获取HTTP工厂实例
	httpFactory, err := factory.GetHTTPFactory()
	if err != nil {
		t.Fatalf("Failed to get HTTP factory: %v", err)
	}
	fmt.Println("HTTP factory created successfully")

	// 7. 获取Doris工厂实例
	dorisFactory, err := factory.GetDorisFactory(httpFactory)
	if err != nil {
		t.Fatalf("Failed to get Doris factory: %v", err)
	}
	fmt.Println("Doris factory created successfully")

	// 检查Doris表结构
	fmt.Println("Checking Doris table structure...")

	// 获取Doris配置
	cfg := config.GetConfig()
	database := "X6_STOCK_DEV"
	table := "A3_CLOTHING_LOG"
	skipHeader := false // 改为true，跳过CSV头行
	url := fmt.Sprintf("http://%s:%d/api/%s/%s/_stream_load", cfg.GetDoris().FEHost, cfg.GetDoris().FEPort, database, table)
	fmt.Printf("Doris stream load URL: %s\n", url)

	// 计算需要的页数
	totalPages := (pageConfig.TotalRows + pageConfig.PageSize - 1) / pageConfig.PageSize
	fmt.Printf("预计总共需要查询 %d 页\n", totalPages)

	// 创建任务通道和结果通道
	taskChan := make(chan QueryTask, totalPages)
	resultChan := make(chan QueryTask, totalPages)
	doneChan := make(chan bool)

	var wg sync.WaitGroup
	var mu sync.Mutex

	// 执行统计变量
	totalQueryTime := time.Duration(0)
	totalSaveTime := time.Duration(0)
	totalRowsInserted := 0
	completedTasks := 0

	// 启动工作线程
	for i := 0; i < pageConfig.MaxWorkers; i++ {
		wg.Add(1)
		go func(workerID int) {
			defer wg.Done()

			// 每个worker创建自己的HTTP客户端
			httpClient := httpFactory.CreateClient()

			for task := range taskChan {
				fmt.Printf("Worker %d 处理第 %d 页 (行 %d-%d, CLOTHING_ID > %s)...\n",
					workerID, task.Page, task.StartRow, task.EndRow, task.LastClothingID)

				// 记录查询开始时间
				queryStartTime := time.Now()

				// 准备查询请求
				queryRequest := types.QueryRequest{
					SQL:              task.QuerySQL,
					PositionalParams: task.QueryParams,
					WriterHeader:     false,
				}

				// 发送POST请求到 /api/query/csv 获取CSV格式数据
				resp, err := httpClient.PostWithAuth(
					"http://localhost:8080/api/query/csv/param",
					queryRequest,
					"123", // Bearer Token
					nil,
				)

				if err != nil {
					task.Error = fmt.Errorf("第%d页查询失败: %v", task.Page, err)
					resultChan <- task
					continue
				}

				if resp.StatusCode() != 200 {
					task.Error = fmt.Errorf("第%d页查询请求失败, 状态码: %d", task.Page, resp.StatusCode())
					resultChan <- task
					continue
				}

				// 获取CSV数据
				csvData := string(resp.Body())
				if len(csvData) == 0 {
					task.Error = fmt.Errorf("第%d页没有数据", task.Page)
					resultChan <- task
					continue
				}
				log.Printf("csvData:\n%s", csvData)
				// 记录查询结束时间
				queryEndTime := time.Now()
				queryDuration := queryEndTime.Sub(queryStartTime)
				task.QueryTime = queryDuration
				task.CSVData = csvData

				// 插入数据到Doris
				saveStartTime := time.Now()
				err = dorisFactory.InsertCSV(database, table, csvData, skipHeader)
				if err != nil {
					task.Error = fmt.Errorf("第%d页数据插入Doris失败: %v", task.Page, err)
					resultChan <- task
					continue
				}

				// 记录保存结束时间
				saveEndTime := time.Now()
				saveDuration := saveEndTime.Sub(saveStartTime)
				task.SaveTime = saveDuration

				resultChan <- task
			}
		}(i + 1)
	}

	// 启动结果处理协程
	go func() {
		for task := range resultChan {
			mu.Lock()

			if task.Error != nil {
				fmt.Printf("❌ 第 %d 页处理失败: %v\n", task.Page, task.Error)
			} else {
				// 计算本页数据行数
				estimatedRows := task.EndRow - task.StartRow + 1
				// 如果数据不足，按实际估算
				if len(task.CSVData) < estimatedRows*50 { // 保守估计每行至少50字符
					estimatedRows = len(task.CSVData) / 50
				}

				totalQueryTime += task.QueryTime
				totalSaveTime += task.SaveTime
				totalRowsInserted += estimatedRows
				completedTasks++

				fmt.Printf("✅ Worker 完成第 %d 页 (行 %d-%d, CLOTHING_ID > %s)\n",
					task.Page, task.StartRow, task.EndRow, task.LastClothingID)
				fmt.Printf("   查询耗时: %v, 保存耗时: %v\n", task.QueryTime, task.SaveTime)
				fmt.Printf("   估算数据行数: %d\n", estimatedRows)

				// 从CSV数据中提取最后一行的CLOTHING_ID
				if task.CSVData != "" {
					// 简单实现：取最后一行第一列的第一个记录
					// 这里假设CSV格式是逗号分隔，没有标题行
					lines := strings.Split(strings.TrimSpace(task.CSVData), "\n")
					if len(lines) > 0 {
						// 取最后一行（如果有多行数据）
						lastLine := lines[len(lines)-1]
						fields := strings.Split(lastLine, ",")
						if len(fields) > 0 {
							// 更新任务的LastClothingID
							task.LastClothingID = strings.TrimSpace(fields[0]) // CLOTHING_ID是第一列
							fmt.Printf("   提取到最后一个CLOTHING_ID: %s\n", task.LastClothingID)
						}
					}
				}
			}

			mu.Unlock()

			// 如果所有任务都完成了，发送完成信号
			if completedTasks >= totalPages {
				doneChan <- true
				break
			}
		}
	}()

	// 生成任务并发送到任务通道
	fmt.Println("\n📋 开始生成查询任务...")

	// 初始CLOTHING_ID为"0"
	lastClothingID := "A"

	for page := 1; page <= totalPages; page++ {
		startRow := pageConfig.StartRow + (page-1)*pageConfig.PageSize + 1
		endRow := pageConfig.StartRow + page*pageConfig.PageSize

		// 检查不超过要查询的总行数
		maxEndRow := pageConfig.StartRow + pageConfig.TotalRows
		if endRow > maxEndRow {
			endRow = maxEndRow
		}

		// 生成查询SQL（使用参数模式）
		querySQL, queryParams := getSQLWithPagination(startRow, endRow, lastClothingID)

		task := QueryTask{
			Page:           page,
			StartRow:       startRow,
			EndRow:         endRow,
			QuerySQL:       querySQL,
			QueryParams:    queryParams,
			LastClothingID: lastClothingID,
		}

		fmt.Printf("生成第 %d 页任务 (行 %d-%d), 使用CLOTHING_ID > '%s'\n",
			page, startRow, endRow, lastClothingID)
		taskChan <- task

		// 注意：在实际场景中，lastClothingID需要在任务完成后更新
		// 这里简化处理，每个任务使用相同的lastClothingID
		// 更好的做法是使用任务队列和结果回调来更新lastClothingID
	}

	// 关闭任务通道，通知worker没有更多任务
	close(taskChan)

	// 等待所有worker完成
	wg.Wait()

	// 关闭结果通道
	close(resultChan)

	// 等待结果处理完成
	<-doneChan

	// 记录总结束时间
	totalEndTime := time.Now()
	totalDuration := totalEndTime.Sub(totalStartTime)

	// 打印性能统计
	fmt.Println("\n📊 性能统计:")
	fmt.Printf("   完成页数: %d/%d\n", completedTasks, totalPages)
	fmt.Printf("   总查询耗时: %v\n", totalQueryTime)
	fmt.Printf("   总保存耗时: %v\n", totalSaveTime)
	fmt.Printf("   总耗时: %v\n", totalDuration)
	fmt.Printf("   估算插入总行数: %d\n", totalRowsInserted)
	if completedTasks > 0 {
		fmt.Printf("   平均每页查询耗时: %v\n", totalQueryTime/time.Duration(completedTasks))
		fmt.Printf("   平均每页保存耗时: %v\n", totalSaveTime/time.Duration(completedTasks))
		fmt.Printf("   平均每秒处理行数: %.2f\n", float64(totalRowsInserted)/totalDuration.Seconds())
	}

	fmt.Println("✅ 所有数据成功插入到 Doris!")
}

// getSQLWithPagination 生成带分页的SQL语句（参数模式）
// 返回SQL语句和参数映射
func getSQLWithPagination1(startRow, endRow int, lastClothingID string) (string, []interface{}) {
	sql := `SELECT
    CLOTHING_ID,
    CLOTHING_YEAR,
    CLOTHING_NAME

   FROM (
    SELECT a.*, ROWNUM as rn
    FROM (
        SELECT *
        FROM X6_STOCK_DEV.A3_CLOTHING 
        WHERE CLOTHING_ID > :1
        ORDER BY CLOTHING_ID
    ) a
    WHERE ROWNUM <= :2
)
WHERE rn > :3`

	// 创建参数映射
	params := []interface{}{
		lastClothingID,
		endRow,
		startRow - 1, // WHERE rn > :start_row 所以是startRow-1
	}

	return sql, params
}