pixeldrain/gocqlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Added tests, example and updated a few types

Browse Source
This commit is contained in:
Pavle Kosutic
2021-12-12 09:34:17 +01:00
committed by Michal Jan Matczuk
parent 2942397ab6
commit a62ba24cf9
5 changed files with 250 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cmd/schemagen/schemagen.go
View File

@@ -5,6 +5,7 @@ import (
_ "embed"
"flag"
"fmt"
"go/format"
"html/template"
"io/ioutil"
"log"
@@ -84,7 +85,13 @@ func renderTemplate(md *gocql.KeyspaceMetadata) ([]byte, error) {
imports := make([]string, 0)
for _, t := range md.Tables {
for _, c := range t.Columns {
if c.Validator == "uuid" && !existsInSlice(imports, "github.com/gocql/gocql") {
if (c.Validator == "timestamp" || c.Validator == "date" || c.Validator == "duration" || c.Validator == "time") && !existsInSlice(imports, "time") {
imports = append(imports, "time")
}
if c.Validator == "decimal" && !existsInSlice(imports, "gopkg.in/inf.v0") {
imports = append(imports, "gopkg.in/inf.v0")
}
if c.Validator == "duration" && !existsInSlice(imports, "github.com/gocql/gocql") {
imports = append(imports, "github.com/gocql/gocql")
}
}
@@ -101,8 +108,7 @@ func renderTemplate(md *gocql.KeyspaceMetadata) ([]byte, error) {
if err = t.Execute(buf, data); err != nil {
return nil, fmt.Errorf("template: %w", err)
}
//return format.Source(buf.Bytes())
return buf.Bytes(), nil
return format.Source(buf.Bytes())
}
func createSession() (gocqlx.Session, error) {