110 lines
3.1 KiB
Go
110 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/genex/trading-service/internal/interface/http/handler"
|
|
"github.com/genex/trading-service/internal/interface/http/middleware"
|
|
"github.com/genex/trading-service/internal/matching"
|
|
)
|
|
|
|
func main() {
|
|
logger, _ := zap.NewProduction()
|
|
defer logger.Sync()
|
|
|
|
port := os.Getenv("PORT")
|
|
if port == "" {
|
|
port = "3003"
|
|
}
|
|
|
|
// Initialize matching engine
|
|
engine := matching.NewEngine()
|
|
|
|
// Setup Gin router
|
|
r := gin.New()
|
|
r.Use(gin.Recovery())
|
|
|
|
// Health endpoints
|
|
r.GET("/health", func(c *gin.Context) {
|
|
c.JSON(200, gin.H{"status": "ok", "service": "trading-service", "timestamp": time.Now().UTC().Format(time.RFC3339)})
|
|
})
|
|
r.GET("/health/ready", func(c *gin.Context) { c.JSON(200, gin.H{"status": "ready"}) })
|
|
r.GET("/health/live", func(c *gin.Context) { c.JSON(200, gin.H{"status": "alive"}) })
|
|
|
|
// API routes
|
|
api := r.Group("/api/v1")
|
|
|
|
tradeHandler := handler.NewTradeHandler(engine)
|
|
|
|
trades := api.Group("/trades")
|
|
trades.Use(middleware.JWTAuth())
|
|
{
|
|
trades.POST("/orders", tradeHandler.PlaceOrder)
|
|
trades.DELETE("/orders/:id", tradeHandler.CancelOrder)
|
|
}
|
|
|
|
// Public orderbook
|
|
api.GET("/trades/orderbook/:couponId", tradeHandler.GetOrderBook)
|
|
|
|
// Admin routes (require JWT + admin role)
|
|
adminTradeHandler := handler.NewAdminTradeHandler(engine)
|
|
adminMMHandler := handler.NewAdminMMHandler(engine)
|
|
|
|
admin := api.Group("/admin")
|
|
admin.Use(middleware.JWTAuth(), middleware.RequireAdmin())
|
|
{
|
|
// Trade administration
|
|
adminTrades := admin.Group("/trades")
|
|
adminTrades.GET("/stats", adminTradeHandler.GetTradingStats)
|
|
adminTrades.GET("/orders", adminTradeHandler.ListOrders)
|
|
adminTrades.GET("/volume-trend", adminTradeHandler.GetVolumeTrend)
|
|
adminTrades.POST("/orders/:id/investigate", adminTradeHandler.InvestigateOrder)
|
|
|
|
// Market maker administration
|
|
mm := admin.Group("/mm")
|
|
mm.GET("/list", adminMMHandler.ListMarketMakers)
|
|
mm.GET("/:id/details", adminMMHandler.GetMarketMakerDetails)
|
|
mm.POST("/:id/suspend", adminMMHandler.SuspendMarketMaker)
|
|
mm.POST("/:id/resume", adminMMHandler.ResumeMarketMaker)
|
|
mm.GET("/liquidity-pools", adminMMHandler.GetLiquidityPools)
|
|
mm.GET("/order-book-depth", adminMMHandler.GetOrderBookDepth)
|
|
mm.GET("/health-indicators", adminMMHandler.GetHealthIndicators)
|
|
}
|
|
|
|
server := &http.Server{
|
|
Addr: ":" + port,
|
|
Handler: r,
|
|
ReadTimeout: 15 * time.Second,
|
|
WriteTimeout: 15 * time.Second,
|
|
IdleTimeout: 60 * time.Second,
|
|
}
|
|
|
|
go func() {
|
|
logger.Info("Trading Service starting", zap.String("port", port))
|
|
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
logger.Fatal("Server failed", zap.Error(err))
|
|
}
|
|
}()
|
|
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
<-quit
|
|
|
|
logger.Info("Shutting down gracefully...")
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
defer cancel()
|
|
|
|
if err := server.Shutdown(ctx); err != nil {
|
|
logger.Fatal("Server forced to shutdown", zap.Error(err))
|
|
}
|
|
logger.Info("Trading Service stopped")
|
|
}
|