Skip to content

Commit 363b32f

Browse files
committed
feat: add grpc client method for updating parameters
1 parent fb3f486 commit 363b32f

File tree

5 files changed

+273
-193
lines changed

5 files changed

+273
-193
lines changed

infrastructure/grpc_client/client.go

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@ package grpcclient
22

33
import (
44
"context"
5-
65
mpb "github.com/dezh-tech/immortal/infrastructure/grpc_client/gen"
6+
"github.com/dezh-tech/immortal/infrastructure/grpc_client/params_keeper"
77
"google.golang.org/grpc"
88
"google.golang.org/grpc/credentials/insecure"
99
"google.golang.org/grpc/metadata"
@@ -16,9 +16,10 @@ type Client struct {
1616
id string
1717
config Config
1818
conn *grpc.ClientConn
19+
parametersKeeper params_keeper.ParametersKeeper
1920
}
2021

21-
func New(endpoint string, cfg Config) (IClient, error) {
22+
func New(endpoint string, cfg Config, keeper params_keeper.ParametersKeeper) (IClient, error) {
2223
conn, err := grpc.NewClient(endpoint, grpc.WithTransportCredentials(insecure.NewCredentials()))
2324
if err != nil {
2425
return nil, err
@@ -30,6 +31,7 @@ func New(endpoint string, cfg Config) (IClient, error) {
3031
LogService: mpb.NewLogClient(conn),
3132
config: cfg,
3233
conn: conn,
34+
parametersKeeper: keeper,
3335
}, nil
3436
}
3537

@@ -61,3 +63,12 @@ func (c *Client) AddLog(ctx context.Context, msg, stack string) (*mpb.AddLogResp
6163
Stack: stack,
6264
})
6365
}
66+
67+
func (c *Client) UpdateParameters(ctx context.Context, newParams *mpb.GetParametersResponse) error {
68+
select {
69+
case <-ctx.Done():
70+
return ctx.Err()
71+
default:
72+
return c.parametersKeeper.LoadParameters(newParams)
73+
}
74+
}

0 commit comments

Comments
 (0)