Refactor data service: rename streaming files, update gRPC service methods, and enhance stream management

services/data_service/cmd/main.go

@@ -1,6 +1,10 @@
 package main
 
 import (
+	"fmt"
+	"log"
+	"net"
+
 	pb "gitlab.michelsen.id/phillmichelsen/tessera/pkg/pb/data_service"
 	"gitlab.michelsen.id/phillmichelsen/tessera/services/data_service/internal/manager"
 	"gitlab.michelsen.id/phillmichelsen/tessera/services/data_service/internal/provider/binance"
@@ -8,41 +12,55 @@ import (
 	"gitlab.michelsen.id/phillmichelsen/tessera/services/data_service/internal/server"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/reflection"
-	"log"
-	"net"
 )
 
 func main() {
+	fmt.Println("Starting Data Service...")
 	// Setup
 	r := router.NewRouter()
 	m := manager.NewManager(r)
 	binanceFutures := binance.NewFuturesWebsocket()
 	m.AddProvider("binance_futures_websocket", binanceFutures)
 
-	// gRPC Server
-	grpcServer := grpc.NewServer()
+	// gRPC Control Server
+	grpcControlServer := grpc.NewServer()
 	go func() {
-		pb.RegisterDataServiceStreamingServer(grpcServer, server.NewGRPCStreamingServer(m))
-		reflection.Register(grpcServer)
+		pb.RegisterDataServiceControlServer(grpcControlServer, server.NewGRPCControlServer(m))
+		reflection.Register(grpcControlServer)
 		grpcLis, err := net.Listen("tcp", ":50051")
+		if err != nil {
+			log.Fatalf("Failed to listen for gRPC control: %v", err)
+		}
+		log.Println("gRPC control server listening on :50051")
+		if err := grpcControlServer.Serve(grpcLis); err != nil {
+			log.Fatalf("Failed to serve gRPC control: %v", err)
+		}
+	}()
+
+	// gRPC Streaming Server
+	grpcStreamingServer := grpc.NewServer()
+	go func() {
+		pb.RegisterDataServiceStreamingServer(grpcStreamingServer, server.NewGRPCStreamingServer(m))
+		reflection.Register(grpcStreamingServer)
+		grpcLis, err := net.Listen("tcp", ":50052")
 		if err != nil {
 			log.Fatalf("Failed to listen for gRPC: %v", err)
 		}
-		log.Println("gRPC server listening on :50051")
-		if err := grpcServer.Serve(grpcLis); err != nil {
+		log.Println("gRPC streaming server listening on :50052")
+		if err := grpcStreamingServer.Serve(grpcLis); err != nil {
 			log.Fatalf("Failed to serve gRPC: %v", err)
 		}
 	}()
 
-	// Socket Server
-	socketServer := server.NewSocketStreamingServer(m)
+	// Socket Streaming Server
+	socketStreamingServer := server.NewSocketStreamingServer(m)
 	go func() {
 		socketLis, err := net.Listen("tcp", ":6000")
 		if err != nil {
 			log.Fatalf("Failed to listen for socket: %v", err)
 		}
 		log.Println("Socket server listening on :6000")
-		if err := socketServer.Serve(socketLis); err != nil {
+		if err := socketStreamingServer.Serve(socketLis); err != nil {
 			log.Fatalf("Socket server error: %v", err)
 		}
 	}()