83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
package models
|
|
|
|
import "slices"
|
|
|
|
// driver model
|
|
type Driver struct {
|
|
Type string `json:"type"` // driver type name of driver in collection
|
|
Buses []*Bus `json:"buses,omitempty"` // name of driver bus
|
|
}
|
|
|
|
func (d *Driver) AddNewBus(name string) *Bus {
|
|
for _, b := range d.Buses {
|
|
if b.Name == name {
|
|
return b
|
|
}
|
|
}
|
|
b := &Bus{Name: name}
|
|
d.Buses = append(d.Buses, b)
|
|
return b
|
|
}
|
|
|
|
func (d *Driver) AddBuses(buses []*Bus) {
|
|
next:
|
|
for _, newBus := range buses {
|
|
for _, currentBus := range d.Buses {
|
|
if currentBus.Name != newBus.Name {
|
|
continue
|
|
}
|
|
currentBus.AddAddress(newBus.Address...)
|
|
if newBus.Topic == nil {
|
|
continue next
|
|
}
|
|
currentBus.AddSubscription(newBus.Topic.Subscribe...)
|
|
currentBus.AddPublish(newBus.Topic.Publish...)
|
|
continue next
|
|
}
|
|
d.Buses = append(d.Buses, newBus)
|
|
}
|
|
}
|
|
|
|
func (d *Driver) RemoveBuses(buses []*Bus) {
|
|
for _, bus := range buses {
|
|
for _, currentBus := range d.Buses {
|
|
if currentBus.Name != bus.Name {
|
|
continue
|
|
}
|
|
|
|
currentBus.RemoveAddress(bus.Address...)
|
|
if bus.Topic != nil {
|
|
currentBus.RemoveSubscription(bus.Topic.Subscribe...)
|
|
currentBus.RemovePublish(bus.Topic.Publish...)
|
|
if len(currentBus.Address) == 0 {
|
|
var remove bool = true
|
|
if currentBus.Topic != nil {
|
|
remove = len(currentBus.Topic.Subscribe) == 0 && len(currentBus.Topic.Publish) == 0
|
|
}
|
|
if remove {
|
|
if len(d.Buses) == 1 {
|
|
d.Buses = []*Bus{}
|
|
return
|
|
}
|
|
for i, b := range d.Buses {
|
|
if bus.Name == b.Name {
|
|
d.Buses = slices.Delete(d.Buses, i, i+1)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (d *Driver) GetBus(name string) *Bus {
|
|
for _, b := range d.Buses {
|
|
if b.Name == name {
|
|
return b
|
|
}
|
|
}
|
|
return nil
|
|
}
|