1
0
mirror of https://github.com/ManyakRus/crud_generator.git synced 2024-12-23 12:44:13 +02:00
crud_generator/pkg/dbmeta/meta_unknown.go
Nikitin Aleksandr a6ea64ee3e новый
2023-10-24 18:03:04 +03:00

91 lines
2.0 KiB
Go

package dbmeta
import (
"database/sql"
"fmt"
"regexp"
"strconv"
"strings"
"github.com/jimsmart/schema"
)
// LoadUnknownMeta fetch db meta data for unknown database type
func LoadUnknownMeta(db *sql.DB, sqlType, sqlDatabase, tableName string) (DbTableMeta, error) {
m := &dbTableMeta{
sqlType: sqlType,
sqlDatabase: sqlDatabase,
tableName: tableName,
}
cols, err := schema.ColumnTypes(db, sqlDatabase, tableName)
if err != nil {
return nil, err
}
m.columns = make([]*columnMeta, len(cols))
infoSchema, err := LoadTableInfoFromMSSqlInformationSchema(db, tableName)
if err != nil {
fmt.Printf("NOTICE unable to load InformationSchema table: %s error: %v\n", tableName, err)
}
for i, v := range cols {
nullable, ok := v.Nullable()
if !ok {
nullable = false
}
isAutoIncrement := false
isPrimaryKey := i == 0
colDDL := v.DatabaseTypeName()
defaultVal := ""
columnType, columnLen := ParseSQLType(v.DatabaseTypeName())
if columnLen == -1 {
dbType := strings.ToLower(v.DatabaseTypeName())
if strings.Contains(dbType, "varchar") {
re := regexp.MustCompile(`[-]?\d[\d,]*[\.]?[\d{2}]*`)
submatchall := re.FindAllString(dbType, -1)
if len(submatchall) > 0 {
i, err := strconv.Atoi(submatchall[0])
if err == nil {
columnLen = int64(i)
}
}
}
}
if infoSchema != nil {
infoSchemaColInfo, ok := infoSchema[v.Name()]
if ok {
if infoSchemaColInfo.ColumnDefault != nil {
defaultVal = BytesToString(infoSchemaColInfo.ColumnDefault.([]uint8))
defaultVal = cleanupDefault(defaultVal)
}
}
}
colMeta := &columnMeta{
index: i,
name: v.Name(),
databaseTypeName: columnType,
nullable: nullable,
isPrimaryKey: isPrimaryKey,
isAutoIncrement: isAutoIncrement,
colDDL: colDDL,
defaultVal: defaultVal,
columnType: columnType,
columnLen: columnLen,
}
m.columns[i] = colMeta
}
m.ddl = BuildDefaultTableDDL(tableName, m.columns)
m = updateDefaultPrimaryKey(m)
return m, nil
}