implemented sql flag

This commit is contained in:
marco 2016-02-06 22:46:59 +01:00
parent cd583bfe41
commit ef9a220565
2 changed files with 10 additions and 7 deletions

View File

@ -72,11 +72,9 @@ The generated code is exactly the same as the Stringer tool plus the mentioned a
The usage of Enumer is the same as Stringer, so you can refer to the [Stringer docs](https://godoc.org/golang.org/x/tools/cmd/stringer) The usage of Enumer is the same as Stringer, so you can refer to the [Stringer docs](https://godoc.org/golang.org/x/tools/cmd/stringer)
for more information. for more information.
There is only one flag added: `noJSON`. If this flag is set to true (i.e. `enumer -type=Pill -noJSON`), There are two flags added: `noJSON` and `sql`. If the noJSON flag is set to true (i.e. `enumer -type=Pill -noJSON`),
the JSON related methods won't be generated. the JSON related methods won't be generated. And if the sql flag is set to true, the Scanner and Valuer interface will
be implemented to seamlessly use the enum in a database model.
## Additional functions of this fork
This fork additionally implements the Scanner and Valuer interface to use a enum seamlessly in a database model.
## Inspiring projects ## Inspiring projects
* [Stringer](https://godoc.org/golang.org/x/tools/cmd/stringer) * [Stringer](https://godoc.org/golang.org/x/tools/cmd/stringer)

View File

@ -83,6 +83,7 @@ import (
var ( var (
typeNames = flag.String("type", "", "comma-separated list of type names; must be set") typeNames = flag.String("type", "", "comma-separated list of type names; must be set")
noJSON = flag.Bool("noJSON", false, "if true, json marshaling methods will NOT be included. Default: false") noJSON = flag.Bool("noJSON", false, "if true, json marshaling methods will NOT be included. Default: false")
sql = flag.Bool("sql", false, "if true, the Scanner and Valuer interface will be implemented.")
output = flag.String("output", "", "output file name; default srcdir/<type>_string.go") output = flag.String("output", "", "output file name; default srcdir/<type>_string.go")
) )
@ -135,10 +136,12 @@ func main() {
g.Printf("\n") g.Printf("\n")
g.Printf("import (\n") g.Printf("import (\n")
g.Printf("\t\"fmt\"\n") g.Printf("\t\"fmt\"\n")
g.Printf("\t\"database/sql/driver\"\n")
if !*noJSON { if !*noJSON {
g.Printf("\t\"encoding/json\"\n") g.Printf("\t\"encoding/json\"\n")
} }
if *sql {
g.Printf("\t\"database/sql/driver\"\n")
}
g.Printf(")\n") g.Printf(")\n")
// Run generate for each type. // Run generate for each type.
@ -324,7 +327,9 @@ func (g *Generator) generate(typeName string, includeJSON bool) {
} }
// SQL // SQL
g.addValueAndScanMethod(typeName) if *sql {
g.addValueAndScanMethod(typeName)
}
} }
// splitIntoRuns breaks the values into runs of contiguous sequences. // splitIntoRuns breaks the values into runs of contiguous sequences.