tg/cmd/tg-broker/clients.go

80 lines
1.2 KiB
Go
Raw Normal View History

2018-04-03 09:51:06 +00:00
package main
import (
"bufio"
"encoding/json"
"fmt"
"log"
"net"
2018-04-03 09:54:00 +00:00
"github.com/hamcha/tg"
2018-04-03 09:51:06 +00:00
)
var clients []net.Conn
func startClientsServer(bind string) {
listener, err := net.Listen("tcp", bind)
assert(err)
// Accept loop
for {
c, err := listener.Accept()
if err != nil {
log.Printf("Can't accept client: %s\n", err.Error())
continue
}
clients = append(clients, c)
go handleClient(c)
}
}
func handleClient(c net.Conn) {
b := bufio.NewReader(c)
defer c.Close()
// Start reading messages
buf := make([]byte, 0)
for {
bytes, isPrefix, err := b.ReadLine()
if err != nil {
break
}
buf = append(buf, bytes...)
if isPrefix {
continue
}
// Get command
var cmd tg.ClientCommand
err = json.Unmarshal(buf, &cmd)
if err != nil {
log.Printf("[handleClient] Can't parse JSON: %s\r\n", err.Error())
log.Printf("%s\n", string(buf))
continue
}
// Empty buffer
buf = []byte{}
executeClientCommand(cmd, c)
}
removeCon(c)
}
func removeCon(c net.Conn) {
for i, con := range clients {
if c == con {
clients = append(clients[:i], clients[i+1:]...)
}
}
}
func broadcast(message string) {
for _, c := range clients {
_, err := fmt.Fprintln(c, message)
if err != nil {
removeCon(c)
}
}
}