Method parseCreateStmt

database/postgres/parser.go:193–299

(stmt *pgquery.CreateStmt)

}

func (p PostgresParser) parseCreateStmt(stmt *pgquery.CreateStmt) (parser.Statement, error) {
	if stmt.Constraints != nil {
		return nil, fmt.Errorf("unhandled node in parseCreateStmt: %#v", stmt)
	}

	tableName, err := p.parseTableName(stmt.Relation)
	if err != nil {
		return nil, err
	}

	var columns []*parser.ColumnDefinition
	var indexes []*parser.IndexDefinition
	var foreignKeys []*parser.ForeignKeyDefinition
	var checks []*parser.CheckDefinition
	var exclusions []*parser.ExclusionDefinition
	for _, elt := range stmt.TableElts {
		switch node := elt.Node.(type) {
		case *pgquery.Node_ColumnDef:
			column, foreignKey, err := p.parseColumnDef(node.ColumnDef, tableName)
			if err != nil {
				return nil, err
			}
			columns = append(columns, column)
			if foreignKey != nil {
				foreignKeys = append(foreignKeys, foreignKey)
			}
		case *pgquery.Node_Constraint:
			var indexCols []parser.IndexColumn
			for _, key := range node.Constraint.Keys {
				indexCol := parser.IndexColumn{
					Column:    parser.NewIdent(key.Node.(*pgquery.Node_String_).String_.Sval, false),
					Direction: "asc",
				}
				indexCols = append(indexCols, indexCol)
			}
			switch node.Constraint.Contype {
			case pgquery.ConstrType_CONSTR_PRIMARY:
				index := &parser.IndexDefinition{
					Info: &parser.IndexInfo{
						Type:      "primary key",
						Name:      parser.NewIdent(node.Constraint.Conname, false),
						Unique:    true,
						Primary:   true,
						Clustered: true,
					},
					Columns: indexCols,
					Options: []*parser.IndexOption{},
				}
				indexes = append(indexes, index)
			case pgquery.ConstrType_CONSTR_UNIQUE:
				index := &parser.IndexDefinition{
					Info: &parser.IndexInfo{
						Type:   "UNIQUE",
						Name:   parser.NewIdent(node.Constraint.Conname, false),
						Unique: true,
					},
					Columns: indexCols,
					Options: []*parser.IndexOption{},