fix:update feature-flag - enable database
This commit is contained in:
parent
0a7b336cca
commit
188ebf853c
2
Makefile
2
Makefile
@ -106,7 +106,7 @@ docker-compose-up:
|
|||||||
# Stop Docker Compose services for local development
|
# Stop Docker Compose services for local development
|
||||||
docker-compose-down:
|
docker-compose-down:
|
||||||
@echo "Stopping all development services..."
|
@echo "Stopping all development services..."
|
||||||
docker-compose down
|
docker-compose down -v
|
||||||
|
|
||||||
# Run Docker Compose for production
|
# Run Docker Compose for production
|
||||||
docker-compose-prod-up:
|
docker-compose-prod-up:
|
||||||
|
|||||||
@ -24,10 +24,14 @@ type HTTPService struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func NewHTTPService(cfg *config.Config, db *database.Database) *HTTPService {
|
func NewHTTPService(cfg *config.Config, db *database.Database) *HTTPService {
|
||||||
|
var gormDB *gorm.DB
|
||||||
|
if db != nil {
|
||||||
|
gormDB = db.DB
|
||||||
|
}
|
||||||
return &HTTPService{
|
return &HTTPService{
|
||||||
server: http.NewServer(cfg, db.DB),
|
server: http.NewServer(cfg, gormDB), // gormDB can be nil here
|
||||||
cfg: cfg,
|
cfg: cfg,
|
||||||
db: db,
|
db: db, // db itself (the *database.Database wrapper) can also be nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -108,26 +112,44 @@ func main() {
|
|||||||
}
|
}
|
||||||
lifecycleMgr := lifecycle.New(shutdownTimeout)
|
lifecycleMgr := lifecycle.New(shutdownTimeout)
|
||||||
|
|
||||||
// Initialize database connection
|
var gormDB *gorm.DB // Declare gormDB for database connection
|
||||||
db, err := database.NewConnection(&cfg.Database)
|
var dbInstance *database.Database // Declare dbInstance for HTTPService and other potential users
|
||||||
if err != nil {
|
|
||||||
logger.WithError(err).Fatal("Failed to connect to database")
|
if feature.IsEnabled(feature.EnableDatabase) {
|
||||||
|
logger.Info("Feature flag 'enable_database' is true. Initializing database connection...")
|
||||||
|
// Initialize database connection
|
||||||
|
var dbErr error
|
||||||
|
gormDB, dbErr = database.NewConnection(&cfg.Database)
|
||||||
|
if dbErr != nil {
|
||||||
|
logger.WithError(dbErr).Fatal("Failed to connect to database")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run database migrations
|
||||||
|
if err := database.Migrate(cfg.Database); err != nil { // Migrate still needs cfg.Database for path etc.
|
||||||
|
logger.WithError(err).Fatal("Failed to migrate database")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Register database cleanup on shutdown
|
||||||
|
lifecycleMgr.Register(&databaseService{db: gormDB})
|
||||||
|
|
||||||
|
// Prepare dbInstance for HTTPService
|
||||||
|
dbInstance = &database.Database{DB: gormDB, Config: &cfg.Database}
|
||||||
|
} else {
|
||||||
|
logger.Info("Feature flag 'enable_database' is false. Skipping database initialization.")
|
||||||
|
// gormDB remains nil
|
||||||
|
// dbInstance remains nil
|
||||||
|
// No databaseService is registered with lifecycleMgr if DB is disabled.
|
||||||
}
|
}
|
||||||
|
|
||||||
// Run database migrations
|
// Initialize HTTP service
|
||||||
if err := database.Migrate(cfg.Database); err != nil {
|
// NewHTTPService and its chain (http.NewServer, http.SetupRouter)
|
||||||
logger.WithError(err).Fatal("Failed to migrate database")
|
// must be able to handle a nil dbInstance or a dbInstance with a nil DB.
|
||||||
}
|
// The NewHTTPService function already checks if its db argument (or db.DB) is nil.
|
||||||
|
httpService := NewHTTPService(cfg, dbInstance) // dbInstance can be nil here
|
||||||
// Register database cleanup on shutdown
|
|
||||||
lifecycleMgr.Register(&databaseService{db: db})
|
|
||||||
|
|
||||||
// Initialize HTTP service with database
|
|
||||||
httpService := NewHTTPService(cfg, &database.Database{DB: db})
|
|
||||||
if httpService == nil {
|
if httpService == nil {
|
||||||
logger.Fatal("Failed to create HTTP service")
|
logger.Fatal("Failed to create HTTP service")
|
||||||
}
|
}
|
||||||
lifecycleMgr.Register(httpService)
|
lifecycleMgr.Register(httpService) // HTTP service is always registered
|
||||||
|
|
||||||
// Start all services
|
// Start all services
|
||||||
logger.Info("Đang khởi động các dịch vụ...")
|
logger.Info("Đang khởi động các dịch vụ...")
|
||||||
|
|||||||
@ -19,11 +19,11 @@ server:
|
|||||||
|
|
||||||
database:
|
database:
|
||||||
driver: "postgres"
|
driver: "postgres"
|
||||||
host: "localhost"
|
host: "postgres"
|
||||||
port: 5432
|
port: 5432
|
||||||
username: "postgres"
|
username: "postgres"
|
||||||
password: "postgres"
|
password: "postgres"
|
||||||
database: "ulflow"
|
database: "ulflow_zee"
|
||||||
ssl_mode: "disable"
|
ssl_mode: "disable"
|
||||||
max_open_conns: 25
|
max_open_conns: 25
|
||||||
max_idle_conns: 5
|
max_idle_conns: 5
|
||||||
|
|||||||
@ -26,58 +26,30 @@ services:
|
|||||||
container_name: ulflow-postgres
|
container_name: ulflow-postgres
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
environment:
|
environment:
|
||||||
POSTGRES_USER: ${DB_USER:-user}
|
POSTGRES_USER: postgres
|
||||||
POSTGRES_PASSWORD: ${DB_PASSWORD:-password}
|
POSTGRES_PASSWORD: Minhtuyen0605
|
||||||
POSTGRES_DB: ${DB_NAME:-ulflow_db}
|
POSTGRES_DB: ulflow_zee
|
||||||
ports:
|
ports:
|
||||||
- "5432:5432"
|
- "5432:5432"
|
||||||
volumes:
|
volumes:
|
||||||
- postgres-data:/var/lib/postgresql/data
|
- postgres-data:/var/lib/postgresql/data
|
||||||
networks:
|
networks:
|
||||||
- ulflow-network
|
- ulflow-network
|
||||||
|
adminer:
|
||||||
# Gitea (Git Server, CI/CD, Registry)
|
image: adminer:latest # Hoặc adminer:4.8.1 (chọn phiên bản cụ thể nếu muốn)
|
||||||
gitea:
|
container_name: ulflow-adminer
|
||||||
image: gitea/gitea:1.21
|
|
||||||
container_name: ulflow-gitea
|
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
environment:
|
|
||||||
- USER_UID=1000
|
|
||||||
- USER_GID=1000
|
|
||||||
- GITEA__database__DB_TYPE=postgres
|
|
||||||
- GITEA__database__HOST=postgres:5432
|
|
||||||
- GITEA__database__NAME=gitea
|
|
||||||
- GITEA__database__USER=${DB_USER:-user}
|
|
||||||
- GITEA__database__PASSWD=${DB_PASSWORD:-password}
|
|
||||||
ports:
|
ports:
|
||||||
- "3001:3000"
|
- "8080:8080" # Ánh xạ cổng 8080 của container ra cổng 8080 của host
|
||||||
- "2222:22"
|
environment:
|
||||||
volumes:
|
ADMINER_DEFAULT_SERVER: postgres # Tên service của PostgreSQL trong Docker Compose
|
||||||
- gitea-data:/data
|
|
||||||
- /etc/timezone:/etc/timezone:ro
|
|
||||||
- /etc/localtime:/etc/localtime:ro
|
|
||||||
depends_on:
|
depends_on:
|
||||||
- postgres
|
- postgres # Đảm bảo postgres khởi động trước adminer
|
||||||
networks:
|
|
||||||
- ulflow-network
|
|
||||||
|
|
||||||
# Gitea Runner for CI/CD
|
|
||||||
gitea-runner:
|
|
||||||
image: gitea/act_runner:latest
|
|
||||||
container_name: ulflow-gitea-runner
|
|
||||||
restart: unless-stopped
|
|
||||||
volumes:
|
|
||||||
- /var/run/docker.sock:/var/run/docker.sock
|
|
||||||
- gitea-runner-data:/data
|
|
||||||
depends_on:
|
|
||||||
- gitea
|
|
||||||
networks:
|
networks:
|
||||||
- ulflow-network
|
- ulflow-network
|
||||||
|
|
||||||
volumes:
|
volumes:
|
||||||
postgres-data:
|
postgres-data:
|
||||||
gitea-data:
|
|
||||||
gitea-runner-data:
|
|
||||||
go-modules:
|
go-modules:
|
||||||
|
|
||||||
networks:
|
networks:
|
||||||
|
|||||||
@ -107,6 +107,7 @@ func (l *ViperConfigLoader) loadEnvFile(v *viper.Viper) error {
|
|||||||
if err := os.Setenv(key, val); err != nil {
|
if err := os.Setenv(key, val); err != nil {
|
||||||
return fmt.Errorf("failed to set environment variable %s: %w", key, err)
|
return fmt.Errorf("failed to set environment variable %s: %w", key, err)
|
||||||
}
|
}
|
||||||
|
fmt.Println("Loaded environment variable:", key, "=", val)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
@ -4,6 +4,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
"zee/internal/adapter/postgres"
|
"zee/internal/adapter/postgres"
|
||||||
"zee/internal/helper/config"
|
"zee/internal/helper/config"
|
||||||
|
"zee/internal/helper/logger" // Added logger import
|
||||||
"zee/internal/service"
|
"zee/internal/service"
|
||||||
"zee/internal/transport/http/handler"
|
"zee/internal/transport/http/handler"
|
||||||
"zee/internal/transport/http/middleware"
|
"zee/internal/transport/http/middleware"
|
||||||
@ -31,60 +32,72 @@ func SetupRouter(cfg *config.Config, db *gorm.DB) *gin.Engine {
|
|||||||
securityCfg := middleware.DefaultSecurityConfig()
|
securityCfg := middleware.DefaultSecurityConfig()
|
||||||
securityCfg.Apply(router)
|
securityCfg.Apply(router)
|
||||||
|
|
||||||
// Khởi tạo repositories
|
// Khởi tạo các handlers không phụ thuộc DB
|
||||||
userRepo := postgres.NewUserRepository(db)
|
|
||||||
roleRepo := postgres.NewRoleRepository(db)
|
|
||||||
|
|
||||||
// Get JWT configuration from config
|
|
||||||
jwtSecret := "your-secret-key" // Default fallback
|
|
||||||
accessTokenExpire := 24 * time.Hour
|
|
||||||
|
|
||||||
// Override with config values if available
|
|
||||||
if cfg.JWT.Secret != "" {
|
|
||||||
jwtSecret = cfg.JWT.Secret
|
|
||||||
}
|
|
||||||
if cfg.JWT.AccessTokenExpire > 0 {
|
|
||||||
accessTokenExpire = time.Duration(cfg.JWT.AccessTokenExpire) * time.Minute
|
|
||||||
}
|
|
||||||
|
|
||||||
// Khởi tạo services
|
|
||||||
authSvc := service.NewAuthService(
|
|
||||||
userRepo,
|
|
||||||
roleRepo,
|
|
||||||
jwtSecret,
|
|
||||||
accessTokenExpire,
|
|
||||||
)
|
|
||||||
|
|
||||||
// Khởi tạo middleware
|
|
||||||
authMiddleware := middleware.NewAuthMiddleware(authSvc)
|
|
||||||
_ = authMiddleware // TODO: Use authMiddleware when needed
|
|
||||||
|
|
||||||
// Khởi tạo các handlers
|
|
||||||
healthHandler := handler.NewHealthHandler(cfg)
|
healthHandler := handler.NewHealthHandler(cfg)
|
||||||
authHandler := handler.NewAuthHandler(authSvc)
|
|
||||||
|
|
||||||
// Đăng ký các routes
|
// Đăng ký các routes không phụ thuộc DB
|
||||||
|
|
||||||
// Health check routes (public)
|
|
||||||
router.GET("/ping", healthHandler.Ping)
|
router.GET("/ping", healthHandler.Ping)
|
||||||
router.GET("/health", healthHandler.HealthCheck)
|
router.GET("/health", healthHandler.HealthCheck)
|
||||||
|
|
||||||
// Auth routes (public)
|
// Các thành phần và routes phụ thuộc vào DB
|
||||||
authGroup := router.Group("/api/v1/auth")
|
if db != nil {
|
||||||
{
|
logger.Info("Database connection is available. Initializing DB-dependent services and routes...")
|
||||||
authGroup.POST("/register", authHandler.Register)
|
// Khởi tạo repositories
|
||||||
authGroup.POST("/login", authHandler.Login)
|
userRepo := postgres.NewUserRepository(db)
|
||||||
authGroup.POST("/refresh", authHandler.RefreshToken)
|
roleRepo := postgres.NewRoleRepository(db)
|
||||||
authGroup.POST("/logout", authMiddleware.Authenticate(), authHandler.Logout)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Protected API routes
|
// Get JWT configuration from config
|
||||||
api := router.Group("/api/v1")
|
jwtSecret := "your-secret-key" // Default fallback
|
||||||
api.Use(authMiddleware.Authenticate())
|
accessTokenExpire := 24 * time.Hour
|
||||||
{
|
|
||||||
// Ví dụ về protected endpoints
|
// Override with config values if available
|
||||||
// api.GET("/profile", userHandler.GetProfile)
|
if cfg.JWT.Secret != "" {
|
||||||
// api.PUT("/profile", userHandler.UpdateProfile)
|
jwtSecret = cfg.JWT.Secret
|
||||||
|
}
|
||||||
|
if cfg.JWT.AccessTokenExpire > 0 {
|
||||||
|
accessTokenExpire = time.Duration(cfg.JWT.AccessTokenExpire) * time.Minute
|
||||||
|
}
|
||||||
|
|
||||||
|
// Khởi tạo services
|
||||||
|
authSvc := service.NewAuthService(
|
||||||
|
userRepo,
|
||||||
|
roleRepo,
|
||||||
|
jwtSecret,
|
||||||
|
accessTokenExpire,
|
||||||
|
)
|
||||||
|
|
||||||
|
// Khởi tạo middleware
|
||||||
|
authMiddleware := middleware.NewAuthMiddleware(authSvc)
|
||||||
|
|
||||||
|
// Khởi tạo các handlers phụ thuộc DB
|
||||||
|
authHandler := handler.NewAuthHandler(authSvc)
|
||||||
|
|
||||||
|
// Auth routes (public)
|
||||||
|
authGroup := router.Group("/api/v1/auth")
|
||||||
|
{
|
||||||
|
authGroup.POST("/register", authHandler.Register)
|
||||||
|
authGroup.POST("/login", authHandler.Login)
|
||||||
|
authGroup.POST("/refresh", authHandler.RefreshToken)
|
||||||
|
authGroup.POST("/logout", authMiddleware.Authenticate(), authHandler.Logout)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Protected API routes
|
||||||
|
api := router.Group("/api/v1")
|
||||||
|
api.Use(authMiddleware.Authenticate())
|
||||||
|
{
|
||||||
|
// Ví dụ về protected endpoints
|
||||||
|
// api.GET("/profile", userHandler.GetProfile)
|
||||||
|
// api.PUT("/profile", userHandler.UpdateProfile)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
logger.Info("Database is disabled via feature flag or connection failed. DB-dependent routes (e.g., /api/v1/auth, /api/v1/*) will not be available.")
|
||||||
|
// Optionally, register placeholder routes that return 503 Service Unavailable
|
||||||
|
// router.Any("/api/v1/auth/*any", func(c *gin.Context) {
|
||||||
|
// c.JSON(http.StatusServiceUnavailable, gin.H{"error": "Authentication service is currently disabled"})
|
||||||
|
// })
|
||||||
|
// router.Any("/api/v1/*any", func(c *gin.Context) {
|
||||||
|
// c.JSON(http.StatusServiceUnavailable, gin.H{"error": "API service is currently disabled due to database unavailability"})
|
||||||
|
// })
|
||||||
}
|
}
|
||||||
|
|
||||||
return router
|
return router
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user