interceptors: add ban

health: set services serving on start and not available on close

Signed-off-by: Adphi <philippe.adrien.nousse@gmail.com>

interceptors/interceptors.go

@@ -1,6 +1,8 @@
 package interceptors
 
 import (
+	"context"
+
 	"google.golang.org/grpc"
 )
 
@@ -18,3 +20,16 @@ type Interceptors interface {
 	ServerInterceptors
 	ClientInterceptors
 }
+
+func NewContextServerStream(ctx context.Context, ss grpc.ServerStream) grpc.ServerStream {
+	return &ContextWrapper{ServerStream: ss, ctx: ctx}
+}
+
+type ContextWrapper struct {
+	grpc.ServerStream
+	ctx context.Context
+}
+
+func (w *ContextWrapper) Context() context.Context {
+	return w.ctx
+}