Browse Source
new scene system is kind of working. I haven't tested changing between scenes and Transition effect yet, though.
pull/1/head
new scene system is kind of working. I haven't tested changing between scenes and Transition effect yet, though.
pull/1/head
9 changed files with 380 additions and 140 deletions
-
21app/api/devices.go
-
15app/server.go
-
12app/services/events.go
-
81app/services/publish.go
-
275app/services/publisher/publisher.go
-
66app/services/publisher/scene.go
-
3internal/mysql/devicerepo.go
-
20models/device.go
-
9models/scene.go
@ -1,81 +0,0 @@ |
|||
package services |
|||
|
|||
import ( |
|||
"context" |
|||
"git.aiterp.net/lucifer/new-server/app/config" |
|||
"git.aiterp.net/lucifer/new-server/models" |
|||
"log" |
|||
"sync" |
|||
"time" |
|||
) |
|||
|
|||
func StartPublisher() { |
|||
ctx := context.Background() |
|||
|
|||
go func() { |
|||
for devices := range config.PublishChannel { |
|||
if len(devices) == 0 { |
|||
continue |
|||
} |
|||
|
|||
lists := make(map[int][]models.Device, 4) |
|||
for _, device := range devices { |
|||
lists[device.BridgeID] = append(lists[device.BridgeID], device) |
|||
} |
|||
|
|||
ctx, cancel := context.WithTimeout(ctx, time.Second * 30) |
|||
|
|||
bridges, err := config.BridgeRepository().FetchAll(ctx) |
|||
if err != nil { |
|||
log.Println("Publishing error (1): " + err.Error()) |
|||
cancel() |
|||
continue |
|||
} |
|||
|
|||
wg := sync.WaitGroup{} |
|||
for _, devices := range lists { |
|||
wg.Add(1) |
|||
|
|||
go func(devices []models.Device) { |
|||
defer wg.Done() |
|||
|
|||
var bridge models.Bridge |
|||
for _, bridge2 := range bridges { |
|||
if bridge2.ID == devices[0].BridgeID { |
|||
bridge = bridge2 |
|||
} |
|||
} |
|||
if bridge.ID == 0 { |
|||
log.Println("Unknown bridge") |
|||
return |
|||
} |
|||
|
|||
if bridge.Driver == models.DTLIFX { |
|||
return |
|||
} |
|||
|
|||
bridge, err := config.BridgeRepository().Find(ctx, devices[0].BridgeID) |
|||
if err != nil { |
|||
log.Println("Publishing error (1): " + err.Error()) |
|||
return |
|||
} |
|||
|
|||
driver, err := config.DriverProvider().Provide(bridge.Driver) |
|||
if err != nil { |
|||
log.Println("Publishing error (2): " + err.Error()) |
|||
return |
|||
} |
|||
|
|||
err = driver.Publish(ctx, bridge, devices) |
|||
if err != nil { |
|||
log.Println("Publishing error (3): " + err.Error()) |
|||
return |
|||
} |
|||
}(devices) |
|||
} |
|||
|
|||
wg.Wait() |
|||
cancel() |
|||
} |
|||
}() |
|||
} |
|||
@ -0,0 +1,275 @@ |
|||
package publisher |
|||
|
|||
import ( |
|||
"context" |
|||
"git.aiterp.net/lucifer/new-server/app/config" |
|||
"git.aiterp.net/lucifer/new-server/models" |
|||
"log" |
|||
"sync" |
|||
"time" |
|||
) |
|||
|
|||
type Publisher struct { |
|||
mu sync.Mutex |
|||
sceneData map[int]*models.Scene |
|||
scenes []*Scene |
|||
sceneAssignment map[int]*Scene |
|||
started map[int]bool |
|||
pending map[int][]models.Device |
|||
waiting map[int]chan struct{} |
|||
} |
|||
|
|||
func (p *Publisher) ReloadScenes(ctx context.Context) error { |
|||
scenes, err := config.SceneRepository().FetchAll(ctx) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
p.mu.Lock() |
|||
for _, scene := range scenes { |
|||
p.sceneData[scene.ID] = &scene |
|||
} |
|||
p.mu.Unlock() |
|||
|
|||
return nil |
|||
} |
|||
|
|||
func (p *Publisher) ReloadDevices(ctx context.Context) error { |
|||
devices, err := config.DeviceRepository().FetchByReference(ctx, models.RKAll, "") |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
p.Publish(devices...) |
|||
|
|||
return nil |
|||
} |
|||
|
|||
func (p *Publisher) Publish(devices ...models.Device) { |
|||
if len(devices) == 0 { |
|||
return |
|||
} |
|||
|
|||
p.mu.Lock() |
|||
defer p.mu.Unlock() |
|||
|
|||
for _, device := range devices { |
|||
if !p.started[device.BridgeID] { |
|||
p.started[device.BridgeID] = true |
|||
go p.runBridge(device.BridgeID) |
|||
} |
|||
|
|||
p.reassignDevice(device) |
|||
|
|||
if p.sceneAssignment[device.ID] != nil { |
|||
p.sceneAssignment[device.ID].UpsertDevice(device) |
|||
} else { |
|||
p.pending[device.BridgeID] = append(p.pending[device.BridgeID], device) |
|||
if p.waiting[device.BridgeID] != nil { |
|||
close(p.waiting[device.BridgeID]) |
|||
p.waiting[device.BridgeID] = nil |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
func (p *Publisher) PublishChannel(ch <-chan []models.Device) { |
|||
for list := range ch { |
|||
p.Publish(list...) |
|||
} |
|||
} |
|||
|
|||
func (p *Publisher) Run() { |
|||
ticker := time.NewTicker(time.Millisecond * 100) |
|||
deleteList := make([]int, 0, 8) |
|||
updatedList := make([]models.Device, 0, 16) |
|||
|
|||
for range ticker.C { |
|||
deleteList = deleteList[:0] |
|||
updatedList = updatedList[:0] |
|||
|
|||
p.mu.Lock() |
|||
for i, scene := range p.scenes { |
|||
if !scene.endTime.IsZero() && time.Now().After(scene.endTime) { |
|||
deleteList = append(deleteList, i-len(deleteList)) |
|||
updatedList = append(updatedList, scene.AllDevices()...) |
|||
|
|||
for _, device := range scene.AllDevices() { |
|||
p.sceneAssignment[device.ID] = nil |
|||
} |
|||
continue |
|||
} |
|||
|
|||
if scene.Due() { |
|||
updatedList = append(updatedList, scene.Run()...) |
|||
updatedList = append(updatedList, scene.UnaffectedDevices()...) |
|||
} |
|||
} |
|||
|
|||
for _, i := range deleteList { |
|||
p.scenes = append(p.scenes[:i], p.scenes[i+1:]...) |
|||
} |
|||
|
|||
for _, device := range updatedList { |
|||
if !p.started[device.BridgeID] { |
|||
p.started[device.BridgeID] = true |
|||
go p.runBridge(device.BridgeID) |
|||
} |
|||
|
|||
p.pending[device.BridgeID] = append(p.pending[device.BridgeID], device) |
|||
if p.waiting[device.BridgeID] != nil { |
|||
close(p.waiting[device.BridgeID]) |
|||
p.waiting[device.BridgeID] = nil |
|||
} |
|||
} |
|||
p.mu.Unlock() |
|||
} |
|||
} |
|||
|
|||
// reassignDevice re-evaluates the device's scene assignment config. It will return whether the scene changed, which
|
|||
// should trigger an update.
|
|||
func (p *Publisher) reassignDevice(device models.Device) bool { |
|||
var selectedAssignment *models.DeviceSceneAssignment |
|||
for _, assignment := range device.SceneAssignments { |
|||
duration := time.Duration(assignment.DurationMS) * time.Millisecond |
|||
if duration <= 0 || time.Now().Before(assignment.StartTime.Add(duration)) { |
|||
selectedAssignment = &assignment |
|||
} |
|||
} |
|||
|
|||
if selectedAssignment == nil { |
|||
if p.sceneAssignment[device.ID] != nil { |
|||
p.sceneAssignment[device.ID].RemoveDevice(device) |
|||
delete(p.sceneAssignment, device.ID) |
|||
|
|||
// Scene changed
|
|||
return true |
|||
} |
|||
|
|||
// Stop here, no scene should be assigned.
|
|||
return false |
|||
} |
|||
|
|||
if p.sceneAssignment[device.ID] != nil { |
|||
scene := p.sceneAssignment[device.ID] |
|||
if scene.data.ID == selectedAssignment.SceneID && scene.group == selectedAssignment.Group { |
|||
// Current assignment is good.
|
|||
return false |
|||
} |
|||
|
|||
p.sceneAssignment[device.ID].RemoveDevice(device) |
|||
delete(p.sceneAssignment, device.ID) |
|||
} |
|||
|
|||
for _, scene := range p.scenes { |
|||
if scene.data.ID == selectedAssignment.SceneID && scene.group == selectedAssignment.Group { |
|||
p.sceneAssignment[device.ID] = scene |
|||
return true |
|||
} |
|||
} |
|||
|
|||
newScene := &Scene{ |
|||
data: p.sceneData[selectedAssignment.SceneID], |
|||
group: selectedAssignment.Group, |
|||
startTime: selectedAssignment.StartTime, |
|||
endTime: selectedAssignment.StartTime.Add(time.Duration(selectedAssignment.DurationMS) * time.Millisecond), |
|||
roleMap: make(map[int]int, 16), |
|||
roleList: make(map[int][]models.Device, 16), |
|||
due: true, |
|||
} |
|||
p.sceneAssignment[device.ID] = newScene |
|||
p.scenes = append(p.scenes, newScene) |
|||
|
|||
if selectedAssignment.DurationMS <= 0 { |
|||
newScene.endTime = time.Time{} |
|||
} |
|||
|
|||
return true |
|||
} |
|||
|
|||
func (p *Publisher) runBridge(id int) { |
|||
defer func() { |
|||
p.mu.Lock() |
|||
p.started[id] = false |
|||
p.mu.Unlock() |
|||
}() |
|||
|
|||
bridge, err := config.BridgeRepository().Find(context.Background(), id) |
|||
if err != nil { |
|||
log.Println("Failed to get bridge data:", err) |
|||
return |
|||
} |
|||
|
|||
driver, err := config.DriverProvider().Provide(bridge.Driver) |
|||
if err != nil { |
|||
log.Println("Failed to get bridge driver:", err) |
|||
log.Println("Maybe Lucifer needs to be updated.") |
|||
return |
|||
} |
|||
|
|||
devices := make(map[int]models.Device) |
|||
|
|||
for { |
|||
p.mu.Lock() |
|||
if len(p.pending[id]) == 0 { |
|||
if p.waiting[id] == nil { |
|||
p.waiting[id] = make(chan struct{}) |
|||
} |
|||
waitCh := p.waiting[id] |
|||
p.mu.Unlock() |
|||
<-waitCh |
|||
p.mu.Lock() |
|||
} |
|||
|
|||
updates := p.pending[id] |
|||
p.pending[id] = p.pending[id][:0:0] |
|||
p.mu.Unlock() |
|||
|
|||
// Only allow the latest update per device (this avoids slow bridges causing a backlog of cations).
|
|||
for key := range devices { |
|||
delete(devices, key) |
|||
} |
|||
for _, update := range updates { |
|||
devices[update.ID] = update |
|||
} |
|||
updates = updates[:0] |
|||
for _, value := range devices { |
|||
updates = append(updates, value) |
|||
} |
|||
|
|||
err := driver.Publish(context.Background(), bridge, updates) |
|||
if err != nil { |
|||
log.Println("Failed to publish to driver:", err) |
|||
|
|||
p.mu.Lock() |
|||
p.pending[id] = append(updates, p.pending[id]...) |
|||
p.mu.Unlock() |
|||
return |
|||
} |
|||
} |
|||
} |
|||
|
|||
var publisher = Publisher{ |
|||
sceneData: make(map[int]*models.Scene), |
|||
scenes: make([]*Scene, 0, 16), |
|||
sceneAssignment: make(map[int]*Scene, 16), |
|||
started: make(map[int]bool), |
|||
pending: make(map[int][]models.Device), |
|||
waiting: make(map[int]chan struct{}), |
|||
} |
|||
|
|||
func Initialize(ctx context.Context) error { |
|||
err :=publisher.ReloadScenes(ctx) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
err = publisher.ReloadDevices(ctx) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
go publisher.PublishChannel(config.PublishChannel) |
|||
go publisher.Run() |
|||
|
|||
return nil |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue