Skip to content

Commit cadce21

Browse files
committed
cleaner code; should pass lint tests now
1 parent 170bbb2 commit cadce21

File tree

3 files changed

+28
-19
lines changed

3 files changed

+28
-19
lines changed

pkg/gofr/grpc.go

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ func (g *grpcServer) Run(c *container.Container) {
139139
if g.server == nil {
140140
if err := g.createServer(); err != nil {
141141
c.Logger.Fatalf("failed to create gRPC server: %v", err)
142-
c.Metrics().NewCounter("grpc_server_errors_total", "").Inc()
142+
c.Metrics().IncrementCounter(ctx, "grpc_server_errors_total")
143143
return
144144
}
145145
}
@@ -151,22 +151,22 @@ func (g *grpcServer) Run(c *container.Container) {
151151
listener, err := net.Listen("tcp", addr)
152152
if err != nil {
153153
c.Logger.Errorf("error in starting gRPC server at %s: %s", addr, err)
154-
c.Metrics().NewCounter("grpc_server_errors_total", "").Inc()
155-
c.Metrics().NewGauge("grpc_server_status", "").Set(0)
154+
c.Metrics().IncrementCounter(ctx, "grpc_server_errors_total")
155+
c.Metrics().SetGauge("grpc_server_status", 0)
156156
return
157157
}
158158

159-
c.Metrics().NewGauge("grpc_server_status", "").Set(1)
159+
c.Metrics().SetGauge("grpc_server_status", 1)
160160

161161
if err := g.server.Serve(listener); err != nil {
162162
c.Logger.Errorf("error in starting gRPC server at %s: %s", addr, err)
163-
c.Metrics().NewCounter("grpc_server_errors_total", "").Inc()
164-
c.Metrics().NewGauge("grpc_server_status", "").Set(0)
163+
c.Metrics().IncrementCounter(ctx, "grpc_server_errors_total")
164+
c.Metrics().SetGauge("grpc_server_status", 0)
165165
return
166166
}
167167

168168
c.Logger.Infof("gRPC server stopped on %s", addr)
169-
c.Metrics().NewGauge("grpc_server_status", "").Set(0)
169+
c.Metrics().SetGauge("grpc_server_status", 0)
170170
}
171171

172172
func (g *grpcServer) Shutdown(ctx context.Context) error {
@@ -203,8 +203,8 @@ func (a *App) RegisterService(desc *grpc.ServiceDesc, impl any) {
203203

204204
a.container.Logger.Infof("registering gRPC Server: %s", desc.ServiceName)
205205
a.grpcServer.server.RegisterService(desc, impl)
206-
207-
a.container.Metrics().NewCounter("grpc_services_registered_total", "").Inc()
206+
207+
a.container.Metrics().IncrementCounter(ctx, "grpc_services_registered_total")
208208

209209
err := injectContainer(impl, a.container)
210210
if err != nil {

pkg/gofr/grpc/log_test.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,7 @@ func TestRPCLog_PrettyPrint(t *testing.T) {
9494
// Check if ID is coming
9595
assert.Contains(t, log, `1`)
9696
}
97+
9798
func TestRPCLog_PrettyPrintWithStreamType(t *testing.T) {
9899
var buf bytes.Buffer
99100

pkg/gofr/grpc_test.go

Lines changed: 18 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,8 @@ func TestNewGRPCServer(t *testing.T) {
3939
Logger: logging.NewLogger(logging.DEBUG),
4040
}
4141
cfg := testutil.NewServerConfigs(t)
42-
g := newGRPCServer(&c, 9999, cfg)
42+
g, err := newGRPCServer(&c, 9999, cfg)
43+
require.NoError(t, err)
4344

4445
assert.NotNil(t, g, "TEST Failed.\n")
4546
}
@@ -48,7 +49,8 @@ func TestGRPCServer_AddServerOptions(t *testing.T) {
4849
Logger: logging.NewLogger(logging.DEBUG),
4950
}
5051
cfg := testutil.NewServerConfigs(t)
51-
g := newGRPCServer(&c, 9999, cfg)
52+
g, err := newGRPCServer(&c, 9999, cfg)
53+
require.NoError(t, err)
5254

5355
option1 := grpc.ConnectionTimeout(5 * time.Second)
5456
option2 := grpc.MaxRecvMsgSize(1024 * 1024)
@@ -63,7 +65,8 @@ func TestGRPCServer_AddUnaryInterceptors(t *testing.T) {
6365
Logger: logging.NewLogger(logging.DEBUG),
6466
}
6567
cfg := testutil.NewServerConfigs(t)
66-
g := newGRPCServer(&c, 9999, cfg)
68+
g, err := newGRPCServer(&c, 9999, cfg)
69+
require.NoError(t, err)
6770

6871
interceptor1 := func(ctx context.Context, req any, _ *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (any, error) {
6972
return handler(ctx, req)
@@ -83,7 +86,7 @@ func TestGRPCServer_CreateServer(t *testing.T) {
8386
Logger: logging.NewLogger(logging.DEBUG),
8487
}
8588
cfg := testutil.NewServerConfigs(t)
86-
g := newGRPCServer(&c, 9999, cfg)
89+
g, err := newGRPCServer(&c, 9999, cfg)
8790
require.NoError(t, err)
8891

8992
g.createServer()
@@ -98,7 +101,8 @@ func TestGRPCServer_RegisterService(t *testing.T) {
98101
Logger: logging.NewLogger(logging.DEBUG),
99102
}
100103
cfg := testutil.NewServerConfigs(t)
101-
g := newGRPCServer(&c, 9999, cfg)
104+
g, err := newGRPCServer(&c, 9999, cfg)
105+
require.NoError(t, err)
102106
g.createServer()
103107

104108
healthServer := health.NewServer()
@@ -138,7 +142,7 @@ func TestGRPC_ServerRun(t *testing.T) {
138142

139143
g := &grpcServer{
140144
port: tc.port,
141-
cconfig: testutil.NewServerConfigs(t),
145+
cconfig: getConfigs(t),
142146
}
143147

144148
go func() {
@@ -161,7 +165,8 @@ func TestGRPC_ServerShutdown(t *testing.T) {
161165
Logger: logging.NewLogger(logging.DEBUG),
162166
}
163167
cfg := testutil.NewServerConfigs(t)
164-
g := newGRPCServer(&c, 9999, cfg)
168+
g, err := newGRPCServer(&c, 9999, cfg)
169+
require.NoError(t, err)
165170

166171
go g.Run(&c)
167172

@@ -181,7 +186,8 @@ func TestGRPC_ServerShutdown_ContextCanceled(t *testing.T) {
181186
Logger: logging.NewLogger(logging.DEBUG),
182187
}
183188
cfg := testutil.NewServerConfigs(t)
184-
g := newGRPCServer(&c, 9999, cfg)
189+
g, err := newGRPCServer(&c, 9999, cfg)
190+
require.NoError(t, err)
185191

186192
go g.Run(&c)
187193

@@ -270,7 +276,8 @@ func TestGRPC_Shutdown_BeforeStart(t *testing.T) {
270276
c := &container.Container{Logger: logger}
271277

272278
cfg := testutil.NewServerConfigs(t)
273-
g := newGRPCServer(c, 9999, cfg)
279+
g, err := newGRPCServer(c, 9999, cfg)
280+
require.NoError(t, err)
274281

275282
ctx, cancel := context.WithTimeout(t.Context(), 500*time.Millisecond)
276283
defer cancel()
@@ -348,7 +355,8 @@ func TestApp_WithReflection(t *testing.T) {
348355
app := New()
349356
app.container = c
350357
cfg := testutil.NewServerConfigs(t)
351-
app.grpcServer = newGRPCServer(c, 9999, cfg)
358+
app.grpcServer, err = newGRPCServer(c, 9999, cfg)
359+
require.NoError(t, err)
352360
app.grpcServer.createServer()
353361

354362
services := app.grpcServer.server.GetServiceInfo()

0 commit comments

Comments
 (0)