summaryrefslogtreecommitdiff
path: root/src/rhctl/switch_control.go
blob: f52e9a2bc8eb6a055765879622deb191cc4d9812 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
//
//  rhctl
//
//  Copyright (C) 2009-2016 Christian Pointner <equinox@helsinki.at>
//
//  This file is part of rhctl.
//
//  rhctl is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  any later version.
//
//  rhctl is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with rhctl. If not, see <http://www.gnu.org/licenses/>.
//

package main

import (
	"fmt"

	"github.com/btittelbach/pubsub"
)

type CommandType uint8

const (
	CmdStatus CommandType = iota
	CmdServer
	CmdSwitch
)

func (c CommandType) String() string {
	switch c {
	case CmdStatus:
		return "status"
	case CmdServer:
		return "server"
	case CmdSwitch:
		return "switch"
	}
	return "unknown"
}

type Command struct {
	Type     CommandType
	Args     []string
	Response chan<- interface{}
}

type SwitchControl struct {
	sw       *AudioSwitch
	servers  []*PlayoutServer
	Updates  *pubsub.PubSub
	Commands chan *Command
}

func (ctrl *SwitchControl) handleCommand(cmd *Command) {
	switch cmd.Type {
	case CmdStatus:
	case CmdServer:
	case CmdSwitch:
		if len(cmd.Args) == 0 {
			rhl.Printf("SwitchCTRL: ignoring empty raw switch command")
			return
		}
		c, err := NewSwitchCommandFromStrings(cmd.Args[0], cmd.Args[1:]...)
		if err != nil {
			cmd.Response <- fmt.Errorf("switch command syntax error: %s", err.Error())
			return
		}
		c.Response = cmd.Response
		ctrl.sw.Commands <- c
	}
}

func handleServer(in <-chan ServerStatus, out chan<- ServerStatus) {
	for {
		update := <-in
		out <- update
	}
}

func (ctrl *SwitchControl) Run() {
	rhdl.Printf("SwitchCTRL: handler running...")

	serverUpdates := make(chan ServerStatus, 8)
	for _, srv := range ctrl.servers {
		go handleServer(srv.Updates, serverUpdates)
	}

	for {
		select {
		case update := <-ctrl.sw.Updates:
			rhdl.Printf("got update from switch: %+v", update)
			for _, srv := range ctrl.servers {
				srv.SwitchUpdates <- update
			}
			ctrl.Updates.Pub(update, "switch:"+update.Type.String())
		case status := <-serverUpdates:
			rhdl.Printf("got server status update: %+v", status)
			ctrl.Updates.Pub(status, "server:status")
			// TODO: recalculate overall status and send out commands to switch
		case cmd := <-ctrl.Commands:
			ctrl.handleCommand(cmd)
		}
	}
}

func SwitchControlInit(conf *Config, sw *AudioSwitch, servers []*PlayoutServer) (ctrl *SwitchControl) {
	ctrl = &SwitchControl{}
	ctrl.sw = sw
	ctrl.servers = servers
	ctrl.Updates = pubsub.NewNonBlocking(32)
	ctrl.Commands = make(chan *Command, 8)
	return
}