*: gofumpt

Author: tolgaozen

integration-test/integration_test.go

@@ -53,7 +53,6 @@ func healthCheck(attempts int) error {
 
 // HTTP POST: /translation/do-translate.
 func TestHTTPCheckRequest(t *testing.T) {
-
 	body := `{
     	"schema": "entity user {} \n\nentity account {\n    // roles \n    relation admin @user    \n    relation member @user    \n    relation parent_account @account\n\n    action add_member = admin or parent_account.add_member\n    action delete_member = admin\n\n}"
 	}`
@@ -90,5 +89,4 @@ func TestHTTPCheckRequest(t *testing.T) {
 		Expect().Status().Equal(http.StatusOK),
 		Expect().Body().JSON().JQ(".can").Equal("RESULT_DENIED"),
 	)
-
 }

internal/config/config.go

@@ -26,9 +26,9 @@ type (
 
 	// Server contains the configurations for both HTTP and gRPC servers.
 	Server struct {
-		Address string `mapstructure:"address"` // Address for the server
-		HTTP    `mapstructure:"http"`           // HTTP server configuration
-		GRPC    `mapstructure:"grpc"`           // gRPC server configuration
+		Address string                `mapstructure:"address"` // Address for the server
+		HTTP    `mapstructure:"http"` // HTTP server configuration
+		GRPC    `mapstructure:"grpc"` // gRPC server configuration
 	}
 
 	// HTTP contains configuration for the HTTP server.
@@ -301,11 +301,9 @@ func DefaultConfig() *Config {
 }
 
 func isYAML(file string) error {
-
 	ext := filepath.Ext(file)
 	if ext != "yaml" {
 		return errors.New("file is not yaml")
 	}
 	return nil
-
 }

internal/engines/consistent/hashring.go

@@ -19,29 +19,30 @@ import (
 
 // Hashring is a wrapper around the consistent hash implementation that
 type Hashring struct {
-	checker          invoke.Check
-	gossip           gossip.IGossip
-	consistent       hash.Consistent
-	localNodeAddress string
-	l                *logger.Logger
+	checker           invoke.Check
+	gossip            gossip.IGossip
+	consistent        hash.Consistent
+	localNodeAddress  string
+	connectionOptions []grpc.DialOption
+	l                 *logger.Logger
 }
 
 // NewCheckEngineWithHashring creates a new instance of EngineKeyManager by initializing an EngineKeys
 // struct with the provided cache.Cache instance.
-func NewCheckEngineWithHashring(checker invoke.Check, consistent *hash.ConsistentHash, g *gossip.Gossip, port string, l *logger.Logger) (invoke.Check, error) {
+func NewCheckEngineWithHashring(checker invoke.Check, consistent *hash.ConsistentHash, g *gossip.Gossip, port string, l *logger.Logger, options ...grpc.DialOption) (invoke.Check, error) {
 	// Return a new instance of EngineKeys with the provided cache
-
 	ip, err := gossip.ExternalIP()
 	if err != nil {
 		return nil, err
 	}
 
 	return &Hashring{
-		checker:          checker,
-		localNodeAddress: ip + ":" + port,
-		gossip:           g,
-		consistent:       consistent,
-		l:                l,
+		checker:           checker,
+		localNodeAddress:  ip + ":" + port,
+		gossip:            g,
+		consistent:        consistent,
+		connectionOptions: options,
+		l:                 l,
 	}, nil
 }
 
@@ -96,7 +97,7 @@ func (c *Hashring) Check(ctx context.Context, request *base.PermissionCheckReque
 // forwardRequestGetToNode forwards a request to the responsible node
 func (c *Hashring) forwardRequestToNode(ctx context.Context, node string, request *base.PermissionCheckRequest) (*base.PermissionCheckResponse, error) {
 	// Set up a connection to the server.
-	conn, err := grpc.Dial(node, grpc.WithInsecure(), grpc.WithBlock())
+	conn, err := grpc.DialContext(ctx, node, c.connectionOptions...)
 	if err != nil {
 		log.Fatalf("did not connect: %v", err)
 	}