feat(main): Updated template bootstrap

Author: ekkinox

go.mod

@@ -4,6 +4,7 @@ go 1.23
 
 require (
 	github.com/ankorstore/yokai/config v1.5.0
+	github.com/ankorstore/yokai/fxconfig v1.3.0
 	github.com/ankorstore/yokai/fxcore v1.9.0
 	github.com/ankorstore/yokai/fxgrpcserver v1.3.0
 	github.com/ankorstore/yokai/grpcserver v1.2.0
@@ -17,7 +18,6 @@ require (
 )
 
 require (
-	github.com/ankorstore/yokai/fxconfig v1.3.0 // indirect
 	github.com/ankorstore/yokai/fxgenerate v1.2.0 // indirect
 	github.com/ankorstore/yokai/fxhealthcheck v1.1.0 // indirect
 	github.com/ankorstore/yokai/fxlog v1.1.0 // indirect

internal/bootstrap.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"testing"
 
+	"github.com/ankorstore/yokai/fxconfig"
 	"github.com/ankorstore/yokai/fxcore"
 	"github.com/ankorstore/yokai/fxgrpcserver"
 	"go.uber.org/fx"
@@ -34,7 +35,11 @@ func Run(ctx context.Context) {
 func RunTest(tb testing.TB, options ...fx.Option) {
 	tb.Helper()
 
-	tb.Setenv("APP_CONFIG_PATH", fmt.Sprintf("%s/configs", RootDir))
-
-	Bootstrapper.RunTestApp(tb, fx.Options(options...))
+	Bootstrapper.RunTestApp(
+		tb,
+		// config lookup
+		fxconfig.AsConfigPath(fmt.Sprintf("%s/configs/", RootDir)),
+		// test options
+		fx.Options(options...),
+	)
 }

internal/service/example.go

@@ -47,30 +47,37 @@ func (s *ExampleService) ExampleStreaming(stream proto.ExampleService_ExampleStr
 	logger := log.CtxLogger(ctx)
 
 	for {
-		req, err := stream.Recv()
+		select {
+		case <-ctx.Done():
+			logger.Info().Msg("rpc context cancelled")
 
-		if errors.Is(err, io.EOF) {
-			logger.Info().Msg("end of rpc")
+			return ctx.Err()
+		default:
+			req, err := stream.Recv()
 
-			return nil
-		}
+			if errors.Is(err, io.EOF) {
+				logger.Info().Msg("end of rpc")
 
-		if err != nil {
-			logger.Error().Err(err).Msgf("error while receiving: %v", err)
-		}
+				return nil
+			}
+
+			if err != nil {
+				logger.Error().Err(err).Msgf("error while receiving: %v", err)
+			}
 
-		logger.Info().Msgf("received: %s", req.Text)
+			logger.Info().Msgf("received: %s", req.Text)
 
-		span.AddEvent(fmt.Sprintf("received: %s", req.Text))
+			span.AddEvent(fmt.Sprintf("received: %s", req.Text))
 
-		err = stream.Send(&proto.ExampleResponse{
-			Text: fmt.Sprintf("response from %s: you sent %s", s.config.AppName(), req.Text),
-		})
+			err = stream.Send(&proto.ExampleResponse{
+				Text: fmt.Sprintf("response from %s: you sent %s", s.config.AppName(), req.Text),
+			})
 
-		if err != nil {
-			logger.Error().Err(err).Msgf("error while sending: %v", err)
+			if err != nil {
+				logger.Error().Err(err).Msgf("error while sending: %v", err)
 
-			return err
+				return err
+			}
 		}
 	}
 }