feat: some refactoring

cmd/apiserver/config/config.go

@@ -0,0 +1,78 @@
+package config
+
+import (
+	"flag"
+	"log"
+	"time"
+
+	"github.com/BurntSushi/toml"
+)
+
+type Server struct {
+	Address      string        `toml:"address"`
+	ReadTimeout  time.Duration `toml:"read_timeout"`
+	WriteTimeout time.Duration `toml:"write_timeout"`
+	IdleTimeout  time.Duration `toml:"idle_timeout"`
+}
+
+type Logging struct {
+	Level  string `toml:"level"`
+	Format string `toml:"format"`
+	Output string `toml:"output"`
+}
+
+type Request struct {
+	MaxBodySize int64 `toml:"max_body_size"`
+}
+
+type Config struct {
+	Server  Server  `toml:"server"`
+	Logging Logging `toml:"logging"`
+	Request Request `toml:"request"`
+}
+
+var (
+	configPath string
+)
+
+func init() {
+	flag.StringVar(&configPath, "config-path", "./configs/development.toml", "path to config")
+}
+
+func Load() *Config {
+	flag.Parse()
+
+	var config Config
+
+	if _, err := toml.DecodeFile(configPath, &config); err != nil {
+		log.Fatalf("Failed to load config: %v", err)
+	}
+
+	// Set defaults if not specified
+	if config.Server.Address == "" {
+		config.Server.Address = ":8080"
+	}
+	if config.Server.ReadTimeout == 0 {
+		config.Server.ReadTimeout = 10 * time.Second
+	}
+	if config.Server.WriteTimeout == 0 {
+		config.Server.WriteTimeout = 10 * time.Second
+	}
+	if config.Server.IdleTimeout == 0 {
+		config.Server.IdleTimeout = 60 * time.Second
+	}
+	if config.Logging.Level == "" {
+		config.Logging.Level = "info"
+	}
+	if config.Logging.Format == "" {
+		config.Logging.Format = "json"
+	}
+	if config.Logging.Output == "" {
+		config.Logging.Output = "stdout"
+	}
+	if config.Request.MaxBodySize == 0 {
+		config.Request.MaxBodySize = 10 * 1024 * 1024 // 10MB
+	}
+
+	return &config
+}

cmd/apiserver/main.go

@@ -1,23 +1,31 @@
 package main
 
 import (
-	"flag"
 	"log"
 
-	"git.nwaifu.su/sergey/MyGoServer/internal/app/apiserver"
+	"git.nwaifu.su/sergey/MyGoServer/cmd/apiserver/config"
+	"git.nwaifu.su/sergey/MyGoServer/internal/apiserver/server"
 )
 
-var (
-	configPath string
-)
+// @title Some GoLang server
+// @version 1.0
+// @description This is some GoLang server.
+// @termsOfService http://swagger.io/terms/
 
-func init() {
-	flag.StringVar(&configPath, "config-path", "/go/bin/configs/server.toml", "path to config")
-}
+// @contact.name Sergey Elpashev
+// @contact.url https://nwaifu.su
+// @contact.email mrsedan@nwaifu.su
 
+// @license.name MIT
+// @license.url https://opensource.org/license/mit
+
+// @host localhost:8080
+// @BasePath /
 func main() {
-	flag.Parse()
-	if err := apiserver.Start(); err != nil {
+	cfg := config.Load()
+
+	srv := server.NewServer(cfg)
+	if err := srv.Start(); err != nil {
 		log.Fatal(err)
 	}
 }