54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package config
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
type DHConfig struct {
|
|
DiskFetchFrequency int `json:"diskFetchFrequency"`
|
|
MaxHistoryAge int `json:"maxHistoryAge"`
|
|
DatabaseFilePath string
|
|
Listen string
|
|
}
|
|
|
|
func GetConfiguration() DHConfig {
|
|
// Load .env file if it exists
|
|
if err := godotenv.Load(); err != nil {
|
|
log.Println("No .env file found")
|
|
}
|
|
|
|
config := DHConfig{
|
|
DiskFetchFrequency: 5, // default value
|
|
MaxHistoryAge: 2592000, // default value
|
|
DatabaseFilePath: "./data.sqlite",
|
|
|
|
Listen: ":8080",
|
|
}
|
|
|
|
if val, exists := os.LookupEnv("DISK_FETCH_FREQUENCY"); exists {
|
|
if intValue, err := strconv.Atoi(val); err == nil {
|
|
config.DiskFetchFrequency = intValue
|
|
}
|
|
}
|
|
|
|
if val, exists := os.LookupEnv("MAX_HISTORY_AGE"); exists {
|
|
if intValue, err := strconv.Atoi(val); err == nil {
|
|
config.MaxHistoryAge = intValue
|
|
}
|
|
}
|
|
|
|
if val, exists := os.LookupEnv("LISTEN"); exists {
|
|
config.Listen = val
|
|
}
|
|
|
|
if val, exists := os.LookupEnv("DATABASE_FILE_PATH"); exists {
|
|
config.DatabaseFilePath = val
|
|
}
|
|
|
|
return config
|
|
}
|