👌 chore: added execute batch with transaction #1

Author: pnguyen215

postgresconn/postgresconn_service.go

@@ -24,6 +24,8 @@ type PostgresService interface {
 	ShowProcedureContent(procedure string) (string, error)
 	ExplainAnalysis(query string) (string, error)
 	ExplainAnalysisFromFile(filename string) (string, error)
+	ExecuteBatch(statements []string) error
+	ExecuteBatchWithTransaction(statements []string) error
 }
 
 type postgresServiceImpl struct {
@@ -186,3 +188,52 @@ func (p *postgresServiceImpl) ExplainAnalysisFromFile(filename string) (string,
 	query := string(bytes)
 	return p.ExplainAnalysis(query)
 }
+
+func (p *postgresServiceImpl) ExecuteBatch(statements []string) error {
+	if len(statements) == 0 {
+		return fmt.Errorf("missing statements")
+	}
+	tx, err := p.dbConn.Beginx()
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if p := recover(); p != nil {
+			tx.Rollback()
+			panic(p)
+		} else if err != nil {
+			tx.Rollback()
+		} else {
+			err = tx.Commit()
+		}
+	}()
+	for _, statement := range statements {
+		_, err := tx.Exec(statement)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func (p *postgresServiceImpl) ExecuteBatchWithTransaction(statements []string) error {
+	tx, err := p.dbConn.Beginx()
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if p := recover(); p != nil {
+			tx.Rollback()
+			panic(p)
+		} else if err != nil {
+			tx.Rollback()
+		} else {
+			err = tx.Commit()
+		}
+	}()
+	err = p.ExecuteBatch(statements)
+	if err != nil {
+		return err
+	}
+	return nil
+}