query-orchestration/cmd/grpc/main.go

package main

import (
	"context"
	"gotemplate/api/grpc/controllers"
	"gotemplate/api/grpc/spec"
	"gotemplate/internal/collector"
	"gotemplate/internal/database"
	"gotemplate/internal/database/repository"
	"gotemplate/internal/export"
	"gotemplate/internal/otel"
	"gotemplate/internal/query"
	"log"
	"net"
	"strconv"

	_ "github.com/lib/pq"
	"google.golang.org/grpc"
)

func main() {
	ctx := context.Background()

	closeTracer := otel.Init(ctx)
	defer closeTracer()

	database.RunMigrations()

	host := "0.0.0.0"
	port := 8080
	address := net.JoinHostPort(host, strconv.Itoa(port))

	lis, err := net.Listen("tcp", address)
	if err != nil {
		log.Fatalf("failed to listen: %v", err)
	}

	pool := database.GetDBPool(ctx)
	db := repository.New(pool)

	grpcServer := grpc.NewServer()
	spec.RegisterQueryServiceServer(grpcServer, controllers.NewQueryController(*query.New(db)))
	spec.RegisterExportServiceServer(grpcServer, controllers.NewExportController(*export.New(db)))
	spec.RegisterJobCollectorServiceServer(grpcServer, controllers.NewJobCollectorController(*collector.New(db)))

	log.Printf("Listening on port %d", port)
	if err := grpcServer.Serve(lis); err != nil {
		log.Fatalf("Failed to serve: %v", err)
	}
}