feat(grpc): add global config. (#125)

Author: SwimmingRieux

cmd/relay/relay.go

@@ -19,7 +19,7 @@ import (
 
 // Relay keeps all concepts such as server, database and manages them.
 type Relay struct {
-	config          config.Config
+	config          *config.Config
 	websocketServer *websocket.Server
 	grpcServer      *grpc.Server
 	database        *database.Database
@@ -68,17 +68,21 @@ func New(cfg *config.Config) (*Relay, error) {
 		return nil, err
 	}
 
-	h := repository.New(cfg.Handler, db, meili, c)
+	h := repository.New(&cfg.Handler, db, meili, c)
 
-	ws, err := websocket.New(cfg.WebsocketServer, h, m, r, c)
+	ws, err := websocket.New(&cfg.WebsocketServer, h, m, r, c)
 	if err != nil {
 		return nil, err
 	}
 
-	gs := grpc.New(&cfg.GRPCServer, r, db, time.Now())
+	keeper := grpc.ParametersKeeper{
+		Handler:         &cfg.Handler,
+		WebsocketServer: &cfg.WebsocketServer,
+	}
+	gs := grpc.New(&cfg.GRPCServer, r, db, time.Now(), keeper)
 
 	return &Relay{
-		config:          *cfg,
+		config:          cfg,
 		websocketServer: ws,
 		database:        db,
 		redis:           r,

config/config.go

@@ -3,14 +3,16 @@ package config
 import (
 	"os"
 
-	"github.com/dezh-tech/immortal/delivery/grpc"
 	"github.com/dezh-tech/immortal/delivery/websocket"
+
+	"github.com/dezh-tech/immortal/repository"
+
+	"github.com/dezh-tech/immortal/delivery/grpc"
 	"github.com/dezh-tech/immortal/infrastructure/database"
 	grpcclient "github.com/dezh-tech/immortal/infrastructure/grpc_client"
 	"github.com/dezh-tech/immortal/infrastructure/meilisearch"
 	"github.com/dezh-tech/immortal/infrastructure/redis"
 	"github.com/dezh-tech/immortal/pkg/logger"
-	"github.com/dezh-tech/immortal/repository"
 	"github.com/joho/godotenv"
 	"gopkg.in/yaml.v3"
 )

config/parameters.go

@@ -4,40 +4,33 @@ import (
 	"github.com/dezh-tech/immortal/delivery/websocket"
 	mpb "github.com/dezh-tech/immortal/infrastructure/grpc_client/gen"
 	"github.com/dezh-tech/immortal/pkg/utils"
-	"github.com/dezh-tech/immortal/repository"
 )
 
-type Parameters struct {
-	Handler         *repository.Config
-	WebsocketServer *websocket.Config
-}
-
 func (c *Config) LoadParameters(params *mpb.GetParametersResponse) error {
 	url, err := utils.ParseURL(params.Url)
 	if err != nil {
 		return err
 	}
 
-	c.WebsocketServer.URL = url
+	c.WebsocketServer.SetURL(url)
 
-	c.WebsocketServer.Limitation = &websocket.Limitation{
-		MaxMessageLength:    params.Limitations.MaxMessageLength,
-		MaxSubscriptions:    params.Limitations.MaxSubscriptions,
-		MaxSubidLength:      params.Limitations.MaxSubidLength,
-		MinPowDifficulty:    params.Limitations.MinPowDifficulty,
-		AuthRequired:        params.Limitations.AuthRequired,
-		PaymentRequired:     params.Limitations.PaymentRequired,
-		RestrictedWrites:    params.Limitations.RestrictedWrites,
-		MaxEventTags:        params.Limitations.MaxEventTags,
-		MaxContentLength:    params.Limitations.MaxContentLength,
-		CreatedAtLowerLimit: params.Limitations.CreatedAtLowerLimit,
-		CreatedAtUpperLimit: params.Limitations.CreatedAtUpperLimit,
-	}
+	c.WebsocketServer.SetLimitation(
+		&websocket.Limitation{
+			MaxMessageLength:    params.Limitations.MaxMessageLength,
+			MaxSubscriptions:    params.Limitations.MaxSubscriptions,
+			MaxSubidLength:      params.Limitations.MaxSubidLength,
+			MinPowDifficulty:    params.Limitations.MinPowDifficulty,
+			AuthRequired:        params.Limitations.AuthRequired,
+			PaymentRequired:     params.Limitations.PaymentRequired,
+			RestrictedWrites:    params.Limitations.RestrictedWrites,
+			MaxEventTags:        params.Limitations.MaxEventTags,
+			MaxContentLength:    params.Limitations.MaxContentLength,
+			CreatedAtLowerLimit: params.Limitations.CreatedAtLowerLimit,
+			CreatedAtUpperLimit: params.Limitations.CreatedAtUpperLimit,
+		})
 
-	c.Handler = repository.Config{
-		DefaultQueryLimit: params.Limitations.DefaultQueryLimit,
-		MaxQueryLimit:     params.Limitations.MaxQueryLimit,
-	}
+	c.Handler.SetMaxQueryLimit(params.Limitations.MaxQueryLimit)
+	c.Handler.SetDefaultQueryLimit(params.Limitations.DefaultQueryLimit)
 
 	return nil
 }