package l_export_async

import (
	"fmt"
	"os"
	"testing"
)

func Test_Merge(t *testing.T) {
	path := "/tmp/f63b2047-dd7b-11f0-b1a8-00155d5ef0f92960656377/csv/"
	outputDir := "/tmp/f63b2047-dd7b-11f0-b1a8-00155d5ef0f92960656377/xlsx/aaa.xlsx"
	csvFiles, err := listFiles(path)
	if err != nil {
		panic(err)
	}
	m := NewMerge(
		Reader{Files: csvFiles, Index: len(csvFiles) - 1},
		Writer{File: outputDir, Limit: 30000, BufferSize: 3000},
		NewLogPrint(nil),
	)
	err = m.Merge()
	t.Log(err)
}

func listFiles(dirPath string) ([]string, error) {
	entries, err := os.ReadDir(dirPath)
	if err != nil {
		return nil, fmt.Errorf("failed to read directory: %w", err)
	}

	var files []string
	for _, entry := range entries {
		if !entry.IsDir() { // 只添加文件（排除子目录）
			files = append(files, dirPath+entry.Name())
		}
	}
	return files, nil
}