50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package tg
|
|
|
|
import (
|
|
"bufio"
|
|
"encoding/json"
|
|
"io"
|
|
"log"
|
|
)
|
|
|
|
// UpdateHandler is an update handler for webhook updates
|
|
type UpdateHandler func(broker *Broker, message APIMessage)
|
|
|
|
// BrokerCallback is a callback for broker responses to client requests
|
|
type BrokerCallback func(broker *Broker, update BrokerUpdate)
|
|
|
|
// CreateBrokerClient creates a connection to a broker and sends all webhook updates to a given function.
|
|
// This is the intended way to create clients, please refer to examples for how to make a simple client.
|
|
func CreateBrokerClient(brokerAddr string, updateFn UpdateHandler) error {
|
|
broker, err := ConnectToBroker(brokerAddr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer broker.Close()
|
|
|
|
in := bufio.NewReader(broker.Socket)
|
|
for {
|
|
bytes, _, err := in.ReadLine()
|
|
if err != nil {
|
|
break
|
|
}
|
|
|
|
var update BrokerUpdate
|
|
err = json.Unmarshal(bytes, &update)
|
|
if err != nil {
|
|
log.Printf("[tg - CreateBrokerClient] ERROR reading JSON: %s\r\n", err.Error())
|
|
log.Printf("%s\n", string(bytes))
|
|
continue
|
|
}
|
|
|
|
if update.Callback == nil {
|
|
// It's a generic message: dispatch to UpdateHandler
|
|
go updateFn(broker, *(update.Message))
|
|
} else {
|
|
// It's a response to a request: retrieve callback and call it
|
|
go broker.SpliceCallback(*(update.Callback))(broker, update)
|
|
}
|
|
}
|
|
return io.EOF
|
|
}
|