Device reconnection handling #15
+7
-42
@@ -2,7 +2,6 @@ package services
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
|
|
||||||
@@ -24,7 +23,6 @@ type DeviceService struct {
|
|||||||
// Output
|
// Output
|
||||||
Messages chan string
|
Messages chan string
|
||||||
// Callbacks
|
// Callbacks
|
||||||
OnPeripheralFound func(string)
|
|
||||||
// Telemetry service data fetchers
|
// Telemetry service data fetchers
|
||||||
telemetryProvider func() (string, error)
|
telemetryProvider func() (string, error)
|
||||||
}
|
}
|
||||||
@@ -44,12 +42,18 @@ func NewDeviceService(logger *slog.Logger, msg chan string) *DeviceService {
|
|||||||
go dev.FindDevices()
|
go dev.FindDevices()
|
||||||
|
|
||||||
// Set the callbacks for PSS
|
// Set the callbacks for PSS
|
||||||
dev.PSS.OnDeviceFound = dev.deviceFound
|
dev.PSS.telemetryProvider = dev.getTelemetryProvider
|
||||||
|
|
||||||
return dev
|
return dev
|
||||||
}
|
}
|
||||||
|
|
||||||
// Getters [START] -------------------------------------------------------------
|
// Getters [START] -------------------------------------------------------------
|
||||||
|
// This function is currently only being used by PSS, we may have to find a better
|
||||||
|
// pattern for this
|
||||||
|
func (ds *DeviceService) getTelemetryProvider() (string, error) {
|
||||||
|
return ds.telemetryProvider()
|
||||||
|
}
|
||||||
|
|
||||||
func (ds *DeviceService) GetDevices() []peripheral.Peripheral {
|
func (ds *DeviceService) GetDevices() []peripheral.Peripheral {
|
||||||
snapshot := ds.PSS.GetStates()
|
snapshot := ds.PSS.GetStates()
|
||||||
peripherals := make([]peripheral.Peripheral, 0, len(snapshot))
|
peripherals := make([]peripheral.Peripheral, 0, len(snapshot))
|
||||||
@@ -138,45 +142,6 @@ func (ds *DeviceService) transmit(ctx context.Context) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ds *DeviceService) deviceFound(pname string) {
|
|
||||||
ds.OnPeripheralFound(pname)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Callbacks [START] -----------------------------------------------------------
|
// Callbacks [START] -----------------------------------------------------------
|
||||||
|
|
||||||
// ProviderFoundCallback should be called once the telemetry service finds a provider
|
|
||||||
// Here we need to setup our devices. Some devices might have different settings for
|
|
||||||
// different sims
|
|
||||||
func (ds *DeviceService) ProviderFoundCallback(name string) {
|
|
||||||
ds.Messages <- "Device services got triggered by a provider being found\n"
|
|
||||||
for _, peripheral := range ds.PSS.GetStates() {
|
|
||||||
// ds.Messages <- fmt.Sprintf("dev: %s, SETUP: %t, STATE: %d\n",
|
|
||||||
// peripheral.device.Name, peripheral.Setup, peripheral.State)
|
|
||||||
if peripheral.State != DeviceIsConnected || peripheral.Setup {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
// The device is connected and still needs to run the setup
|
|
||||||
ds.Messages <- fmt.Sprintf("device '%s' needs to be setup\n", peripheral.device.Name)
|
|
||||||
|
|
||||||
provider, err := ds.telemetryProvider()
|
|
||||||
if err != nil {
|
|
||||||
// Can't setup anything
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
err = peripheral.Peripheral.Setup(provider)
|
|
||||||
if err != nil {
|
|
||||||
// TODO: log do something
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
err = ds.PSS.UpdatePeripheralSetupState(peripheral.device.Name, true)
|
|
||||||
if err != nil {
|
|
||||||
// TODO: log do something
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Callbacks [END] -------------------------------------------------------------
|
// Callbacks [END] -------------------------------------------------------------
|
||||||
|
|||||||
@@ -57,7 +57,7 @@ type PeripheralStateStore struct {
|
|||||||
// Messaging for UI and stuff
|
// Messaging for UI and stuff
|
||||||
Messages chan string
|
Messages chan string
|
||||||
// Callbacks
|
// Callbacks
|
||||||
OnDeviceFound func(string)
|
telemetryProvider func() (string, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewPeripheralStateStore(
|
func NewPeripheralStateStore(
|
||||||
@@ -177,7 +177,7 @@ func (pss *PeripheralStateStore) setDeviceConnected(pname string, per peripheral
|
|||||||
pss.mu.Unlock()
|
pss.mu.Unlock()
|
||||||
|
|
||||||
pss.Messages <- fmt.Sprintf("Device successfuly connected: %s\n", pname)
|
pss.Messages <- fmt.Sprintf("Device successfuly connected: %s\n", pname)
|
||||||
pss.OnDeviceFound(pname)
|
// pss.OnDeviceFound(pname)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (pss *PeripheralStateStore) setDeviceTimedOut(pname string) {
|
func (pss *PeripheralStateStore) setDeviceTimedOut(pname string) {
|
||||||
@@ -239,20 +239,49 @@ func (pss *PeripheralStateStore) handleDeviceReconnected(pname string) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
if err != nil {
|
// Update the peripheral state
|
||||||
pss.Logger.Error("failed to setup peripheral", "peripheral", pname, "error", err)
|
|
||||||
return ErrFailedToSetupPeripheral
|
|
||||||
}
|
|
||||||
|
|
||||||
// Around here I believe I need to swap the states so the peripheral is setup
|
|
||||||
pss.setDeviceConnected(pname, state.Peripheral)
|
pss.setDeviceConnected(pname, state.Peripheral)
|
||||||
|
pss.UpdatePeripheralSetupState(pname, false)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// handleDeviceConnected will handle the device setup after it connects
|
// handleDeviceConnected will handle the device setup after it connects
|
||||||
|
// NOTE: should this be a state after Connected?
|
||||||
|
// Connected -> Unconfigured -> Configured I believe this would work nicely
|
||||||
|
// THIS IS A TODO ↑↑↑↑↑↑
|
||||||
func (pss *PeripheralStateStore) handleDeviceConnected(pname string) error {
|
func (pss *PeripheralStateStore) handleDeviceConnected(pname string) error {
|
||||||
// We need to query wheter we have a telemetry provider running or not
|
// Things to do once the device is connected
|
||||||
|
// 1. Setup
|
||||||
|
state, err := pss.GetState(pname)
|
||||||
|
if err != nil {
|
||||||
|
// We need to log something here or something
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !state.Setup {
|
||||||
|
err = pss.setupPeripheral(state)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pss *PeripheralStateStore) setupPeripheral(state *PeripheralState) error {
|
||||||
|
provider, err := pss.telemetryProvider()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = state.Peripheral.Setup(provider)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = pss.UpdatePeripheralSetupState(state.device.Name, true)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -25,11 +25,9 @@ func NewOrchestrator(logger *slog.Logger) (*Orchestrator, error) {
|
|||||||
go telemService.FindProvider(telemService.CtxMonitor)
|
go telemService.FindProvider(telemService.CtxMonitor)
|
||||||
|
|
||||||
// Setup device service callbacks
|
// Setup device service callbacks
|
||||||
devService.OnPeripheralFound = telemService.PeripheralFoundCallback
|
|
||||||
devService.telemetryProvider = telemService.GetTelemetryProviderName
|
devService.telemetryProvider = telemService.GetTelemetryProviderName
|
||||||
|
|
||||||
// Setup telemetry service callbacks
|
// Setup telemetry service callbacks
|
||||||
telemService.OnProviderFound = devService.ProviderFoundCallback
|
|
||||||
telemService.peripheralProvider = devService.GetDevices
|
telemService.peripheralProvider = devService.GetDevices
|
||||||
|
|
||||||
return &Orchestrator{
|
return &Orchestrator{
|
||||||
|
|||||||
@@ -36,7 +36,6 @@ type TelemetryService struct {
|
|||||||
CtxHealthcheck context.Context
|
CtxHealthcheck context.Context
|
||||||
healthCheckCancel context.CancelFunc
|
healthCheckCancel context.CancelFunc
|
||||||
// Callbacks
|
// Callbacks
|
||||||
OnProviderFound func(string)
|
|
||||||
// Devices data request
|
// Devices data request
|
||||||
peripheralProvider func() []peripheral.Peripheral
|
peripheralProvider func() []peripheral.Peripheral
|
||||||
}
|
}
|
||||||
@@ -277,8 +276,4 @@ func (t *TelemetryService) IsStreaming() bool {
|
|||||||
|
|
||||||
// Callbacks [START] -----------------------------------------------------------
|
// Callbacks [START] -----------------------------------------------------------
|
||||||
|
|
||||||
func (t *TelemetryService) PeripheralFoundCallback(pname string) {
|
|
||||||
t.Messages <- "Telemetry service callback for peripheral found called\n"
|
|
||||||
}
|
|
||||||
|
|
||||||
// Callbacks [END] -------------------------------------------------------------
|
// Callbacks [END] -------------------------------------------------------------
|
||||||
|
|||||||
@@ -26,9 +26,6 @@ func (t *TelemetryService) onFindProvider(prov telem.TelemetryProvider) {
|
|||||||
// Start the healthcheck on our provider so we can drop it if it stops
|
// Start the healthcheck on our provider so we can drop it if it stops
|
||||||
t.CtxHealthcheck, t.healthCheckCancel = context.WithCancel(context.Background())
|
t.CtxHealthcheck, t.healthCheckCancel = context.WithCancel(context.Background())
|
||||||
go t.ProviderMonitor(t.CtxHealthcheck)
|
go t.ProviderMonitor(t.CtxHealthcheck)
|
||||||
|
|
||||||
// Tell the devices service we got a provider
|
|
||||||
t.OnProviderFound(prov.Name())
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *TelemetryService) onProviderStopsMidStream() {
|
func (t *TelemetryService) onProviderStopsMidStream() {
|
||||||
|
|||||||
Reference in New Issue
Block a user