You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
754 B
Go
39 lines
754 B
Go
package storage
|
|
|
|
import "errors"
|
|
|
|
// implements Storage interface
|
|
type InMemoryStorage struct{}
|
|
|
|
var memory = map[string]*Device{}
|
|
|
|
func NewMemoryStorage() *InMemoryStorage {
|
|
return &InMemoryStorage{}
|
|
}
|
|
|
|
func (s *InMemoryStorage) Save(device *Device) error {
|
|
memory[device.Name] = device
|
|
return nil
|
|
}
|
|
|
|
func (s *InMemoryStorage) List() ([]*Device, error) {
|
|
devices := []*Device{}
|
|
for _, device := range memory {
|
|
devices = append(devices, device)
|
|
}
|
|
return devices, nil
|
|
}
|
|
|
|
func (s *InMemoryStorage) Get(name string) (*Device, error) {
|
|
device, ok := memory[name]
|
|
if !ok {
|
|
return nil, errors.New("device doesn't exist")
|
|
}
|
|
return device, nil
|
|
}
|
|
|
|
func (s *InMemoryStorage) Delete(device *Device) error {
|
|
delete(memory, device.Name)
|
|
return nil
|
|
}
|