feat(do): implement WithResult method

coderyw · Jan 28, 2022 · c62696d · c62696d
1 parent b4c83a4
commit c62696d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/do.go b/do.go
@@ -618,6 +618,12 @@ func (d *DO) ScanRows(rows *sql.Rows, dest interface{}) error {
 	return d.db.Model(d.newResultPointer()).ScanRows(rows, dest)
 }
 
+func (d DO) WithResult(fc func(tx Dao)) ResultInfo {
+	d.db = d.db.Set("", "")
+	fc(&d)
+	return ResultInfo{RowsAffected: d.db.RowsAffected, Error: d.db.Error}
+}
+
 func (d *DO) newResultPointer() interface{} {
 	if d.modelType == nil {
 		return nil