starfleet.go

package main

import (
	"net/http"

	"github.com/rs/zerolog/log"
)

type StarFleetConfig struct {
	Middleware MiddlewareConfig `json:"middleware"`
	Workers    []WorkerConfig   `json:"workers"`
}
type StarFleet struct {
	middleware     Middleware
	requestCounter RequestCounterMiddleware
	workerPool     WorkerPool
}

func New(config StarFleetConfig) *StarFleet {
	return &StarFleet{
		middleware:     NewMiddleware(config.Middleware),
		requestCounter: NewRequestCounterMiddleware(),
		workerPool:     NewWorkerPool(config.Workers),
	}
}

func (sf *StarFleet) Run() {
	sf.workerPool.Run()

	http.HandleFunc("/dashboard", sf.handleDashboard)
	http.HandleFunc("/dashboard-stats", sf.handleDashboardStats)
	http.HandleFunc("/dashboard-request-counter", sf.handleDashboardRequestCounter)
	http.HandleFunc("/dashboard-revive/", sf.handleDashboardRevive)

	http.HandleFunc("/generate", sf.middleware.Middleware(sf.requestCounter.Middleware(sf.handleGenerate)))
	http.HandleFunc("/queue", sf.handleQueue)

	log.Info().Msg("Listening on port :8080")
	log.Fatal().Err(http.ListenAndServe(":8080", nil)).Msg("fatal error has occurred on port :8080")
}