add remove driver function
This commit is contained in:
@@ -16,9 +16,7 @@ func (b *Bus) AddAddress(address ...uint) {
|
|||||||
}
|
}
|
||||||
b.Address = append(b.Address, a)
|
b.Address = append(b.Address, a)
|
||||||
slices.Sort(b.Address)
|
slices.Sort(b.Address)
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bus) AddSubscription(sub ...string) {
|
func (b *Bus) AddSubscription(sub ...string) {
|
||||||
@@ -34,3 +32,25 @@ func (b *Bus) AddPublish(pub ...string) {
|
|||||||
}
|
}
|
||||||
b.Topic.AddPublish(pub...)
|
b.Topic.AddPublish(pub...)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *Bus) RemoveAddress(address ...uint) {
|
||||||
|
if len(b.Address) == 1 {
|
||||||
|
b.Address = []uint{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, toDelete := range address {
|
||||||
|
for i, a := range b.Address {
|
||||||
|
if a == toDelete {
|
||||||
|
b.Address = slices.Delete(b.Address, i, i+1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bus) RemoveSubscription(sub ...string) {
|
||||||
|
b.Topic.RemoveSubscription(sub...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bus) RemovePublish(pub ...string) {
|
||||||
|
b.Topic.RemovePublish(pub...)
|
||||||
|
}
|
||||||
|
@@ -1,5 +1,7 @@
|
|||||||
package models
|
package models
|
||||||
|
|
||||||
|
import "slices"
|
||||||
|
|
||||||
// driver model
|
// driver model
|
||||||
type Driver struct {
|
type Driver struct {
|
||||||
Type string `json:"type"` // driver type name of driver in collection
|
Type string `json:"type"` // driver type name of driver in collection
|
||||||
@@ -36,6 +38,38 @@ next:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (d *Driver) RemoveBus(bus *Bus) {
|
||||||
|
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 {
|
func (d *Driver) GetBus(name string) *Bus {
|
||||||
for _, b := range d.Buses {
|
for _, b := range d.Buses {
|
||||||
if b.Name == name {
|
if b.Name == name {
|
||||||
|
@@ -27,3 +27,31 @@ func (t *Topic) AddPublish(pubs ...string) {
|
|||||||
slices.Sort(t.Publish)
|
slices.Sort(t.Publish)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (t *Topic) RemoveSubscription(subs ...string) {
|
||||||
|
if len(t.Subscribe) == 1 {
|
||||||
|
t.Subscribe = []string{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, toDelete := range subs {
|
||||||
|
for i, a := range t.Subscribe {
|
||||||
|
if a == toDelete {
|
||||||
|
t.Subscribe = slices.Delete(t.Subscribe, i, i+1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *Topic) RemovePublish(pubs ...string) {
|
||||||
|
if len(t.Publish) == 1 {
|
||||||
|
t.Publish = []string{}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, toDelete := range pubs {
|
||||||
|
for i, a := range t.Publish {
|
||||||
|
if a == toDelete {
|
||||||
|
t.Publish = slices.Delete(t.Publish, i, i+1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Reference in New Issue
Block a user