86 lines
1.8 KiB
Go
86 lines
1.8 KiB
Go
package cache
|
|
|
|
import (
|
|
"jannex/robot-control-manager/modules/structs"
|
|
"jannex/robot-control-manager/modules/utils"
|
|
"sort"
|
|
"sync"
|
|
|
|
"git.ex.umbach.dev/Alex/roese-utils/rspagination"
|
|
)
|
|
|
|
// list of robots that are connected for the first time
|
|
// to this server and are not yet in the database (authorized)
|
|
var unauthorizedRobots = make(map[string]*structs.UnauthorizedRobot)
|
|
var urMu sync.RWMutex
|
|
|
|
func AddUnauthorizedRobot(robot *structs.UnauthorizedRobot) {
|
|
urMu.Lock()
|
|
defer urMu.Unlock()
|
|
|
|
unauthorizedRobots[robot.Id] = robot
|
|
}
|
|
|
|
func GetUnauthorizedRobots() map[string]*structs.UnauthorizedRobot {
|
|
urMu.RLock()
|
|
defer urMu.RUnlock()
|
|
|
|
return unauthorizedRobots
|
|
}
|
|
|
|
func GetAllUnauthorizedRobots(query rspagination.PageQuery) structs.UnauthorizedRobotsResponse {
|
|
urMu.RLock()
|
|
defer urMu.RUnlock()
|
|
|
|
var r []structs.UnauthorizedRobot
|
|
|
|
for _, v := range unauthorizedRobots {
|
|
r = append(r, *v)
|
|
}
|
|
|
|
// sort after connected at
|
|
sort.SliceStable(r, func(i, j int) bool {
|
|
return r[i].ConnectedAt.After(r[j].ConnectedAt)
|
|
})
|
|
|
|
start, end := rspagination.GetPage(len(r), query.Page, utils.UnauthorizedRobotsPageLimit)
|
|
|
|
return structs.UnauthorizedRobotsResponse{
|
|
UnauthorizedRobots: r[start:end],
|
|
TotalPages: rspagination.CalculateTotalPages(len(r), utils.UnauthorizedRobotsPageLimit),
|
|
}
|
|
}
|
|
|
|
func IsUnauthorizedRobotInList(robotId string) bool {
|
|
urMu.RLock()
|
|
defer urMu.RUnlock()
|
|
|
|
for _, r := range unauthorizedRobots {
|
|
if r.Id == robotId {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func GetUnauthorizedRobotById(robotId string) structs.UnauthorizedRobot {
|
|
urMu.RLock()
|
|
defer urMu.RUnlock()
|
|
|
|
for _, r := range unauthorizedRobots {
|
|
if r.Id == robotId {
|
|
return *r
|
|
}
|
|
}
|
|
|
|
return structs.UnauthorizedRobot{}
|
|
}
|
|
|
|
func RemoveUnauthorizedRobotById(robotId string) {
|
|
urMu.Lock()
|
|
defer urMu.Unlock()
|
|
|
|
delete(unauthorizedRobots, robotId)
|
|
}
|