wip: grpc api
Signed-off-by: Evan Hazlett <ejhazlett@gmail.com>
This commit is contained in:
parent
1dd748e3f2
commit
2eba8d6511
174 changed files with 22012 additions and 11410 deletions
49
datastore/memory/service.go
Normal file
49
datastore/memory/service.go
Normal file
|
@ -0,0 +1,49 @@
|
|||
package memory
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/ehazlett/element/api/types"
|
||||
)
|
||||
|
||||
func (m *Memory) SaveService(service *types.Service) error {
|
||||
m.m.Lock()
|
||||
m.service[service.ID] = service
|
||||
m.m.Unlock()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Memory) DeleteService(id string) error {
|
||||
m.m.Lock()
|
||||
if _, exists := m.service[id]; exists {
|
||||
delete(m.service, id)
|
||||
}
|
||||
m.m.Unlock()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Memory) GetServices() ([]*types.Service, error) {
|
||||
s := []*types.Service{}
|
||||
for _, v := range m.service {
|
||||
s = append(s, v)
|
||||
}
|
||||
|
||||
return s, nil
|
||||
}
|
||||
|
||||
func (m *Memory) GetServiceByHost(host string) (*types.Service, error) {
|
||||
services, err := m.GetServices()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for _, service := range services {
|
||||
for _, h := range service.Hosts {
|
||||
if strings.Index(h, host) > -1 {
|
||||
return service, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil, nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue