projects/weather-api
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

optimized mongodb

Browse source
This commit is contained in:
Joel Schmid 2021-04-25 00:31:27 +02:00
parent 442f205a2b
commit de11211de1
3 changed files with 25 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
storage/inmemory-sensor-registry.go
View file

@ -48,9 +48,9 @@ func (registry *inmemorySensorRegistry) ResolveSensorById(sensorId uuid.UUID) (*
return nil, errors.New("sensor does not exist") return nil, errors.New("sensor does not exist")
} }
func (registry *inmemorySensorRegistry) ExistSensor(sensor *WeatherSensor) (bool, error) { func (registry *inmemorySensorRegistry) ExistSensor(sensorId uuid.UUID) (bool, error) {
for _, s := range registry.weatherSensors { for _, s := range registry.weatherSensors {
if s.Id == sensor.Id { if s.Id == sensorId {
return true, nil return true, nil
} }
} }

42
storage/mongodb-storage.go
View file

@ -72,45 +72,45 @@ func (registry *mongodbSensorRegistry) RegisterSensorByName(name string) (*Weath
} }
func (registry *mongodbSensorRegistry) ExistSensorName(name string) (bool, error) { func (registry *mongodbSensorRegistry) ExistSensorName(name string) (bool, error) {
sensors, err := registry.GetSensors() ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
cursor, err := registry.sensorCollection.Find(ctx, bson.M{"name": name})
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
return false, err return false, err
} }
for _, s := range sensors {
if s.Name == name { return cursor.Next(ctx), nil
return true, nil
}
}
return false, nil
} }
func (registry *mongodbSensorRegistry) ResolveSensorById(sensorId uuid.UUID) (*WeatherSensor, error) { func (registry *mongodbSensorRegistry) ResolveSensorById(sensorId uuid.UUID) (*WeatherSensor, error) {
sensors, err := registry.GetSensors() ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
cursor, err := registry.sensorCollection.Find(ctx, bson.M{"id": sensorId})
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
return nil, err return nil, err
} }
for _, s := range sensors {
if s.Id == sensorId { if !cursor.Next(ctx) {
return s, nil
}
}
return nil, errors.New("sensor does not exist") return nil, errors.New("sensor does not exist")
} }
func (registry *mongodbSensorRegistry) ExistSensor(sensor *WeatherSensor) (bool, error) { var sensor *WeatherSensor
sensors, err := registry.GetSensors() if err = cursor.Decode(&sensor); err != nil {
log.Fatal(err)
return nil, err
}
return sensor, nil
}
func (registry *mongodbSensorRegistry) ExistSensor(sensorId uuid.UUID) (bool, error) {
ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
cursor, err := registry.sensorCollection.Find(ctx, bson.M{"id": sensorId})
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
return false, err return false, err
} }
for _, s := range sensors {
if s.Id == sensor.Id { return cursor.Next(ctx), nil
return true, nil
}
}
return false, nil
} }
func (registry *mongodbSensorRegistry) GetSensors() ([]*WeatherSensor, error) { func (registry *mongodbSensorRegistry) GetSensors() ([]*WeatherSensor, error) {

3
storage/sensor-registry.go
View file

@ -4,7 +4,8 @@ import "github.com/google/uuid"
type SensorRegistry interface { type SensorRegistry interface {
RegisterSensorByName(string) (*WeatherSensor, error) RegisterSensorByName(string) (*WeatherSensor, error)
ExistSensor(*WeatherSensor) (bool, error) ExistSensor(sensorId uuid.UUID) (bool, error)
ExistSensorName(name string) (bool, error)
ResolveSensorById(uuid.UUID) (*WeatherSensor, error) ResolveSensorById(uuid.UUID) (*WeatherSensor, error)
DeleteSensor(uuid.UUID) error DeleteSensor(uuid.UUID) error
UpdateSensor(*WeatherSensor) error UpdateSensor(*WeatherSensor) error