env 환경변수 사용

프로젝트에서 DB접속주소, ID/PW 등의 정보들을 포함하여 사용할 경우가 있습니다.
Git으로 프로젝트를 올리면 외부에서 DB접속이 가능하고 ID/PW의 계정정보까지 확인이 가능한 단점이 있어 config.json파일이나 env환경변수를 사용하여 보안을 강화하는법이 있습니다.

 

환경변수 직접 가져와서 사용하기

host := os.Getenv("DB_URL")
port := os.Getenv("DB_PORT")
user := os.Getenv("DB_ID")
password := os.Getenv("DB_PW")

#DB연결
connString := fmt.Sprintf("host=%s port=%d user=%s password=%s sslmode=disable", host, port, user, password)
DB, err := sql.Open("postgres", connString)
if err != nil {
    panic(err)
}

 

Viper패키지를 사용하여 환경변수 사용

• .env에서 읽어오기

viper.SetConfigFile(".env")
err := viper.ReadInConfig()

host := viper.Get("HOST")
port := viper.Get("PORT")

 

• config.json 파일에서 읽어오기

viper.AddConfigPath("./config")
viper.SetConfigName("config")
viper.SetConfigType("json")
viper.ReadInConfig()

port := viper.Get("db.port")

- config경로 내부 config.json 파일을 읽어옵니다.