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.
150 lines
3.7 KiB
Go
150 lines
3.7 KiB
Go
package system
|
|
|
|
import (
|
|
"FRMS/internal/pkg/manager"
|
|
pb "dmac/pkg/grpc"
|
|
"dmac/pkg/logging"
|
|
"time"
|
|
|
|
//"FRMS/internal/pkg/device"
|
|
"context"
|
|
"fmt"
|
|
_ "log"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// manager stuff
|
|
|
|
type Manager interface {
|
|
Start() error // status checks
|
|
Exit() error
|
|
Timeout() (time.Duration, error) // TO Generator
|
|
}
|
|
|
|
func NewManager(max int) Manager {
|
|
// takes a heartbeat and max connection attempts
|
|
return manager.New(max)
|
|
}
|
|
|
|
type ReactorManager struct {
|
|
Manager // base manager interface
|
|
// *ClientManager // client manager (OUTDATED)
|
|
*Client // access to ID etc
|
|
// StatusMon *StatusMonitor putting on pause
|
|
// *ReactorDevices
|
|
Config *viper.Viper // config to update
|
|
Err chan error
|
|
}
|
|
|
|
// type ReactorDevices struct {
|
|
// // device struct
|
|
// Devices map[int]DeviceManager
|
|
// sync.RWMutex
|
|
// }
|
|
|
|
func NewReactorManager(cl *Client, config *viper.Viper, errCh chan error) *ReactorManager {
|
|
// making managers
|
|
m := NewManager(6)
|
|
r := &ReactorManager{
|
|
Manager: m,
|
|
Client: cl,
|
|
Config: config,
|
|
Err: errCh,
|
|
}
|
|
return r
|
|
}
|
|
|
|
func (r *ReactorManager) Start() error {
|
|
// allows for extra stuff
|
|
logging.Debug(logging.DStart, "RMA %v starting", r.Id)
|
|
return r.Manager.Start()
|
|
//go r.StatusMon.Send(&DeviceInfo{Id: r.Id, Type: "Reactor", Status: "[green]ONLINE[white]"}, "Reactor")
|
|
}
|
|
|
|
func (r *ReactorManager) Exit() error {
|
|
// allows for extra stuff
|
|
logging.Debug(logging.DExit, "RMA %v exiting", r.Id)
|
|
return r.Manager.Exit()
|
|
//go r.StatusMon.Send(&DeviceInfo{Id: r.Id, Type: "Reactor", Status: "[red]OFFLINE[white]", Data: fmt.Sprintf("Last Seen %v", time.Now().Format("Mon at 03:04:05pm MST"))}, "Reactor")
|
|
}
|
|
|
|
func (r *ReactorManager) UpdateClient(cl *Client) error {
|
|
// this is probably unnessecary
|
|
fmt.Printf("Reactor Manager %d updating client!\n", r.Id)
|
|
r.Client = cl
|
|
return nil
|
|
}
|
|
|
|
func (r *ReactorManager) ReactorStatusHandler(ctx context.Context, req *pb.ReactorStatusPing) (*pb.ReactorStatusResponse, error) {
|
|
// function client will call to update reactor information
|
|
//go r.PingReset()
|
|
fmt.Printf("Recieved ping from %d!\n", req.GetId())
|
|
// update devices/sensors
|
|
for _, dev := range req.GetDevices() {
|
|
fmt.Printf("Device %d is %s ", dev.GetAddr(), dev.GetStatus().String())
|
|
}
|
|
fmt.Printf("\n")
|
|
// go r.UpdateDevices(req.GetDevices())
|
|
|
|
return &pb.ReactorStatusResponse{Id: int32(r.Id)}, nil
|
|
}
|
|
|
|
// // device stuff
|
|
|
|
// type DeviceManager interface {
|
|
// LoadConfig() error
|
|
// UpdateStatus(string) error
|
|
// String() string // printing
|
|
// }
|
|
|
|
// func NewDeviceManager(addr int, config *viper.Viper, prefix string) (DeviceManager, error) {
|
|
// // returns a manager struct
|
|
// return device.NewDeviceManager(addr, config, prefix)
|
|
// }
|
|
|
|
//func (r *ReactorManager) UpdateDevices(devs []*pb.Device) {
|
|
// // pass updates to correct manager
|
|
// r.ReactorDevices.RLock() // read lock only
|
|
// defer r.ReactorDevices.RUnlock()
|
|
|
|
// for _, dev := range devs {
|
|
// // looping over devs
|
|
// if dm, ok := r.ReactorDevices.Devices[int(dev.GetAddr())]; ok {
|
|
// // device manager found
|
|
// go dm.UpdateStatus(dev.GetStatus().String())
|
|
// //fmt.Println(dm)
|
|
// } else {
|
|
// // not found
|
|
// go r.AddDevice(dev, r.Id, r.Config, r.Err)
|
|
// }
|
|
// }
|
|
//}
|
|
|
|
// func (r *ReactorDevices) AddDevice(dev *pb.Device, id int, config *viper.Viper, errCh chan error) {
|
|
|
|
// // setting vars
|
|
// prefix := fmt.Sprintf("reactors.%d.", id)
|
|
// addr := int(dev.GetAddr())
|
|
// var dm DeviceManager
|
|
// var err error
|
|
// // write locking
|
|
// r.Lock()
|
|
// defer r.Unlock()
|
|
|
|
// if dm, err = NewDeviceManager(addr, config, prefix); err != nil {
|
|
// errCh <- err
|
|
// }
|
|
|
|
// // setting status
|
|
// if err = dm.UpdateStatus(dev.GetStatus().String()); err != nil {
|
|
// errCh <- err
|
|
// }
|
|
|
|
// // loading config
|
|
// if err = dm.LoadConfig(); err != nil {
|
|
// errCh <- err
|
|
// }
|
|
// r.Devices[int(addr)] = dm
|
|
// }
|