hongbo-miao/hongbomiao.com

package routes

import (
    "github.com/gin-contrib/logger"
    "github.com/gin-gonic/gin"
    "github.com/hongbo-miao/hongbomiao.com/api-go/internal/graphql_server/controllers"
    "github.com/hongbo-miao/hongbomiao.com/api-go/internal/graphql_server/handlers"
    sharedControllers "github.com/hongbo-miao/hongbomiao.com/api-go/internal/shared/controllers"
    sharedHandlers "github.com/hongbo-miao/hongbomiao.com/api-go/internal/shared/handlers"
    "github.com/minio/minio-go/v7"
    "github.com/redis/go-redis/v9"
    "go.elastic.co/apm/module/apmgin/v2"
    "go.opentelemetry.io/contrib/instrumentation/github.com/gin-gonic/gin/otelgin"
)

func SetupRouter(env string, rdb *redis.Client, minioClient *minio.Client) *gin.Engine {
    r := gin.New()
    r.Use(apmgin.Middleware(r))
    if env == "development" {
        r.Use(handlers.CORSHandler())
    }
    r.Use(logger.SetLogger())
    r.Use(otelgin.Middleware("hm-graphql-server"))
    r.GET("/", sharedControllers.Health)
    r.GET("/metrics", sharedHandlers.PrometheusHandler())
    r.POST("/graphql", handlers.GraphQLHandler(rdb))
    r.POST("/hasura/update-seed", controllers.UpdateSeed)
    r.POST("/hasura/role-event-trigger", controllers.RoleEventTrigger)
    r.POST("/predict", controllers.Predict)
    r.POST("/upload", controllers.Upload(minioClient))
    return r
}