more otel

pkg/actor/disk_storage.go

@@ -1,6 +1,7 @@
 package actor
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"log"
@@ -25,7 +26,7 @@ type DiskStorage[V any] struct {
 }
 
 type LogStorage[V any] interface {
-	LoadEvents(id uint64, grain Grain[V]) error
+	LoadEvents(ctx context.Context, id uint64, grain Grain[V]) error
 	AppendMutations(id uint64, msg ...proto.Message) error
 }
 
@@ -86,7 +87,7 @@ func (s *DiskStorage[V]) logPath(id uint64) string {
 	return filepath.Join(s.path, fmt.Sprintf("%d.events.log", id))
 }
 
-func (s *DiskStorage[V]) LoadEvents(id uint64, grain Grain[V]) error {
+func (s *DiskStorage[V]) LoadEvents(ctx context.Context, id uint64, grain Grain[V]) error {
 	path := s.logPath(id)
 	if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) {
 		// No log -> nothing to replay
@@ -99,7 +100,7 @@ func (s *DiskStorage[V]) LoadEvents(id uint64, grain Grain[V]) error {
 	}
 	defer fh.Close()
 	return s.Load(fh, func(msg proto.Message) {
-		s.registry.Apply(grain, msg)
+		s.registry.Apply(ctx, grain, msg)
 	})
 }