format code (#386)

This commit is contained in:
Kevin Wan
2021-01-14 13:24:24 +08:00
committed by GitHub
parent 744c18b7cb
commit eca4ed2cc0
16 changed files with 114 additions and 96 deletions

View File

@@ -25,7 +25,7 @@ const (
type (
defaultGenerator struct {
//source string
// source string
dir string
console.Console
pkg string
@@ -57,6 +57,7 @@ func NewDefaultGenerator(dir string, cfg *config.Config, opt ...Option) (*defaul
for _, fn := range optionList {
fn(generator)
}
return generator, nil
}
@@ -96,6 +97,7 @@ func (g *defaultGenerator) StartFromInformationSchema(db string, columns map[str
m[table.Name.Source()] = code
}
return g.createFile(m)
}
@@ -130,6 +132,7 @@ func (g *defaultGenerator) createFile(modelList map[string]string) error {
return err
}
}
// generate error file
varFilename, err := format.FileNamingFormat(g.cfg.NamingFormat, "vars")
if err != nil {
@@ -168,16 +171,15 @@ func (g *defaultGenerator) genFromDDL(source string, withCache bool) (map[string
}
m[table.Name.Source()] = code
}
return m, nil
}
type (
Table struct {
parser.Table
CacheKey map[string]Key
ContainsUniqueKey bool
}
)
type Table struct {
parser.Table
CacheKey map[string]Key
ContainsUniqueKey bool
}
func (g *defaultGenerator) genModel(in parser.Table, withCache bool) (string, error) {
if len(in.PrimaryKey.Name.Source()) == 0 {
@@ -292,5 +294,6 @@ func wrapWithRawString(v string) string {
} else if len(v) == 1 {
v = v + "`"
}
return v
}