add register data mechanism

This commit is contained in:
Pedro Pérez 2025-10-09 23:18:32 +02:00
parent b012db856c
commit fcc0d06f96
3 changed files with 81 additions and 3 deletions

View File

@ -15,6 +15,7 @@ const (
subjectSensorsGet = "sensors.get"
subjectSensorsValuesGet = "sensors.values.get"
subjectSensorsList = "sensors.list"
subjectSensorsData = "sensors.data."
)
type Handlers struct {
@ -48,8 +49,23 @@ func handleRequest[Req any, Res any](msg *nats.Msg, handler func(Req) (Res, erro
msg.Respond(response)
}
func handlePublish[Req any](msg *nats.Msg, handler func(Req) error) {
var req Req
if len(msg.Data) > 0 {
if err := json.Unmarshal(msg.Data, &req); err != nil {
slog.Error("failed to unmarshal message", "error", err)
return
}
}
if err := handler(req); err != nil {
slog.Error("handler error", "error", err)
}
}
func (h *Handlers) SetupEndpoints() *Handlers {
h.register()
h.registerData()
h.update()
h.get()
h.getValues()
@ -74,6 +90,25 @@ func (h *Handlers) register() {
})
}
func (h *Handlers) registerData() {
h.NATS.Subscribe(subjectSensorsData+"*", func(msg *nats.Msg) {
handlePublish(msg, func(data SensorData) error {
if err := data.Validate(); err != nil {
slog.Error("error validating sensor data", "error", err)
return err
}
if err := h.service.RegisterSensorData(data); err != nil {
slog.Error("failed to save sensor data", "error", err, "sensor_id", data.SensorID)
return err
}
slog.Debug("sensor data saved", "sensor_id", data.SensorID, "value", data.Value)
return nil
})
})
}
func (h *Handlers) update() {
h.NATS.Subscribe(subjectSensorsUpdate, func(msg *nats.Msg) {
handleRequest(msg, func(req Sensor) (Sensor, error) {

View File

@ -35,8 +35,11 @@ func (p *pgxRepo) CreateSensor(s Sensor) error {
return err
}
const createSensorDataQuery = `insert into registry (sensor_id, value, created_at) values ($1, $2, $3)`
func (p *pgxRepo) CreateSensorData(s SensorData) error {
return nil
_, err := p.Exec(context.Background(), createSensorDataQuery, s.SensorID, s.Value, s.Timestamp)
return err
}
func (p *pgxRepo) UpdateSensor(s Sensor) error {
@ -106,7 +109,32 @@ func (i *inMemory) CreateSensor(s Sensor) error {
return nil
}
func (p *inMemory) CreateSensorData(s SensorData) error {
func (i *inMemory) CreateSensorData(data SensorData) error {
i.mu.Lock()
defer i.mu.Unlock()
sensor, exists := i.sensors[data.SensorID]
if !exists {
return ErrSensorNotFound
}
if sensor.SensorData == nil {
sensor.SensorData = &map[int]SensorData{}
}
key := int(data.Timestamp.Unix())
(*sensor.SensorData)[key] = data
if len(*sensor.SensorData) > 100 {
oldestKey := key
for k := range *sensor.SensorData {
if k < oldestKey {
oldestKey = k
}
}
delete(*sensor.SensorData, oldestKey)
}
return nil
}
@ -169,7 +197,12 @@ func (d *DecoratorRepo) CreateSensor(s Sensor) error {
return nil
}
func (p *DecoratorRepo) CreateSensorData(s SensorData) error {
func (d *DecoratorRepo) CreateSensorData(data SensorData) error {
if err := d.db.CreateSensorData(data); err != nil {
return err
}
_ = d.memory.CreateSensorData(data)
return nil
}

View File

@ -26,6 +26,16 @@ func (s *Service) RegisterSensor(sensor Sensor) error {
return nil
}
func (s *Service) RegisterSensorData(data SensorData) error {
err := s.repo.CreateSensorData(data)
if err != nil {
slog.Error("error registering sensor data")
return err
}
return nil
}
func (s *Service) UpdateSensor(sensor Sensor) error {
s.repo.UpdateSensor(sensor)