forked from pganalyze/pg_query_go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
column_def.go
186 lines (163 loc) · 5.17 KB
/
column_def.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
// Auto-generated from postgres/src/include/nodes/parsenodes.h - DO NOT EDIT
package pg_query
import "encoding/json"
/*
* ColumnDef - column definition (used in various creates)
*
* If the column has a default value, we may have the value expression
* in either "raw" form (an untransformed parse tree) or "cooked" form
* (a post-parse-analysis, executable expression tree), depending on
* how this ColumnDef node was created (by parsing, or by inheritance
* from an existing relation). We should never have both in the same node!
*
* Similarly, we may have a COLLATE specification in either raw form
* (represented as a CollateClause with arg==NULL) or cooked form
* (the collation's OID).
*
* The constraints list may contain a CONSTR_DEFAULT item in a raw
* parsetree produced by gram.y, but transformCreateStmt will remove
* the item and set raw_default instead. CONSTR_DEFAULT items
* should not appear in any subsequent processing.
*/
type ColumnDef struct {
Colname *string `json:"colname"` /* name of column */
TypeName *TypeName `json:"typeName"` /* type of column */
Inhcount int `json:"inhcount"` /* number of times column is inherited */
IsLocal bool `json:"is_local"` /* column has local (non-inherited) def'n */
IsNotNull bool `json:"is_not_null"` /* NOT NULL constraint specified? */
IsFromType bool `json:"is_from_type"` /* column definition came from table type */
IsFromParent bool `json:"is_from_parent"` /* column def came from partition parent */
Storage byte `json:"storage"` /* attstorage setting, or 0 for default */
RawDefault Node `json:"raw_default"` /* default value (untransformed parse tree) */
CookedDefault Node `json:"cooked_default"` /* default value (transformed expr tree) */
Identity byte `json:"identity"` /* attidentity setting */
CollClause *CollateClause `json:"collClause"` /* untransformed COLLATE spec, if any */
CollOid Oid `json:"collOid"` /* collation OID (InvalidOid if not set) */
Constraints List `json:"constraints"` /* other constraints on column */
Fdwoptions List `json:"fdwoptions"` /* per-column FDW options */
Location int `json:"location"` /* parse location, or -1 if none/unknown */
}
func (node ColumnDef) MarshalJSON() ([]byte, error) {
type ColumnDefMarshalAlias ColumnDef
return json.Marshal(map[string]interface{}{
"ColumnDef": (*ColumnDefMarshalAlias)(&node),
})
}
func (node *ColumnDef) UnmarshalJSON(input []byte) (err error) {
var fields map[string]json.RawMessage
err = json.Unmarshal(input, &fields)
if err != nil {
return
}
if fields["colname"] != nil {
err = json.Unmarshal(fields["colname"], &node.Colname)
if err != nil {
return
}
}
if fields["typeName"] != nil {
var nodePtr *Node
nodePtr, err = UnmarshalNodePtrJSON(fields["typeName"])
if err != nil {
return
}
if nodePtr != nil && *nodePtr != nil {
val := (*nodePtr).(TypeName)
node.TypeName = &val
}
}
if fields["inhcount"] != nil {
err = json.Unmarshal(fields["inhcount"], &node.Inhcount)
if err != nil {
return
}
}
if fields["is_local"] != nil {
err = json.Unmarshal(fields["is_local"], &node.IsLocal)
if err != nil {
return
}
}
if fields["is_not_null"] != nil {
err = json.Unmarshal(fields["is_not_null"], &node.IsNotNull)
if err != nil {
return
}
}
if fields["is_from_type"] != nil {
err = json.Unmarshal(fields["is_from_type"], &node.IsFromType)
if err != nil {
return
}
}
if fields["is_from_parent"] != nil {
err = json.Unmarshal(fields["is_from_parent"], &node.IsFromParent)
if err != nil {
return
}
}
if fields["storage"] != nil {
var strVal string
err = json.Unmarshal(fields["storage"], &strVal)
node.Storage = strVal[0]
if err != nil {
return
}
}
if fields["raw_default"] != nil {
node.RawDefault, err = UnmarshalNodeJSON(fields["raw_default"])
if err != nil {
return
}
}
if fields["cooked_default"] != nil {
node.CookedDefault, err = UnmarshalNodeJSON(fields["cooked_default"])
if err != nil {
return
}
}
if fields["identity"] != nil {
var strVal string
err = json.Unmarshal(fields["identity"], &strVal)
node.Identity = strVal[0]
if err != nil {
return
}
}
if fields["collClause"] != nil {
var nodePtr *Node
nodePtr, err = UnmarshalNodePtrJSON(fields["collClause"])
if err != nil {
return
}
if nodePtr != nil && *nodePtr != nil {
val := (*nodePtr).(CollateClause)
node.CollClause = &val
}
}
if fields["collOid"] != nil {
err = json.Unmarshal(fields["collOid"], &node.CollOid)
if err != nil {
return
}
}
if fields["constraints"] != nil {
node.Constraints.Items, err = UnmarshalNodeArrayJSON(fields["constraints"])
if err != nil {
return
}
}
if fields["fdwoptions"] != nil {
node.Fdwoptions.Items, err = UnmarshalNodeArrayJSON(fields["fdwoptions"])
if err != nil {
return
}
}
if fields["location"] != nil {
err = json.Unmarshal(fields["location"], &node.Location)
if err != nil {
return
}
}
return
}