71 lines
2.2 KiB
Go
71 lines
2.2 KiB
Go
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"
|
|
"gitlab.michelsen.id/phillmichelsen/tessera/services/data_service/internal/router"
|
|
"gitlab.michelsen.id/phillmichelsen/tessera/services/data_service/internal/server"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/reflection"
|
|
)
|
|
|
|
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 Control Server
|
|
grpcControlServer := grpc.NewServer()
|
|
go func() {
|
|
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 streaming server listening on :50052")
|
|
if err := grpcStreamingServer.Serve(grpcLis); err != nil {
|
|
log.Fatalf("Failed to serve gRPC: %v", err)
|
|
}
|
|
}()
|
|
|
|
// 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 := socketStreamingServer.Serve(socketLis); err != nil {
|
|
log.Fatalf("Socket server error: %v", err)
|
|
}
|
|
}()
|
|
|
|
// Block main forever
|
|
select {}
|
|
}
|