add context

Author: ad3n

bima.go

@@ -11,7 +11,7 @@ import (
 )
 
 const (
-	Version = "v4.1.6"
+	Version = "v4.1.7"
 
 	HighestPriority = 255
 	LowestPriority  = -255

go.mod

@@ -26,7 +26,7 @@ require (
 	github.com/vcraescu/go-paginator/v2 v2.0.0
 	github.com/youmark/pkcs8 v0.0.0-20201027041543-1326539a0a0a // indirect
 	go.mongodb.org/mongo-driver v1.10.0
-	golang.org/x/net v0.0.0-20220708220712-1185a9018129 // indirect
+	golang.org/x/net v0.0.0-20220708220712-1185a9018129
 	golang.org/x/sync v0.0.0-20220601150217-0de741cfad7f // indirect
 	golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 // indirect
 	golang.org/x/tools v0.1.5 // indirect

interfaces/consumer.go

@@ -3,13 +3,14 @@ package interfaces
 import (
 	"github.com/bimalabs/framework/v4/configs"
 	"github.com/bimalabs/framework/v4/messengers"
+	"golang.org/x/net/context"
 )
 
 type Consumer struct {
 	Messenger *messengers.Messenger
 }
 
-func (q *Consumer) Run(servers []configs.Server) {
+func (q *Consumer) Run(ctx context.Context, servers []configs.Server) {
 	if q.Messenger == nil {
 		return
 	}

interfaces/database.go

@@ -2,12 +2,13 @@ package interfaces
 
 import (
 	"github.com/bimalabs/framework/v4/configs"
+	"golang.org/x/net/context"
 )
 
 type Database struct {
 }
 
-func (d *Database) Run(servers []configs.Server) {
+func (d *Database) Run(ctx context.Context, servers []configs.Server) {
 	if configs.Database == nil {
 		return
 	}

interfaces/elasticsearch.go

@@ -3,13 +3,14 @@ package interfaces
 import (
 	"github.com/bimalabs/framework/v4/configs"
 	"github.com/olivere/elastic/v7"
+	"golang.org/x/net/context"
 )
 
 type Elasticsearch struct {
 	Client *elastic.Client
 }
 
-func (e *Elasticsearch) Run(servers []configs.Server) {
+func (e *Elasticsearch) Run(ctx context.Context, servers []configs.Server) {
 	if e.Client == nil {
 		return
 	}

interfaces/factory.go

@@ -5,11 +5,12 @@ import (
 	"time"
 
 	"github.com/bimalabs/framework/v4/configs"
+	"golang.org/x/net/context"
 )
 
 type (
 	Application interface {
-		Run(servers []configs.Server)
+		Run(ctx context.Context, servers []configs.Server)
 		IsBackground() bool
 		Priority() int
 	}
@@ -34,12 +35,15 @@ func (f *Factory) Run(servers []configs.Server) {
 		return f.applications[i].Priority() > f.applications[j].Priority()
 	})
 
+	ctx, cancel := context.WithCancel(context.Background())
 	for _, application := range f.applications {
 		if application.IsBackground() {
-			go application.Run(servers)
+			go application.Run(ctx, servers)
 		} else {
 			time.Sleep(100 * time.Millisecond)
-			application.Run(servers)
+			application.Run(ctx, servers)
 		}
 	}
+
+	defer cancel()
 }

interfaces/grpc.go

@@ -1,6 +1,7 @@
 package interfaces
 
 import (
+	"context"
 	"log"
 	"net"
 	"strconv"
@@ -20,7 +21,7 @@ type GRpc struct {
 	Debug    bool
 }
 
-func (g *GRpc) Run(servers []configs.Server) {
+func (g *GRpc) Run(ctx context.Context, servers []configs.Server) {
 	var gRpcAddress strings.Builder
 	gRpcAddress.WriteString(":")
 	gRpcAddress.WriteString(strconv.Itoa(g.GRpcPort))
@@ -47,6 +48,13 @@ func (g *GRpc) Run(servers []configs.Server) {
 		grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(streams...)),
 		grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(unaries...)),
 	)
+	go func() {
+		select {
+		case <-ctx.Done():
+			gRpc.Stop()
+		}
+
+	}()
 
 	for _, server := range servers {
 		server.Register(gRpc)