57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package models
|
|
|
|
import "slices"
|
|
|
|
// bus model
|
|
type Bus struct {
|
|
Name string `json:"name"`
|
|
Address []uint `json:"address,omitempty"` // address of bus
|
|
Topic *Topic `json:"topic,omitempty"` // address of bus
|
|
}
|
|
|
|
func (b *Bus) AddAddress(address ...uint) {
|
|
for _, a := range address {
|
|
if slices.Contains(b.Address, a) {
|
|
continue
|
|
}
|
|
b.Address = append(b.Address, a)
|
|
slices.Sort(b.Address)
|
|
}
|
|
}
|
|
|
|
func (b *Bus) AddSubscription(sub ...string) {
|
|
if b.Topic == nil {
|
|
b.Topic = &Topic{}
|
|
}
|
|
b.Topic.AddSubscription(sub...)
|
|
}
|
|
|
|
func (b *Bus) AddPublish(pub ...string) {
|
|
if b.Topic == nil {
|
|
b.Topic = &Topic{}
|
|
}
|
|
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...)
|
|
}
|