hyprland-tools/hypr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added context for subscribe method

Browse Source
This commit is contained in:
Milarin 2024-02-09 21:30:14 +01:00
parent bec954ef31
commit e308674337
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
subscribe_event.go
View File

@ -2,12 +2,13 @@ package hypr
import ( import (
"bufio" "bufio"
"context"
"strings" "strings"
"git.milar.in/milarin/slices" "git.milar.in/milarin/slices"
) )
func (i *Instance) Subscribe(events ...EventType) (<-chan Event, error) { func (i *Instance) Subscribe(ctx context.Context, events ...EventType) (<-chan Event, error) {
r, err := readSocketRaw(i.EventSocketPath(), strings.NewReader("")) r, err := readSocketRaw(i.EventSocketPath(), strings.NewReader(""))
if err != nil { if err != nil {
return nil, err return nil, err
@ -21,7 +22,8 @@ func (i *Instance) Subscribe(events ...EventType) (<-chan Event, error) {
defer close(out) defer close(out)
sc := bufio.NewScanner(r) sc := bufio.NewScanner(r)
for sc.Scan() {
for ctx.Err() == nil && sc.Scan() {
event := parseEvent(sc.Text()) event := parseEvent(sc.Text())
if _, ok := eventMap[event.Type]; ok { if _, ok := eventMap[event.Type]; ok {
out <- event out <- event