blockstream: add Buffer option for NewUnmanaged constructor

goldsky-io · Nov 15, 2022 · 752234e · 752234e
1 parent 5c4ec70
commit 752234e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/blockstream/server.go b/blockstream/server.go
@@ -41,6 +41,13 @@ func ServerOptionWithLogger(logger *zap.Logger) ServerOption {
 	}
 }
 
+func ServerOptionWithBuffer(size int) ServerOption {
+	return func(s *Server) {
+		s.buffer = bstream.NewBuffer("blockserver", s.logger.Named("buffer"))
+		s.bufferSize = size
+	}
+}
+
 type Server struct {
 	headInfo      *headInfo
 	buffer        *bstream.Buffer