77 lines
1.9 KiB
Go
77 lines
1.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/fatih/color"
|
|
jsoniter "github.com/json-iterator/go"
|
|
"github.com/spf13/cobra"
|
|
"github.com/yaoapp/gou"
|
|
"github.com/yaoapp/yao/config"
|
|
"github.com/yaoapp/yao/engine"
|
|
)
|
|
|
|
var importRenew bool
|
|
var importChunk int
|
|
var importCmd = &cobra.Command{
|
|
Use: "import",
|
|
Short: "导入数据(alpha)",
|
|
Long: `导入CSV/Excel数据`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
Boot()
|
|
|
|
if len(args) < 3 {
|
|
fmt.Fprintln(os.Stderr, "参数错误", args)
|
|
fmt.Println("xiang import <数据模型> <文件名称> <导入配置>")
|
|
os.Exit(1)
|
|
}
|
|
|
|
name := args[0]
|
|
filename := args[1]
|
|
configfile := args[2]
|
|
|
|
// 检查数据文件
|
|
if _, err := os.Stat(filename); os.IsNotExist(err) {
|
|
fmt.Println(color.RedString("数据文件不存在: %s", filename))
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, err := os.Stat(configfile); os.IsNotExist(err) {
|
|
fmt.Println(color.RedString("配置文件不存在: %s", configfile))
|
|
os.Exit(1)
|
|
}
|
|
|
|
content, err := ioutil.ReadFile(configfile)
|
|
if err != nil {
|
|
fmt.Println(color.RedString("读取配置文件失败: %s", err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
impt := gou.NewImport(importChunk)
|
|
err = jsoniter.Unmarshal(content, impt)
|
|
if err != nil {
|
|
fmt.Println(color.RedString("解析配置文件=失败: %s", err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
// 加载数据模型
|
|
engine.Load(config.Conf)
|
|
if importRenew {
|
|
gou.Select(name).Migrate(true)
|
|
}
|
|
|
|
resp := impt.InsertCSV(filename, name)
|
|
for _, err := range *resp.Errors {
|
|
fmt.Println(color.RedString("%s line: %d", err.Message, err.Line))
|
|
}
|
|
fmt.Println(color.WhiteString("成功: %d, 失败: %d, 总共: %d", resp.Success, resp.Failure, resp.Total))
|
|
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
importCmd.PersistentFlags().BoolVarP(&importRenew, "renew", "r", false, "清空数据模型原有数据")
|
|
importCmd.PersistentFlags().IntVarP(&importChunk, "chunk", "c", 1000, "每次处理数量(数值越大,处理越快，内存占用越大)")
|
|
}
|