@@ -35,6 +35,13 @@ func identifier(id string) string {
35
35
return strings .ToLower (id )
36
36
}
37
37
38
+ func getTypeName (t parser.IType_nameContext ) string {
39
+ if t == nil {
40
+ return "any"
41
+ }
42
+ return t .GetText ()
43
+ }
44
+
38
45
func NewIdentifier (t string ) * ast.String {
39
46
return & ast.String {Str : identifier (t )}
40
47
}
@@ -72,10 +79,8 @@ func (c *cc) convertAlter_table_stmtContext(n *parser.Alter_table_stmtContext) a
72
79
Name : & name ,
73
80
Subtype : ast .AT_AddColumn ,
74
81
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 ())},
79
84
IsNotNull : hasNotNullConstraint (def .AllColumn_constraint ()),
80
85
},
81
86
})
@@ -113,14 +118,10 @@ func (c *cc) convertCreate_table_stmtContext(n *parser.Create_table_stmtContext)
113
118
}
114
119
for _ , idef := range n .AllColumn_def () {
115
120
if def , ok := idef .(* parser.Column_defContext ); ok {
116
- typeName := "any"
117
- if def .Type_name () != nil {
118
- typeName = def .Type_name ().GetText ()
119
- }
120
121
stmt .Cols = append (stmt .Cols , & ast.ColumnDef {
121
122
Colname : identifier (def .Column_name ().GetText ()),
122
123
IsNotNull : hasNotNullConstraint (def .AllColumn_constraint ()),
123
- TypeName : & ast.TypeName {Name : typeName },
124
+ TypeName : & ast.TypeName {Name : getTypeName ( def . Type_name ()) },
124
125
})
125
126
}
126
127
}
0 commit comments