handler.go

package handler

import (
	"net/http"
	"time"

	"github.com/USACE/instrumentation-api/api/internal/cloud"
	"github.com/USACE/instrumentation-api/api/internal/config"
	"github.com/USACE/instrumentation-api/api/internal/middleware"
	"github.com/USACE/instrumentation-api/api/internal/service"
)

func newHttpClient() *http.Client {
	return &http.Client{
		Timeout: time.Second * 60,
		CheckRedirect: func(req *http.Request, via []*http.Request) error {
			return nil
		},
	}
}

type ApiHandler struct {
	DBService     *service.DBService
	Middleware    middleware.Middleware
	BlobService   cloud.Blob
	PubsubService cloud.Pubsub
	Config        *config.ApiConfig
}

func NewApi(cfg *config.ApiConfig) *ApiHandler {
	ds := service.NewDBService(cfg.DBConfig)

	return &ApiHandler{
		DBService:     ds,
		Middleware:    middleware.NewMiddleware(&cfg.ServerConfig, ds),
		BlobService:   cloud.NewS3Blob(&cfg.AWSS3Config, "", cfg.RoutePrefix),
		PubsubService: cloud.NewSQSPubsub(&cfg.AWSSQSConfig),
		Config:        cfg,
	}
}

type TelemetryHandler struct {
	DBService  *service.DBService
	Middleware middleware.Middleware
}

func NewTelemetry(cfg *config.TelemetryConfig) *TelemetryHandler {
	ds := service.NewDBService(cfg.DBConfig)

	return &TelemetryHandler{
		DBService:  ds,
		Middleware: middleware.NewMiddleware(&cfg.ServerConfig, ds),
	}
}

type AlertCheckHandler struct {
	DBService *service.DBService
	Config    *config.AlertCheckConfig
}

func NewAlertCheck(cfg *config.AlertCheckConfig) *AlertCheckHandler {
	return &AlertCheckHandler{
		DBService: service.NewDBService(cfg.DBConfig),
		Config:    cfg,
	}
}

type DcsLoaderHandler struct {
	PubsubService    cloud.Pubsub
	DcsLoaderService service.DcsLoaderService
}

func NewDcsLoader(cfg *config.DcsLoaderConfig) *DcsLoaderHandler {
	s3Blob := cloud.NewS3Blob(&cfg.AWSS3Config, "", "")
	ps := cloud.NewSQSPubsub(&cfg.AWSSQSConfig).WithBlob(s3Blob)
	apiClient := newHttpClient()

	return &DcsLoaderHandler{
		PubsubService:    ps,
		DcsLoaderService: service.NewDcsLoaderService(apiClient, cfg),
	}
}