57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
|
package directory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
var serviceDirectory sync.Map
|
||
|
|
|
||
|
|
func GetServiceInstance(id string) *ServiceInstance {
|
||
|
|
val, ok := serviceDirectory.Load(id)
|
||
|
|
if ok {
|
||
|
|
return val.(*ServiceInstance)
|
||
|
|
} else {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetServiceInstanceByType(t string) *ServiceInstance {
|
||
|
|
var result *ServiceInstance
|
||
|
|
serviceDirectory.Range(func(key, value any) bool {
|
||
|
|
if value.(*ServiceInstance).Type == t {
|
||
|
|
result = value.(*ServiceInstance)
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func ListServiceInstance() []*ServiceInstance {
|
||
|
|
var result []*ServiceInstance
|
||
|
|
serviceDirectory.Range(func(key, value interface{}) bool {
|
||
|
|
result = append(result, value.(*ServiceInstance))
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func ListServiceInstanceByType(t string) []*ServiceInstance {
|
||
|
|
var result []*ServiceInstance
|
||
|
|
serviceDirectory.Range(func(key, value interface{}) bool {
|
||
|
|
if value.(*ServiceInstance).Type == t {
|
||
|
|
result = append(result, value.(*ServiceInstance))
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func AddServiceInstance(in *ServiceInstance) {
|
||
|
|
serviceDirectory.Store(in.ID, in)
|
||
|
|
}
|
||
|
|
|
||
|
|
func RemoveServiceInstance(id string) {
|
||
|
|
serviceDirectory.Delete(id)
|
||
|
|
}
|