lorm.DoListTx / lorm.DoSetListTx shortcuts

CHANGELOG.txt

@@ -1,3 +1,7 @@
+v0.10.0
+ * lorm.DoListTx / lorm.DoSetListTx shortcuts
+ * lorm.List interface
+
 v0.9.5
  * pq v1.10.2
 

generic.go

@@ -114,6 +114,26 @@ func DoInTx(r Record, fn func() error) error {
 	return fn()
 }
 
+func DoListTx(l List) *Tx {
+	byTx := make(map[*Tx]struct{}, 1)
+	for _, r := range l.Records() {
+		byTx[r.Tx()] = struct{}{}
+	}
+	if len(byTx) > 1 {
+		log.Panicln("invalid transaction number:", len(byTx))
+	}
+	for tx := range byTx {
+		return tx
+	}
+	return nil
+}
+
+func DoSetListTx(tx *Tx, l List) {
+	for _, r := range l.Records() {
+		r.SetTx(tx)
+	}
+}
+
 func DoDelete(r Record, t op.Table) error {
 	args := op.NewArgs()
 	query := op.Delete(t).Where(r.PkCond(&args))

list.go

@@ -0,0 +1,5 @@
+package lorm
+
+type List interface {
+	Records() []Record
+}