Feat/connect db

README.md

@@ -9,3 +9,14 @@
 ```bash
 ./test-db.sh
 ```
+
+## DB 마이그레이션
+
+[ent, atlas migration 가이드](https://entgo.io/docs/versioned-migrations#generating-versioned-migration-files)
+
+schema 작성 후 아래 스크립트 실행
+```bash
+./create_migration.sh <migration_name>
+```
+- atlas 설치 필요
+- migration파일 생성 후 필요에 따라 migratin 파일 수정시 atlas migrate hash로 적용 필요

cmd/main.go

@@ -2,22 +2,33 @@ package main
 
 import (
 	"fmt"
-	"os"
+	"log"
+
+	"github.com/techbloghub/server/config"
+	"github.com/techbloghub/server/internal/database"
 
 	"github.com/gin-gonic/gin"
-	"github.com/joho/godotenv"
+	_ "github.com/lib/pq"
 )
 
 func main() {
-	godotenv.Load(".env")
+	cfg, cfgErr := config.NewConfig()
+	if cfgErr != nil {
+		log.Fatalf("failed to load config: %v", cfgErr)
+	}
+
+	// DB 연결
+	client, errPg := database.ConnectDatabase(cfg)
+	if errPg != nil {
+		log.Fatalf("failed to connect database: %v", errPg)
+	}
+	defer client.Close()
 
-	// PORT 환경변수에서 가져오기
-	// 후에 이런 config값 관리할것들 많아지면 후에 Config struct등으로 분리 고려
-	port := getEnvWithDefault("PORT", "8080")
+	// 서버 실행
 	r := setRouter()
-	err := r.Run(":" + port)
-	if err != nil {
-		fmt.Println("Error while running server: ", err)
+	routerErr := r.Run(":" + cfg.ServerConfig.Port)
+	if routerErr != nil {
+		fmt.Println("Error while running server: ", cfgErr)
 		return
 	}
 }
@@ -29,10 +40,3 @@ func setRouter() *gin.Engine {
 	})
 	return r
 }
-
-func getEnvWithDefault(key, defaultValue string) string {
-	if value := os.Getenv(key); value != "" {
-		return value
-	}
-	return defaultValue
-}

config/config.go

@@ -0,0 +1,97 @@
+package config
+
+import (
+	"fmt"
+	"log"
+	"os"
+
+	"github.com/joho/godotenv"
+)
+
+type Config struct {
+	PostgresConfig
+	ServerConfig
+}
+
+type PostgresConfig struct {
+	Host     string
+	Port     string
+	User     string
+	Password string
+	Db       string
+}
+
+type ServerConfig struct {
+	Port string
+	Env  string
+}
+
+func (cfg *PostgresConfig) ToMap() map[string]string {
+	return map[string]string{
+		"HOST":     cfg.Host,
+		"PORT":     cfg.Port,
+		"USER":     cfg.User,
+		"PASSWORD": cfg.Password,
+		"DB":       cfg.Db,
+	}
+}
+
+func (cfg *ServerConfig) ToMap() map[string]string {
+	return map[string]string{
+		"PORT": cfg.Port,
+		"ENV":  cfg.Env,
+	}
+}
+
+func NewConfig() (*Config, error) {
+	errEnv := godotenv.Load(".env")
+	if errEnv != nil {
+		log.Print("failed to reading .env", errEnv)
+	}
+
+	postgresConf := PostgresConfig{
+		Host:     os.Getenv("POSTGRES_HOST"),
+		Port:     os.Getenv("POSTGRES_PORT"),
+		User:     os.Getenv("POSTGRES_USER"),
+		Password: os.Getenv("POSTGRES_PASSWORD"),
+		Db:       os.Getenv("POSTGRES_DB"),
+	}
+
+	serverConf := ServerConfig{
+		Port: os.Getenv("PORT"),
+		Env:  os.Getenv("ENV"),
+	}
+
+	cfg := &Config{
+		PostgresConfig: postgresConf,
+		ServerConfig:   serverConf,
+	}
+
+	if err := validateEnvs(cfg); err != nil {
+		return nil, err
+	}
+
+	return cfg, nil
+}
+
+func validateEnvs(cfg *Config) error {
+	missingEnvs := []string{}
+
+	missingEnvs = append(missingEnvs, findEmptyValueKeys(cfg.PostgresConfig.ToMap())...)
+	missingEnvs = append(missingEnvs, findEmptyValueKeys(cfg.ServerConfig.ToMap())...)
+
+	if len(missingEnvs) > 0 {
+		return fmt.Errorf("missing envs: %v", missingEnvs)
+	}
+	return nil
+}
+
+func findEmptyValueKeys(m map[string]string) []string {
+	keys := []string{}
+	for k, v := range m {
+		if v == "" {
+			keys = append(keys, k)
+		}
+	}
+	return keys
+}

create_migration.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+
+if [ -z "$1" ]; then
+  echo "Usage: $0 <migration_name>"
+  exit 1
+fi
+
+MIGRATION_NAME=$1
+
+atlas migrate diff "$MIGRATION_NAME" \
+  --dir "file://ent/migrate/migrations" \
+  --to "ent://ent/schema" \
+  --dev-url "docker://postgres/15/test?search_path=public"