temp
This commit is contained in:
@@ -1,148 +1,24 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"sipro-mps/internal/config"
|
||||
"sipro-mps/internal/db"
|
||||
"sipro-mps/internal/marketplace"
|
||||
ozon "sipro-mps/internal/ozon/products"
|
||||
ozon_products "sipro-mps/internal/ozon/products"
|
||||
"sipro-mps/internal/redis"
|
||||
"sipro-mps/internal/tasks/client"
|
||||
wb "sipro-mps/internal/wb/products"
|
||||
ym "sipro-mps/internal/ym/products"
|
||||
"sipro-mps/internal/transport/grpc"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
"github.com/joho/godotenv"
|
||||
"google.golang.org/grpc"
|
||||
"go.uber.org/fx"
|
||||
_ "google.golang.org/grpc/encoding/gzip"
|
||||
)
|
||||
|
||||
func logMessage(level string, format string, a ...interface{}) {
|
||||
const (
|
||||
green = "\033[32m"
|
||||
red = "\033[31m"
|
||||
yellow = "\033[33m"
|
||||
blue = "\033[34m"
|
||||
reset = "\033[0m"
|
||||
)
|
||||
switch level {
|
||||
case "info":
|
||||
fmt.Printf("%s✅ [INFO]%s %s %s\n", green, reset, fmt.Sprintf(format, a...), "ℹ️")
|
||||
case "error":
|
||||
fmt.Printf("%s❌ [ERROR]%s %s %s\n", red, reset, fmt.Sprintf(format, a...), "🚨")
|
||||
default:
|
||||
fmt.Printf("%s[LOG]%s %s\n", blue, reset, fmt.Sprintf(format, a...))
|
||||
}
|
||||
}
|
||||
|
||||
func createGrpcServer(pool *pgxpool.Pool) {
|
||||
|
||||
lis, err := net.Listen("tcp", ":8080")
|
||||
if err != nil {
|
||||
fmt.Printf("failed to listen: %v\n", err)
|
||||
return
|
||||
}
|
||||
grpcServer := grpc.NewServer()
|
||||
|
||||
repo, err := marketplace.RegisterAdapterGRPC(grpcServer, pool)
|
||||
if err != nil {
|
||||
fmt.Printf("failed to register gRPC server: %v\n", err)
|
||||
return
|
||||
}
|
||||
_, err = ozon.RegisterAdapterGRPC(grpcServer, *repo)
|
||||
if err != nil {
|
||||
fmt.Printf("failed to register Ozon Products gRPC server: %v\n", err)
|
||||
return
|
||||
}
|
||||
_, err = wb.RegisterAdapterGRPC(grpcServer, *repo)
|
||||
if err != nil {
|
||||
fmt.Printf("failed to register Wildberries Products gRPC server: %v\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
_, err = ym.RegisterAdapterGRPC(grpcServer, *repo)
|
||||
if err != nil {
|
||||
fmt.Printf("failed to register Yandex Market Products gRPC server: %v\n", err)
|
||||
return
|
||||
|
||||
}
|
||||
|
||||
fmt.Println("gRPC server registered successfully.")
|
||||
// Start serving gRPC requests
|
||||
fmt.Println("gRPC server is starting on port 8080...")
|
||||
|
||||
if err := grpcServer.Serve(lis); err != nil {
|
||||
fmt.Printf("failed to serve: %v\n", err)
|
||||
}
|
||||
fmt.Println("gRPC server created.")
|
||||
}
|
||||
|
||||
func initDotenv() error {
|
||||
err := godotenv.Load()
|
||||
if err != nil {
|
||||
return fmt.Errorf("error loading .env file: %w", err)
|
||||
}
|
||||
logMessage("info", "Dotenv file loaded successfully. 🌱")
|
||||
return nil
|
||||
}
|
||||
|
||||
func initRedisClient(ctx context.Context) error {
|
||||
err := redis.InitClient(ctx)
|
||||
if err != nil {
|
||||
return fmt.Errorf("error initializing Redis client: %w", err)
|
||||
}
|
||||
//defer redis.CloseClient()
|
||||
logMessage("info", "Redis client initialized successfully. 🟥")
|
||||
return nil
|
||||
}
|
||||
func initRedisLocker() error {
|
||||
err := redis.InitLocker()
|
||||
if err != nil {
|
||||
return fmt.Errorf("error initializing Redis locker: %w", err)
|
||||
}
|
||||
logMessage("info", "Redis locker initialized successfully. 🟥")
|
||||
return nil
|
||||
}
|
||||
|
||||
func main() {
|
||||
err := initDotenv()
|
||||
if err != nil {
|
||||
logMessage("error", "Failed to load .env file: %v", err)
|
||||
}
|
||||
logMessage("info", "Starting the SIPRO Marketplace Server... 🚀1")
|
||||
ctx := context.Background()
|
||||
|
||||
// Initializing the Redis client
|
||||
err = initRedisClient(ctx)
|
||||
if err != nil {
|
||||
logMessage("error", "Failed to initialize Redis client: %v", err)
|
||||
return
|
||||
}
|
||||
defer redis.CloseClient()
|
||||
|
||||
// Initializing the Redis locker
|
||||
err = initRedisLocker()
|
||||
if err != nil {
|
||||
logMessage("error", "Failed to initialize Redis locker: %v", err)
|
||||
return
|
||||
}
|
||||
defer redis.CloseLocker()
|
||||
cfg, err := config.LoadConfig()
|
||||
if err != nil {
|
||||
logMessage("error", "Failed to load configuration: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
client.InitClient(*cfg.Redis)
|
||||
// Initializing pgx connection
|
||||
dbpool, err := pgxpool.New(ctx, os.Getenv("POSTGRES_URL"))
|
||||
if err != nil {
|
||||
logMessage("error", "Failed to connect to PostgreSQL: %v", err)
|
||||
return
|
||||
}
|
||||
defer dbpool.Close()
|
||||
|
||||
createGrpcServer(dbpool)
|
||||
fx.New(
|
||||
config.Module,
|
||||
redis.Module,
|
||||
db.Module,
|
||||
grpc.Module,
|
||||
marketplace.Module,
|
||||
ozon_products.Module,
|
||||
).Run()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user