hub

type Client struct {
	Conn     ConnectionInterface
	Send     chan Message
	Producer kafka.Producer
}

type ClientInterface interface {
	ReadPump(hub TheHub, logger Logger)
	WritePump(logger Logger)
	GetConnection() ConnectionInterface
	SendMessage(msg Message) error
	CloseSend() error
}

type ConnectionInterface interface {
	Close() error
	WriteMessage(int, []byte) error
	ReadMessage() (int, []byte, error)
}

type Hub struct {
	Clients    map[ClientInterface]bool
	Broadcast  chan Message
	Register   chan ClientInterface
	Unregister chan ClientInterface

	Logger Logger

	SaramaConfig *sarama.Config
	// contains filtered or unexported fields
}

type Logger interface {
	Printf(format string, v ...interface{})
}

type Message struct {
	Topic string `json:"topic"`
	Key   string `json:"key"`
	Data  string `json:"data"`
}

type TheHub interface {
	Run(ctx context.Context, logger Logger)
	RegisterClient(client ClientInterface)
	UnregisterClient(client ClientInterface)
	BroadcastMessage(message Message)
	HandleWebSocketMessage(message Message)
	UpgradeWebSocket(c *websocket.Conn, logger Logger)
	IsTest(isTest bool)
}

type WebSocketConnection struct {
	Conn *websocket.Conn
}