u

code/allapp/allapp-go/db/repo/batch.go

@@ -16,6 +16,59 @@ var (
 	ErrBatchAlreadyClosed = errors.New("batch already closed")
 )
 
+const batchUpdateFinanceCategorySortNumber = `-- name: BatchUpdateFinanceCategorySortNumber :batchexec
+UPDATE finance_category_space
+SET sort_number = $1,update_by = $2,update_time = NOW()
+WHERE id = $3
+`
+
+type BatchUpdateFinanceCategorySortNumberBatchResults struct {
+	br     pgx.BatchResults
+	tot    int
+	closed bool
+}
+
+type BatchUpdateFinanceCategorySortNumberParams struct {
+	SortNumber int32
+	UpdateBy   int64
+	ID         int64
+}
+
+func (q *Queries) BatchUpdateFinanceCategorySortNumber(ctx context.Context, arg []BatchUpdateFinanceCategorySortNumberParams) *BatchUpdateFinanceCategorySortNumberBatchResults {
+	batch := &pgx.Batch{}
+	for _, a := range arg {
+		vals := []interface{}{
+			a.SortNumber,
+			a.UpdateBy,
+			a.ID,
+		}
+		batch.Queue(batchUpdateFinanceCategorySortNumber, vals...)
+	}
+	br := q.db.SendBatch(ctx, batch)
+	return &BatchUpdateFinanceCategorySortNumberBatchResults{br, len(arg), false}
+}
+
+func (b *BatchUpdateFinanceCategorySortNumberBatchResults) Exec(f func(int, error)) {
+	defer b.br.Close()
+	for t := 0; t < b.tot; t++ {
+		if b.closed {
+			if f != nil {
+				f(t, ErrBatchAlreadyClosed)
+			}
+			continue
+		}
+		_, err := b.br.Exec()
+		if f != nil {
+			f(t, err)
+		}
+	}
+}
+
+func (b *BatchUpdateFinanceCategorySortNumberBatchResults) Close() error {
+	b.closed = true
+	return b.br.Close()
+}
+
 const batchUpdateSpaceSortNumber = `-- name: BatchUpdateSpaceSortNumber :batchexec
 UPDATE space_member
 SET sort_number = $1