cgimport/main.go

package main

import (
	"fmt"
	"os"

	"git.wecise.com/wecise/cgimport/importer"
	"git.wecise.com/wecise/cgimport/odbc"
)

// 获取配置信息
//
//	默认配置：
//		当前工作目录下 与应用同名的 .conf 文件
//		环境变量
//		命令行参数
var mcfg = odbc.Config
var logger = odbc.Logger

func init() {
	odbc.CommandArgsInfo = func() string {
		return `
datapath=data      # 指定数据文件路径
reload=true        # 重新导入所有数据，不指定则跳过已经导入的文件`
	}
}
func main() {
	if mcfg.GetString("help") != "" {
		fmt.Println(odbc.Usage())
		os.Exit(0)
	}
	// 配置参数
	// 文件目录
	datapath := mcfg.GetString("datapath", mcfg.GetString("cgi.datapath", "data"))
	// 并发数
	parallel := mcfg.GetInt("parallel", mcfg.GetInt("cgi.parallel", 20))
	//
	reload := mcfg.GetBool("reload") || mcfg.GetString("reload") == "reload"
	//
	logger.Info("datapath:   ", datapath)
	logger.Info("parallel:   ", parallel)
	logger.Info("reload:     ", reload)
	// 导入
	totalfilescount, totalrecordscount, totalusetime, filescount, recordscount, usetime, e := importer.ImportDir(datapath, parallel, reload)
	if e != nil {
		logger.Error(e)
		panic(e)
	}
	if totalfilescount == 0 {
		fmt.Println(`not found data files in "` + datapath + `"`)
		return
	}
	// 输出统计信息
	fmt.Println("import", filescount, "files", recordscount, "records", "in", usetime)
	fmt.Println("total import", totalfilescount, "files", totalrecordscount, "records", "in", totalusetime)
	fmt.Println("access", odbc.LogFile, "for detail information")

	// 验证
	importer.Check()
	os.Exit(0)
}