78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/manifoldco/promptui"
|
|
)
|
|
|
|
type opType int32
|
|
|
|
const (
|
|
Add opType = 1
|
|
OverWrite opType = 2
|
|
)
|
|
|
|
var opIndex = map[string]opType{
|
|
"新增": Add,
|
|
"覆盖(有则改，无则增)": OverWrite,
|
|
}
|
|
|
|
type Set struct {
|
|
Op opType
|
|
OverWriteJudColumns []string
|
|
}
|
|
|
|
func setOp(title []string) *Set {
|
|
var set = new(Set)
|
|
for {
|
|
prompt := promptui.Select{
|
|
Label: "请选择需要执行的操作: ",
|
|
Items: []string{"新增"}, // "覆盖(有则改，无则增)"
|
|
}
|
|
_, r, err := prompt.Run()
|
|
if err != nil {
|
|
warning("Prompt failed %v\n", err)
|
|
continue
|
|
}
|
|
set.Op = opIndex[r]
|
|
break
|
|
}
|
|
if set.Op == OverWrite {
|
|
selector := append(title, "继续")
|
|
selector[2] = "继续"
|
|
for {
|
|
r := columnSelect(selector)
|
|
if r != "继续" {
|
|
set.OverWriteJudColumns = append(set.OverWriteJudColumns, r)
|
|
selector = removeElement(selector, r)
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
}
|
|
return set
|
|
}
|
|
|
|
func columnSelect(selector []string) string {
|
|
prompt := promptui.Select{
|
|
Label: "请选择判重字段（可以选择多个，`继续`选项表示选择完了）",
|
|
Items: selector,
|
|
Size: len(selector) + 1,
|
|
}
|
|
_, r, err := prompt.Run()
|
|
if err != nil {
|
|
fatal("Prompt failed %v\n", err)
|
|
|
|
}
|
|
return r
|
|
}
|
|
|
|
func removeElement(slice []string, element string) []string {
|
|
result := slice[:0] // 创建一个空切片，与原切片共享底层数组
|
|
for _, v := range slice {
|
|
if v != element {
|
|
result = append(result, v)
|
|
}
|
|
}
|
|
return result
|
|
}
|