You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
130 lines
3.7 KiB
130 lines
3.7 KiB
package main
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/rs/cors"
|
|
"github.com/Cyclops-Labs/cyclops-4-hpc.git/services/eventsengine/models"
|
|
"github.com/Cyclops-Labs/cyclops-4-hpc.git/services/eventsengine/restapi"
|
|
"github.com/Cyclops-Labs/cyclops-4-hpc.git/services/eventsengine/server/dbManager"
|
|
"github.com/Cyclops-Labs/cyclops-4-hpc.git/services/eventsengine/server/eventManager"
|
|
"github.com/Cyclops-Labs/cyclops-4-hpc.git/services/eventsengine/server/statusManager"
|
|
"github.com/Cyclops-Labs/cyclops-4-hpc.git/services/eventsengine/server/usageManager"
|
|
l "gitlab.com/cyclops-utilities/logging"
|
|
)
|
|
|
|
// getService is the uService instantiation function. A sample of the elements
|
|
// that need to be personalized for a new uService are provided.
|
|
// Returns:
|
|
// - hadler: return the http.handler with the swagger and (optionally) the CORS
|
|
// configured according to the config provided
|
|
// - e: in case of any error it's propagated to the invoker
|
|
func getService() (handler http.Handler, register *prometheus.Registry, e error) {
|
|
|
|
l.Trace.Printf("[MAIN] Intializing service [ %v ] handler\n", strings.Title(service))
|
|
|
|
// TABLE0,...,N have to be customized
|
|
db := dbStart(&models.Event{}, &models.State{})
|
|
mon := statusManager.New(db)
|
|
|
|
// Prometheus Metrics linked to dbParameter
|
|
db.Metrics, register = prometheusStart()
|
|
|
|
// In case of needed, here is where kafka support is started
|
|
// The functions needs to be customized accordingly to the needs of the service
|
|
// And the parts of the service that might need to send messages need to get
|
|
// the channel
|
|
_ = kafkaStart(db, mon)
|
|
|
|
//bp := getBasePath()
|
|
|
|
// Parts of the service HERE
|
|
v := eventManager.New(db, mon, cfg.Events.Filters)
|
|
u := usageManager.New(db, mon)
|
|
|
|
// Initiate the http handler, with the objects that are implementing the business logic.
|
|
h, e := restapi.Handler(restapi.Config{
|
|
StatusManagementAPI: mon,
|
|
EventManagementAPI: v,
|
|
UsageManagementAPI: u,
|
|
Logger: l.Info.Printf,
|
|
AuthKeycloak: AuthKeycloak,
|
|
AuthAPIKeyHeader: AuthAPIKey,
|
|
AuthAPIKeyParam: AuthAPIKey,
|
|
})
|
|
|
|
if e != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
handler = h
|
|
|
|
// CORS
|
|
if cfg.General.CORSEnabled {
|
|
|
|
l.Trace.Printf("[MAIN] Enabling CORS for the service [ %v ]\n", strings.Title(service))
|
|
|
|
handler = cors.New(
|
|
cors.Options{
|
|
Debug: (cfg.General.LogLevel == "DEBUG") || (cfg.General.LogLevel == "TRACE"),
|
|
AllowedOrigins: cfg.General.CORSOrigins,
|
|
AllowedHeaders: cfg.General.CORSHeaders,
|
|
AllowedMethods: cfg.General.CORSMethods,
|
|
}).Handler(h)
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
// kafkaStart handles the initialization of the kafka service.
|
|
// This is a sample function with the most basic usage of the kafka service, it
|
|
// should be redefined to match the needs of the service.
|
|
// Parameters:
|
|
// - db: DbPAramenter reference for interaction with the db.
|
|
// - mon: statusManager parameter reference to interact with the statusManager
|
|
// subsystem
|
|
// Returns:
|
|
// - ch: a interface{} channel to be able to send thing through the kafka topic generated.
|
|
func kafkaStart(db *dbManager.DbParameter, mon *statusManager.StatusManager) (ch chan interface{}) {
|
|
|
|
l.Trace.Printf("[MAIN] Intializing Kafka\n")
|
|
|
|
f := func(db *dbManager.DbParameter, model interface{}) (e error) {
|
|
|
|
if e = db.AddEvent(*model.(*models.Event)); e != nil {
|
|
|
|
l.Warning.Printf("[KAFKA] There was an error while apliying the linked function. Error: %v\n", e)
|
|
|
|
} else {
|
|
|
|
l.Trace.Printf("[KAFKA] Aplication of the linked function done succesfully.\n")
|
|
|
|
}
|
|
return
|
|
|
|
}
|
|
|
|
ch = make(chan interface{})
|
|
|
|
handler := kafkaHandlerConf{
|
|
in: []kafkaPackage{
|
|
{
|
|
topic: cfg.Kafka.TopicsIn[0],
|
|
model: models.Event{},
|
|
function: f,
|
|
saveDB: false,
|
|
},
|
|
},
|
|
}
|
|
|
|
kafkaHandler(db, mon, handler)
|
|
|
|
return
|
|
|
|
}
|
|
|