imagini/cmd/server/server.go

package server

import (
    "time"
    "context"
    "net/http"
    "reichard.io/imagini/internal/db"
    "reichard.io/imagini/internal/api"
    "reichard.io/imagini/internal/auth"
    "reichard.io/imagini/internal/config"
    log "github.com/sirupsen/logrus"
)

type Server struct {
    API         *api.API
    Auth        *auth.AuthManager
    Config      *config.Config
    Database    *db.DBManager
    httpServer  *http.Server
}

func NewServer() *Server {
    c := config.Load()
    db := db.NewMgr(c)
    auth := auth.NewMgr(db, c)
    api := api.NewApi(db, auth)

    return &Server{
        API: api,
        Auth: auth,
        Config: c,
        Database: db,
    }
}

func (s *Server) StartServer() {
    listenAddr := (":" + s.Config.ListenPort)

    s.httpServer = &http.Server{
        Handler: s.API.Router,
        Addr: listenAddr,
    }

    go func() {
        err := s.httpServer.ListenAndServe()
        if err != nil {
            log.Error("Error starting server ", err)
            return
        }
    }()
}

func (s *Server) StopServer() {
    ctx, cancel := context.WithTimeout(context.Background(), 5 * time.Second)
    defer cancel()
    s.httpServer.Shutdown(ctx)
}