WIP

Author: asmyasnikov

internal/engine/postgresql/convert.go

@@ -259,8 +259,8 @@ func convertAlterDatabaseSetStmt(n *pg.AlterDatabaseSetStmt) *ast.Node {
 	return &ast.Node{
 		Node: &ast.Node_AlterDatabaseSetStmt{
 			AlterDatabaseSetStmt: &ast.AlterDatabaseSetStmt{
-				Dbname:   n.Dbname,
-				Setstmt:  convertVariableSetStmtForDatabaseSet(n.Setstmt),
+				Dbname:  n.Dbname,
+				Setstmt: convertVariableSetStmtForDatabaseSet(n.Setstmt),
 			},
 		},
 	}
@@ -1482,12 +1482,12 @@ func convertCreateOpClassItem(n *pg.CreateOpClassItem) *ast.Node {
 	return &ast.Node{
 		Node: &ast.Node_CreateOpClassItem{
 			CreateOpClassItem: &ast.CreateOpClassItem{
-				Itemtype:     int32(n.Itemtype),
-				Name:         convertObjectWithArgs(n.Name),
-				Number:       int32(n.Number),
-				OrderFamily:  convertSlice(n.OrderFamily),
-				ClassArgs:    convertSlice(n.ClassArgs),
-				Storedtype:   convertTypeName(n.Storedtype),
+				Itemtype:    int32(n.Itemtype),
+				Name:        convertObjectWithArgs(n.Name),
+				Number:      int32(n.Number),
+				OrderFamily: convertSlice(n.OrderFamily),
+				ClassArgs:   convertSlice(n.ClassArgs),
+				Storedtype:  convertTypeName(n.Storedtype),
 			},
 		},
 	}
@@ -1535,12 +1535,12 @@ func convertCreatePLangStmt(n *pg.CreatePLangStmt) *ast.Node {
 	return &ast.Node{
 		Node: &ast.Node_CreatePLangStmt{
 			CreatePLangStmt: &ast.CreatePLangStmt{
-				Replace:    n.Replace,
-				Plname:     plname,
-				Plhandler:  convertSlice(n.Plhandler),
-				Plinline:   convertSlice(n.Plinline),
+				Replace:     n.Replace,
+				Plname:      plname,
+				Plhandler:   convertSlice(n.Plhandler),
+				Plinline:    convertSlice(n.Plinline),
 				Plvalidator: convertSlice(n.Plvalidator),
-				Pltrusted:  n.Pltrusted,
+				Pltrusted:   n.Pltrusted,
 			},
 		},
 	}
@@ -2135,10 +2135,10 @@ func convertFuncCall(n *pg.FuncCall) *ast.FuncCall {
 		panic(err)
 	}
 	fc := &ast.FuncCall{
-		Func:           rel.FuncName(),
-		Funcname:       convertSlice(n.Funcname),
-		AggStar:        n.AggStar,
-		Location:       int32(n.Location),
+		Func:     rel.FuncName(),
+		Funcname: convertSlice(n.Funcname),
+		AggStar:  n.AggStar,
+		Location: int32(n.Location),
 	}
 	if args := convertSlice(n.Args); args != nil {
 		fc.Args = args
@@ -2420,15 +2420,15 @@ func convertIntoClause(n *pg.IntoClause) *ast.IntoClause {
 		relRelname = n.Rel.Relname
 	}
 	return &ast.IntoClause{
-		RelCatalogname:  relCatalogname,
-		RelSchemaname:   relSchemaname,
-		RelRelname:      relRelname,
-		ColNames:        convertSlice(n.ColNames),
-		Options:         convertSlice(n.Options),
-		OnCommit:        ast.OnCommitAction(n.OnCommit),
-		TableSpaceName:  tableSpaceName,
-		ViewQuery:       convertNode(n.ViewQuery),
-		SkipData:        n.SkipData,
+		RelCatalogname: relCatalogname,
+		RelSchemaname:  relSchemaname,
+		RelRelname:     relRelname,
+		ColNames:       convertSlice(n.ColNames),
+		Options:        convertSlice(n.Options),
+		OnCommit:       ast.OnCommitAction(n.OnCommit),
+		TableSpaceName: tableSpaceName,
+		ViewQuery:      convertNode(n.ViewQuery),
+		SkipData:       n.SkipData,
 	}
 }
 
@@ -2588,7 +2588,7 @@ func convertNullIfExpr(n *pg.NullIfExpr) *ast.Node {
 				Xpr:          convertNode(n.Xpr),
 				Opno:         &ast.Oid{Value: uint64(n.Opno)},
 				Opfuncid:     &ast.Oid{Value: 0}, // Opfuncid not in pg_query
-				Opresulttype:  &ast.Oid{Value: uint64(n.Opresulttype)},
+				Opresulttype: &ast.Oid{Value: uint64(n.Opresulttype)},
 				Opretset:     &ast.Oid{Value: 0}, // Opretset is bool in pg_query, convert to Oid
 				Opcollid:     &ast.Oid{Value: uint64(n.Opcollid)},
 				Inputcollid:  &ast.Oid{Value: uint64(n.Inputcollid)},
@@ -2856,7 +2856,7 @@ func convertQuery(n *pg.Query) *ast.Query {
 		LimitCount:       convertNode(n.LimitCount),
 		RowMarks:         convertSlice(n.RowMarks),
 		SetOperations:    convertNode(n.SetOperations),
-		ConstraintDeps:  convertSlice(n.ConstraintDeps),
+		ConstraintDeps:   convertSlice(n.ConstraintDeps),
 		WithCheckOptions: convertSlice(n.WithCheckOptions),
 		StmtLocation:     int32(n.StmtLocation),
 		StmtLen:          int32(n.StmtLen),
@@ -2980,11 +2980,11 @@ func convertRangeTblEntry(n *pg.RangeTblEntry) *ast.Node {
 				Lateral:         n.Lateral,
 				Inh:             n.Inh,
 				InFromCl:        n.InFromCl,
-				RequiredPerms:   0, // RequiredPerms not in pg_query
+				RequiredPerms:   0,                  // RequiredPerms not in pg_query
 				CheckAsUser:     &ast.Oid{Value: 0}, // CheckAsUser not in pg_query
-				SelectedCols:    []uint32{}, // SelectedCols not in pg_query
-				InsertedCols:    []uint32{}, // InsertedCols not in pg_query
-				UpdatedCols:     []uint32{}, // UpdatedCols not in pg_query
+				SelectedCols:    []uint32{},         // SelectedCols not in pg_query
+				InsertedCols:    []uint32{},         // InsertedCols not in pg_query
+				UpdatedCols:     []uint32{},         // UpdatedCols not in pg_query
 				SecurityQuals:   convertSlice(n.SecurityQuals),
 			},
 		},
@@ -3310,8 +3310,8 @@ func convertSecLabelStmt(n *pg.SecLabelStmt) *ast.Node {
 	return &ast.Node{
 		Node: &ast.Node_SecLabelStmt{
 			SecLabelStmt: &ast.SecLabelStmt{
-				Objtype: ast.ObjectType(n.Objtype),
-				Object:  convertNode(n.Object),
+				Objtype:  ast.ObjectType(n.Objtype),
+				Object:   convertNode(n.Object),
 				Provider: n.Provider,
 				Label:    n.Label,
 			},
@@ -3324,8 +3324,8 @@ func convertSelectStmt(n *pg.SelectStmt) *ast.SelectStmt {
 		return nil
 	}
 	stmt := &ast.SelectStmt{
-		TargetList:  convertSlice(n.TargetList),
-		FromClause:  convertSlice(n.FromClause),
+		TargetList: convertSlice(n.TargetList),
+		FromClause: convertSlice(n.FromClause),
 	}
 	// Always set these fields, even if empty (for consistency with test expectations)
 	stmt.GroupClause = convertSlice(n.GroupClause)