bug fix

batch3.go

@@ -133,19 +133,21 @@ func (c *Controller[Res]) commit(ctx context.Context, err error) (Res, error) {
 again:
 	if err != nil || atomic.LoadInt32(&c.queue) == 0 {
 		c.cnt = -c.cnt
-		c.ready = true
 
 		if ep, ok := err.(PanicError); ok {
+			c.ready = true
 			c.err = err
 			panic(ep.Panic)
 		} else if err != nil {
+			c.ready = true
 			c.err = err
 		} else {
 			func() {
 				var res Res
 				var err error
 
 				defer func() {
+					c.ready = true
 					c.res, c.err = res, err
 
 					if p := recover(); p != nil {
@@ -260,8 +262,15 @@ func (b *Batch[Res]) Rollback(ctx context.Context, err error) (Res, error) {
 	return b.c.commit(ctx, err)
 }
 
+func AsPanicError(err error) (PanicError, bool) {
+	var pe PanicError
+
+	return pe, errors.As(err, &pe)
+}
+
 func (e PanicError) Error() string {
 	return fmt.Sprintf("panic: %v", e.Panic)
 }
 
-func (noCopy) Lock() {}
+func (*noCopy) Lock()   {}
+func (*noCopy) Unlock() {}

go.mod

@@ -1,3 +1,5 @@
 module nikand.dev/go/batch
 
 go 1.18
+
+retract v0.4.0 // has a bug

multi.go

@@ -40,7 +40,10 @@ func (c *Multi[Res]) Enter(blocking bool) (b Batch[Res], coach, index int) {
 		for coach := range c.cs {
 			b, idx := c.cs[coach].Enter(false)
 			if idx >= 0 {
-				return b, coach, idx
+				return Batch[Res]{
+					c:     b.c,
+					state: b.state,
+				}, coach, idx
 			}
 		}