feat: support custom tag

generator.go

@@ -14,6 +14,7 @@ type input struct {
 	hclPath string
 	outPath string
 	pkg     string
+	tag     string
 }
 
 func generate(s schema.Schema, in input) ([]byte, error) {
@@ -29,7 +30,7 @@ func generate(s schema.Schema, in input) ([]byte, error) {
 		for j := range table.Columns {
 			column := table.Columns[j]
 			tp := toGoTypeString(column.Type)
-			fmt.Fprintf(buf, "\t%s\t%s\t`db:\"%s\"`\n", toCamelCase(column.Name), tp, column.Name)
+			fmt.Fprintf(buf, "\t%s\t%s\t`%s:\"%s\"`\n", toCamelCase(column.Name), tp, in.tag, column.Name)
 		}
 		fmt.Fprintf(buf, "}\n\n")
 	}

generator_test.go

@@ -32,6 +32,7 @@ func Test_generate(t *testing.T) {
 				hclPath: "input.hcl",
 				outPath: "output.go",
 				pkg:     "main",
+				tag:     "db",
 			},
 			out: `// Code generated by github.com/ucpr/atlas-hcl-gen-go. DO NOT EDIT.
 // atlas-hcl-gen-go:

main.go

@@ -27,10 +27,11 @@ func main() {
 }
 
 func run() error {
-	var hclPath, outPath, target, pkg string
+	var hclPath, outPath, target, tag, pkg string
 	flag.StringVar(&hclPath, "i", "", "input file path")
 	flag.StringVar(&outPath, "o", "", "output file path")
 	flag.StringVar(&target, "t", "mysql", "target database")
+	flag.StringVar(&tag, "tag", "db", "tag name")
 	flag.StringVar(&pkg, "package", "main", "package name")
 	flag.Parse()
 
@@ -54,6 +55,7 @@ func run() error {
 		hclPath: hclPath,
 		outPath: outPath,
 		pkg:     pkg,
+		tag:     tag,
 	})
 	if err != nil {
 		return fmt.Errorf("failed to generate: %w", err)