tg/cmd/tg-broker/clients.go

package main

import (
	"bufio"
	"encoding/json"
	"fmt"
	"log"
	"net"

	"github.com/hamcha/tg"
)

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)
		}
	}
}