Merge pull request #160 from erizocosmico/fix/kallax-struct-tag

fix kallax struct tag for custom column names

Author: Miguel Molina

generator/types.go

@@ -591,11 +591,11 @@ func (f *Field) SetFields(sf []*Field) {
 }
 
 // ColumnName returns the SQL valid column name of the field.
-// The struct tag `column` of the field can be use to set the name, otherwise
+// The struct tag `kallax` of the field can be use to set the name, otherwise
 // is the field name converted to lower snake case.
 // If the resultant name is a reserved keyword a _ will be prepended to the name.
 func (f *Field) ColumnName() string {
-	name := f.Tag.Get("column")
+	name := strings.TrimSpace(strings.Split(f.Tag.Get("kallax"), ",")[0])
 	if name == "" {
 		name = toLowerSnakeCase(f.Name)
 	}

generator/types_test.go

@@ -85,9 +85,9 @@ func (s *FieldSuite) TestColumnName() {
 		{"", "FooBar", "foo_bar"},
 		{"", "ID", "id"},
 		{"", "References", "_references"},
-		{`column:"foo"`, "Bar", "foo"},
-		{`column:"References"`, "Bar", "_References"},
-		{`column:"references"`, "Bar", "_references"},
+		{`kallax:"foo"`, "Bar", "foo"},
+		{`kallax:"References"`, "Bar", "_References"},
+		{`kallax:"references"`, "Bar", "_references"},
 	}
 
 	for _, c := range cases {