3 Commits

Author SHA1 Message Date
Adrian Zuercher
0150a0d120 fix tipo 2025-07-29 14:56:37 +02:00
Adrian Zuercher
b9c6aa9d02 change map to slice 2025-07-29 14:20:54 +02:00
Adrian Zuercher
4221815def modify drivers 2025-07-29 12:37:04 +02:00
5 changed files with 75 additions and 35 deletions

View File

@@ -4,28 +4,33 @@ import "slices"
// bus model
type Bus struct {
Name string `json:"name,omitemtpy"`
Address []uint `json:"address,omitemtpy"` // address of bus
Topic *Topic `json:"topic,omitemtpy"` // address of bus
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) {
if !slices.Contains(b.Address, address) {
b.Address = append(b.Address, address)
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) {
func (b *Bus) AddSubscription(sub ...string) {
if b.Topic == nil {
b.Topic = &Topic{}
}
b.Topic.AddSubscription(sub)
b.Topic.AddSubscription(sub...)
}
func (b *Bus) AddPublish(pub string) {
func (b *Bus) AddPublish(pub ...string) {
if b.Topic == nil {
b.Topic = &Topic{}
}
b.Topic.AddPublish(pub)
b.Topic.AddPublish(pub...)
}

View File

@@ -2,24 +2,45 @@ package models
// driver model
type Driver struct {
Type string `json:"type"` // driver type name of driver in collection
Buses map[string]*Bus `json:"buses"` // name of driver bus
Type string `json:"type"` // driver type name of driver in collection
Buses []*Bus `json:"buses,omitempty"` // name of driver bus
}
func (d *Driver) AddBus(name string) *Bus {
if d.Buses == nil {
d.Buses = make(map[string]*Bus)
func (d *Driver) AddNewBus(name string) *Bus {
for _, b := range d.Buses {
if b.Name == name {
return b
}
}
if b, ok := d.Buses[name]; ok {
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)
}
d.Buses[name] = &Bus{Name: name}
return d.Buses[name]
}
func (d *Driver) GetBus(name string) *Bus {
if b, ok := d.Buses[name]; ok {
return b
for _, b := range d.Buses {
if b.Name == name {
return b
}
}
return nil
}

View File

@@ -3,10 +3,18 @@ package models
// collection of drivers ordered in map
type Drivers map[string]*Driver
func (d *Drivers) AddDriver(typ string) *Driver {
func (d *Drivers) AddNewDriver(typ string) *Driver {
if drv, ok := (*d)[typ]; ok {
return drv
}
(*d)[typ] = &Driver{Type: typ}
return (*d)[typ]
}
func (d *Drivers) AddDriver(drv Driver) {
if driver, ok := (*d)[drv.Type]; ok {
driver.AddBuses(drv.Buses)
(*d)[drv.Type] = driver
}
(*d)[drv.Type] = &drv
}

View File

@@ -41,8 +41,8 @@ func (s *Set) NewDriverAddress(typ, bus string, address uint) {
if s.Drivers == nil {
s.Drivers = &Drivers{}
}
drv := s.Drivers.AddDriver(typ)
b := drv.AddBus(bus)
drv := s.Drivers.AddNewDriver(typ)
b := drv.AddNewBus(bus)
b.AddAddress(address)
}
@@ -51,8 +51,8 @@ func (s *Set) NewDriverSubscribe(typ, bus string, sub string) {
if s.Drivers == nil {
s.Drivers = &Drivers{}
}
drv := s.Drivers.AddDriver(typ)
b := drv.AddBus(bus)
drv := s.Drivers.AddNewDriver(typ)
b := drv.AddNewBus(bus)
b.AddSubscription(sub)
}
@@ -61,7 +61,7 @@ func (s *Set) NewDriverPublish(typ, bus string, pub string) {
if s.Drivers == nil {
s.Drivers = &Drivers{}
}
drv := s.Drivers.AddDriver(typ)
b := drv.AddBus(bus)
drv := s.Drivers.AddNewDriver(typ)
b := drv.AddNewBus(bus)
b.AddPublish(pub)
}

View File

@@ -4,20 +4,26 @@ import "slices"
// topic model
type Topic struct {
Publish []string `json:"Publish,omitemtpy"`
Subscribe []string `json:"subscribe,omitemtpy"`
Publish []string `json:"publish,omitempty"`
Subscribe []string `json:"subscribe,omitempty"`
}
func (t *Topic) AddSubscription(sub string) {
if !slices.Contains(t.Subscribe, sub) {
func (t *Topic) AddSubscription(subs ...string) {
for _, sub := range subs {
if slices.Contains(t.Subscribe, sub) {
continue
}
t.Subscribe = append(t.Subscribe, sub)
slices.Sort(t.Subscribe)
}
}
func (t *Topic) AddPublish(pub string) {
if !slices.Contains(t.Publish, pub) {
t.Publish = append(t.Publish, pub)
func (t *Topic) AddPublish(pubs ...string) {
for _, pub := range pubs {
if slices.Contains(t.Publish, pub) {
continue
}
t.Subscribe = append(t.Publish, pub)
slices.Sort(t.Publish)
}
}