New DBAction to run raw SQL statements (#898)

Related to #742

Author: kvch

pkg/migrations/dbactions.go

@@ -673,3 +673,20 @@ func (a *dropDefaultAction) Execute(ctx context.Context) error {
 		pq.QuoteIdentifier(a.column)))
 	return err
 }
+
+type rawSQLAction struct {
+	conn db.DB
+	sql  string
+}
+
+func NewRawSQLAction(conn db.DB, sql string) *rawSQLAction {
+	return &rawSQLAction{
+		conn: conn,
+		sql:  sql,
+	}
+}
+
+func (a *rawSQLAction) Execute(ctx context.Context) error {
+	_, err := a.conn.ExecContext(ctx, a.sql)
+	return err
+}

pkg/migrations/op_raw_sql.go

@@ -21,8 +21,7 @@ func (o *OpRawSQL) Start(ctx context.Context, l Logger, conn db.DB, latestSchema
 		return nil, nil
 	}
 
-	_, err := conn.ExecContext(ctx, o.Up)
-	return nil, err
+	return nil, NewRawSQLAction(conn, o.Up).Execute(ctx)
 }
 
 func (o *OpRawSQL) Complete(ctx context.Context, l Logger, conn db.DB, s *schema.Schema) error {
@@ -32,8 +31,7 @@ func (o *OpRawSQL) Complete(ctx context.Context, l Logger, conn db.DB, s *schema
 		return nil
 	}
 
-	_, err := conn.ExecContext(ctx, o.Up)
-	return err
+	return NewRawSQLAction(conn, o.Up).Execute(ctx)
 }
 
 func (o *OpRawSQL) Rollback(ctx context.Context, l Logger, conn db.DB, s *schema.Schema) error {
@@ -43,8 +41,7 @@ func (o *OpRawSQL) Rollback(ctx context.Context, l Logger, conn db.DB, s *schema
 		return nil
 	}
 
-	_, err := conn.ExecContext(ctx, o.Down)
-	return err
+	return NewRawSQLAction(conn, o.Down).Execute(ctx)
 }
 
 func (o *OpRawSQL) Validate(ctx context.Context, s *schema.Schema) error {