54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package handlers
|
|
|
|
import (
|
|
"btclock/broker"
|
|
ws "btclock/websocket"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"math"
|
|
|
|
"github.com/gofiber/contrib/websocket"
|
|
)
|
|
|
|
// WebSocketHandler implements broker.EventHandler for a websocket connection
|
|
type WebSocketV1Handler struct {
|
|
Cm *ws.ChannelManager
|
|
}
|
|
|
|
func (h *WebSocketV1Handler) Handle(ctx context.Context, event broker.Event) error {
|
|
channel := event.Name()
|
|
payload := map[string]interface{}{}
|
|
|
|
switch event.Name() {
|
|
case "ticker":
|
|
payloadMap := event.Payload().(map[string]interface{})
|
|
if payloadMap["symbol"] != "BTC/USD" {
|
|
return fmt.Errorf("unknown symbol: %s", payloadMap["symbol"])
|
|
}
|
|
payload["bitcoin"] = int(math.Round(payloadMap["last"].(float64)))
|
|
channel = "price-v1"
|
|
case "mempool-fee-rate":
|
|
payload["mempool-blocks"] = []map[string]interface{}{
|
|
{
|
|
"medianFee": int(math.Round(event.Payload().(float64))),
|
|
},
|
|
}
|
|
channel = "blockfee-v1"
|
|
case "mempool-block":
|
|
payload["block"] = map[string]interface{}{
|
|
"height": event.Payload(),
|
|
}
|
|
channel = "blockheight-v1"
|
|
default:
|
|
return fmt.Errorf("unknown event type: %s", event.Name())
|
|
}
|
|
|
|
b, err := json.Marshal(payload)
|
|
if err != nil {
|
|
return fmt.Errorf("error marshalling payload: %v", err)
|
|
}
|
|
|
|
h.Cm.BroadcastTo(channel, b, websocket.TextMessage)
|
|
return nil
|
|
}
|