135 lines
3.8 KiB
Go
135 lines
3.8 KiB
Go
package infrastructure
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"oc-scheduler/conf"
|
|
|
|
oclib "cloud.o-forge.io/core/oc-lib"
|
|
"cloud.o-forge.io/core/oc-lib/models/booking"
|
|
"cloud.o-forge.io/core/oc-lib/models/resources/purchase_resource"
|
|
"cloud.o-forge.io/core/oc-lib/models/utils"
|
|
"cloud.o-forge.io/core/oc-lib/tools"
|
|
)
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// NATS emission
|
|
// ---------------------------------------------------------------------------
|
|
|
|
func EmitNATS(peerID string, message tools.PropalgationMessage) {
|
|
// PB_CLOSE_PLANNER: notify local watchers so streams re-evaluate.
|
|
// Cache mutations (eviction or ownership reset) are the caller's
|
|
// responsibility — see evictAfter and ReleaseRefreshOwnership.
|
|
if message.Action == tools.PB_CLOSE_PLANNER {
|
|
notifyPlannerWatchers(peerID)
|
|
}
|
|
b, _ := json.Marshal(message)
|
|
tools.NewNATSCaller().SetNATSPub(tools.PROPALGATION_EVENT, tools.NATSResponse{
|
|
FromApp: "oc-scheduler",
|
|
Datatype: -1,
|
|
Method: int(tools.PROPALGATION_EVENT),
|
|
Payload: b,
|
|
})
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// NATS listeners
|
|
// ---------------------------------------------------------------------------
|
|
|
|
func ListenNATS() {
|
|
tools.NewNATSCaller().ListenNats(map[tools.NATSMethod]func(tools.NATSResponse){
|
|
tools.PLANNER_EXECUTION: handlePlannerExecution,
|
|
tools.PROPALGATION_EVENT: handlePropagationEvent,
|
|
tools.REMOVE_RESOURCE: handleRemoveResource,
|
|
tools.CREATE_RESOURCE: handleCreateResource,
|
|
tools.CONFIRM_EVENT: handleConfirm,
|
|
})
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Draft timeout
|
|
// ---------------------------------------------------------------------------
|
|
|
|
// draftTimeout deletes a booking or purchase resource if it is still a draft
|
|
// after the 10-minute confirmation window has elapsed.
|
|
func draftTimeout(id string, dt tools.DataType) {
|
|
adminReq := &tools.APIRequest{Admin: true}
|
|
var res utils.DBObject
|
|
var loadErr error
|
|
switch dt {
|
|
case tools.BOOKING:
|
|
res, _, loadErr = booking.NewAccessor(adminReq).LoadOne(id)
|
|
case tools.PURCHASE_RESOURCE:
|
|
res, _, loadErr = purchase_resource.NewAccessor(adminReq).LoadOne(id)
|
|
default:
|
|
return
|
|
}
|
|
if loadErr != nil || res == nil || !res.IsDrafted() {
|
|
return
|
|
}
|
|
switch dt {
|
|
case tools.BOOKING:
|
|
booking.NewAccessor(adminReq).DeleteOne(id)
|
|
case tools.PURCHASE_RESOURCE:
|
|
purchase_resource.NewAccessor(adminReq).DeleteOne(id)
|
|
}
|
|
fmt.Printf("draftTimeout: %s %s deleted (still draft after 10 min)\n", dt.String(), id)
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Kubernetes namespace helper
|
|
// ---------------------------------------------------------------------------
|
|
|
|
func createNamespace(ns string) error {
|
|
/*
|
|
* This function is used to create a namespace.
|
|
* It takes the following parameters:
|
|
* - ns: the namespace you want to create
|
|
*/
|
|
logger := oclib.GetLogger()
|
|
serv, err := tools.NewKubernetesService(
|
|
conf.GetConfig().KubeHost+":"+conf.GetConfig().KubePort, conf.GetConfig().KubeCA,
|
|
conf.GetConfig().KubeCert, conf.GetConfig().KubeData)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
c := context.Background()
|
|
|
|
ok, err := serv.GetNamespace(c, ns)
|
|
if ok != nil && err == nil {
|
|
logger.Debug().Msg("A namespace with name " + ns + " already exists")
|
|
return nil
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = serv.CreateNamespace(c, ns)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = serv.CreateServiceAccount(c, ns)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
role := "argo-role"
|
|
err = serv.CreateRole(c, ns, role,
|
|
[][]string{
|
|
{"coordination.k8s.io"},
|
|
{""},
|
|
{""}},
|
|
[][]string{
|
|
{"leases"},
|
|
{"secrets"},
|
|
{"pods"}},
|
|
[][]string{
|
|
{"get", "create", "update"},
|
|
{"get"},
|
|
{"patch"}})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return serv.CreateRoleBinding(c, ns, "argo-role-binding", role)
|
|
}
|