|
|
package main
|
|
|
|
|
|
import (
|
|
|
"bufio"
|
|
|
"bytes"
|
|
|
"dsTools/Utils/DbUtil"
|
|
|
"dsTools/Utils/FileUtil"
|
|
|
"encoding/json"
|
|
|
"fmt"
|
|
|
"io"
|
|
|
"os"
|
|
|
"strings"
|
|
|
)
|
|
|
|
|
|
func main() {
|
|
|
//1、定义参加创建默认mapping的表有哪些
|
|
|
var createDefaultMappingTables []string
|
|
|
//读取
|
|
|
f, err := os.Open("./Config/createDefaultMapping.txt")
|
|
|
if err != nil {
|
|
|
fmt.Println("没有找到创建默认mapping的数据文件createDefaultMapping.txt！")
|
|
|
return
|
|
|
}
|
|
|
buf := bufio.NewReader(f)
|
|
|
var shouldBreak = false
|
|
|
for {
|
|
|
line, err := buf.ReadString('\n')
|
|
|
if err != nil || io.EOF == err {
|
|
|
shouldBreak = true
|
|
|
}
|
|
|
line = strings.TrimSpace(line)
|
|
|
if strings.Index(line, "#") < 0 && len(line) > 0 {
|
|
|
createDefaultMappingTables = append(createDefaultMappingTables, line)
|
|
|
}
|
|
|
if shouldBreak {
|
|
|
break
|
|
|
}
|
|
|
}
|
|
|
//头部
|
|
|
head := `{
|
|
|
"settings": {
|
|
|
"number_of_shards": 5,
|
|
|
"index.refresh_interval": "5s"
|
|
|
},
|
|
|
"aliases": [
|
|
|
"#indexName#"
|
|
|
],
|
|
|
"mappings": {
|
|
|
"properties": {`
|
|
|
|
|
|
tail := `}}}`
|
|
|
//遍历每一个表
|
|
|
for i := 0; i < len(createDefaultMappingTables); i++ {
|
|
|
tableName := createDefaultMappingTables[i]
|
|
|
//到mysql中查询
|
|
|
sql := "desc " + tableName
|
|
|
list, err := DbUtil.Engine.QueryInterface(sql)
|
|
|
if err != nil {
|
|
|
fmt.Println(err)
|
|
|
}
|
|
|
//这个是在数据中心区别各地上报数据的主键字段，是拼接出来的
|
|
|
str := `"id":{"type":"keyword"},`
|
|
|
//处理每一个字段
|
|
|
for j := 0; j < len(list); j++ {
|
|
|
c := list[j]
|
|
|
field := string(c["Field"].([]byte))
|
|
|
Type := string(c["Type"].([]byte))
|
|
|
Key := string(c["Key"].([]byte))
|
|
|
//如果是主键，那么mapping的字段名称为pk
|
|
|
if Key == "PRI" {
|
|
|
field = "pk"
|
|
|
}
|
|
|
if strings.Index(Type, "int") >= 0 {
|
|
|
str += `"` + field + `": {
|
|
|
"type": "long"
|
|
|
},`
|
|
|
} else if (strings.Index(Type, "datetime")) >= 0 {
|
|
|
str += `"` + field + `": {
|
|
|
"format":"yyyy-MM-dd HH:mm:ss",
|
|
|
"type": "date"
|
|
|
},`
|
|
|
} else if (strings.Index(Type, "date")) >= 0 {
|
|
|
str += `"` + field + `": {
|
|
|
"format":"yyyy-MM-dd",
|
|
|
"type": "date"
|
|
|
},`
|
|
|
} else if (strings.Index(Type, "timestamp")) >= 0 {
|
|
|
str += `"` + field + `": {
|
|
|
"format":"yyyy-MM-dd HH:mm:ss",
|
|
|
"type": "date"
|
|
|
},`
|
|
|
} else {
|
|
|
str += `"` + field + `": {"type": "keyword"},`
|
|
|
}
|
|
|
}
|
|
|
//添加表名
|
|
|
str += `"table_name":{"type": "keyword"}`
|
|
|
//替换别名
|
|
|
head=strings.Replace(head,"#indexName#",tableName,-1)
|
|
|
//输出
|
|
|
p := "MappingDefaultJson/"
|
|
|
//添加头与尾
|
|
|
mappingStr := head + str + tail
|
|
|
|
|
|
//格式化json
|
|
|
var str2 bytes.Buffer
|
|
|
_ = json.Indent(&str2, []byte(mappingStr), "", " ")
|
|
|
//写入json文件
|
|
|
FileUtil.WriteToFile(p+tableName+".json", str2.String())
|
|
|
fmt.Println("正在生成" + tableName + "...")
|
|
|
}
|
|
|
fmt.Println("恭喜，所有初始化mapping文件成功生成！")
|
|
|
}
|
