yves/sqldb
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

added unsigned int to mysql types

Browse Source
This commit is contained in:
ycc 2023-07-04 09:46:33 +02:00
parent 5ce4f94524
commit 7a4c6eb35d
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
db.go
View File

@ -130,7 +130,7 @@ func (db *Db) QueryAssociativeArray(query string) (Rows, error) {
return nil, err return nil, err
} }
m[colName] = i m[colName] = i
case "UNSIGNED BIGINT": case "UNSIGNED BIGINT", "UNSIGNED INT":
u, err := strconv.ParseUint(fmt.Sprintf("%s", *val), 10, 64) u, err := strconv.ParseUint(fmt.Sprintf("%s", *val), 10, 64)
if err != nil { if err != nil {
return nil, err return nil, err
@ -615,7 +615,8 @@ func buildLinks(schema []TableInfo) []Link {
return links return links
} }
func (db *Db) GenerateTemplate(templateFilename string, generatedFilename string) error { // Generate templates from a schema
func (db *Db) GenerateSchemaTemplate(templateFilename string, generatedFilename string) error {
schema, err := db.GetSchema() schema, err := db.GetSchema()
if err != nil { if err != nil {
log.Println(err) log.Println(err)
@ -648,6 +649,7 @@ func (db *Db) GenerateTemplate(templateFilename string, generatedFilename string
return nil return nil
} }
// Generate tables
func (db *Db) GenerateTableTemplates(templateFilename string, outputFolder string, extension string) error { func (db *Db) GenerateTableTemplates(templateFilename string, outputFolder string, extension string) error {
schema, err := db.GetSchema() schema, err := db.GetSchema()
if err != nil { if err != nil {
@ -683,7 +685,7 @@ func FormatForSQL(datatype string, value interface{}) string {
if !strings.Contains(datatype, "char") && len(strval) == 0 { if !strings.Contains(datatype, "char") && len(strval) == 0 {
return "NULL" return "NULL"
} }
if strings.Contains(datatype, "char") || strings.Contains(datatype, "date") || strings.Contains(datatype, "timestamp") { if strings.Contains(datatype, "char") || strings.Contains(datatype, "text") || strings.Contains(datatype, "date") || strings.Contains(datatype, "timestamp") {
return fmt.Sprint(pq.QuoteLiteral(strval)) return fmt.Sprint(pq.QuoteLiteral(strval))
} }
return fmt.Sprint(strval) return fmt.Sprint(strval)

2
my_test.go
View File

@ -217,7 +217,7 @@ func TestMySaveSchema(t *testing.T) {
func TestMyGenerateTemplate(t *testing.T) { func TestMyGenerateTemplate(t *testing.T) {
db := Open("mysql", "test:test@tcp(127.0.0.1:3306)/test?parseTime=true") db := Open("mysql", "test:test@tcp(127.0.0.1:3306)/test?parseTime=true")
defer db.Close() defer db.Close()
err := db.GenerateTemplate("plantuml.tmpl", "schema.puml") err := db.GenerateSchemaTemplate("plantuml.tmpl", "schema.puml")
if err != nil { if err != nil {
fmt.Println(err.Error()) fmt.Println(err.Error())
} }

2
pg_test.go
View File

@ -217,7 +217,7 @@ func TestPgSaveSchema(t *testing.T) {
func TestPgGenerateTemplate(t *testing.T) { func TestPgGenerateTemplate(t *testing.T) {
db := Open("postgres", "host=127.0.0.1 port=5432 user=test password=test dbname=test sslmode=disable") db := Open("postgres", "host=127.0.0.1 port=5432 user=test password=test dbname=test sslmode=disable")
defer db.Close() defer db.Close()
err := db.GenerateTemplate("plantuml.tmpl", "schema.puml") err := db.GenerateSchemaTemplate("plantuml.tmpl", "schema.puml")
if err != nil { if err != nil {
fmt.Println(err.Error()) fmt.Println(err.Error())
} }