@@ -35,6 +35,13 @@ func identifier(id string) string {
3535 return strings .ToLower (id )
3636}
3737
38+ func getTypeName (t parser.IType_nameContext ) string {
39+ if t == nil {
40+ return "any"
41+ }
42+ return t .GetText ()
43+ }
44+
3845func NewIdentifier (t string ) * ast.String {
3946 return & ast.String {Str : identifier (t )}
4047}
@@ -72,10 +79,8 @@ func (c *cc) convertAlter_table_stmtContext(n *parser.Alter_table_stmtContext) a
7279 Name : & name ,
7380 Subtype : ast .AT_AddColumn ,
7481 Def : & ast.ColumnDef {
75- Colname : name ,
76- TypeName : & ast.TypeName {
77- Name : def .Type_name ().GetText (),
78- },
82+ Colname : name ,
83+ TypeName : & ast.TypeName {Name : getTypeName (def .Type_name ())},
7984 IsNotNull : hasNotNullConstraint (def .AllColumn_constraint ()),
8085 },
8186 })
@@ -113,14 +118,10 @@ func (c *cc) convertCreate_table_stmtContext(n *parser.Create_table_stmtContext)
113118 }
114119 for _ , idef := range n .AllColumn_def () {
115120 if def , ok := idef .(* parser.Column_defContext ); ok {
116- typeName := "any"
117- if def .Type_name () != nil {
118- typeName = def .Type_name ().GetText ()
119- }
120121 stmt .Cols = append (stmt .Cols , & ast.ColumnDef {
121122 Colname : identifier (def .Column_name ().GetText ()),
122123 IsNotNull : hasNotNullConstraint (def .AllColumn_constraint ()),
123- TypeName : & ast.TypeName {Name : typeName },
124+ TypeName : & ast.TypeName {Name : getTypeName ( def . Type_name ()) },
124125 })
125126 }
126127 }
0 commit comments