|
|
package main
|
|
|
|
|
|
import (
|
|
|
"dsSupport/Utils/CommonUtil"
|
|
|
"dsSupport/Utils/DbUtil"
|
|
|
"dsSupport/Utils/PgUtil"
|
|
|
"fmt"
|
|
|
)
|
|
|
|
|
|
var db = DbUtil.Engine
|
|
|
var pgDb = PgUtil.Engine
|
|
|
|
|
|
func main() {
|
|
|
//1、读取每一张需要创建的表
|
|
|
sql := "select * from t_dw_table"
|
|
|
list, _ := db.SQL(sql).Query().List()
|
|
|
|
|
|
for i := range list {
|
|
|
tableId := list[i]["table_id"].(int64)
|
|
|
tableName := list[i]["table_name"].(string)
|
|
|
createTableSql := `DROP TABLE IF EXISTS "public"."` + tableName + `";` + "\r\n"
|
|
|
pgDb.SQL(createTableSql).Execute()
|
|
|
|
|
|
createTableSql = `CREATE TABLE "public"."` + tableName + `" (` + "\r\n"
|
|
|
//2、根据表名，获取相应的创建表的信息
|
|
|
sql = `select * from t_dw_table_field where table_id=?`
|
|
|
list2, _ := db.SQL(sql, tableId).Query().List()
|
|
|
for i2 := range list2 {
|
|
|
dataType := list2[i2]["data_type"].(string)
|
|
|
fieldLength := list2[i2]["field_length"].(int64)
|
|
|
isNull := list2[i2]["is_null"].(int64)
|
|
|
fieldName := list2[i2]["field_name"].(string)
|
|
|
var c = ""
|
|
|
if dataType == "int" {
|
|
|
c = "int" + CommonUtil.ConvertInt64ToString(fieldLength)
|
|
|
} else {
|
|
|
c = dataType + "(" + CommonUtil.ConvertInt64ToString(fieldLength) + ") " + ` COLLATE "pg_catalog"."default"`
|
|
|
}
|
|
|
if isNull == 0 {
|
|
|
c += " NOT NULL,"
|
|
|
} else {
|
|
|
c += ","
|
|
|
}
|
|
|
createTableSql += `"` + fieldName + `" ` + c + "\r\n"
|
|
|
}
|
|
|
createTableSql=createTableSql[0:len(createTableSql)-3]
|
|
|
createTableSql += `);` + "\r\n"
|
|
|
_, err := pgDb.SQL(createTableSql).Execute()
|
|
|
if err != nil {
|
|
|
fmt.Println(createTableSql)
|
|
|
panic(err.Error())
|
|
|
}
|
|
|
//主键有哪些
|
|
|
var pks = ""
|
|
|
for i2 := range list2 {
|
|
|
fieldName := list2[i2]["field_name"].(string)
|
|
|
isPk := list2[i2]["is_pk"].(int64)
|
|
|
if isPk == 1 {
|
|
|
pks += fieldName + ","
|
|
|
}
|
|
|
}
|
|
|
//去掉最后一个逗号
|
|
|
pks = pks[0 : len(pks)-1]
|
|
|
//添加主键描述
|
|
|
createTableSql = `ALTER TABLE "public"."` + tableName + `" ADD CONSTRAINT "` + tableName + `_dw_pkey" PRIMARY KEY ("` + pks + `");` + "\r\n"
|
|
|
_, err = pgDb.SQL(createTableSql).Execute()
|
|
|
if err != nil {
|
|
|
panic(err.Error())
|
|
|
}
|
|
|
|
|
|
fmt.Println("恭喜，数据仓库中相应表格创建完毕！")
|
|
|
}
|
|
|
}
|
