eosswedenorg/thalos
1
0
Fork 0
mirror of https://github.com/eosswedenorg/thalos synced 2026-06-19 04:50:02 +02:00
Code Releases Activity

api/client.go: refactor decoding in handler functions to a helper function

Browse source
This commit is contained in:
Henrik Hautakoski 2024-02-04 15:54:10 +01:00
parent 728b03422f
commit 38f885f0df
1 changed files with 18 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

36
api/client.go
View file

@ -45,40 +45,40 @@ func (c *Client) worker(channel Channel, h handler) {
} }
} }
// Action handler // Helper method to decode a message and call OnError on error.
func (c *Client) actHandler(payload []byte) { // Returns true if successfull. false otherwise
var act message.ActionTrace func (c *Client) decode(payload []byte, msg any) bool {
if err := c.decoder(payload, &act); err != nil { if err := c.decoder(payload, msg); err != nil {
if c.OnError != nil { if c.OnError != nil {
c.OnError(err) c.OnError(err)
} }
return return false
}
return true
}
// Action handler
func (c *Client) actHandler(payload []byte) {
var act message.ActionTrace
if ok := c.decode(payload, &act); ok {
c.OnAction(act)
} }
c.OnAction(act)
} }
// TableDelta handler // TableDelta handler
func (c *Client) tableDeltaHandler(payload []byte) { func (c *Client) tableDeltaHandler(payload []byte) {
td := message.TableDelta{} td := message.TableDelta{}
if err := c.decoder(payload, &td); err != nil { if ok := c.decode(payload, &td); ok {
if c.OnError != nil { c.OnTableDelta(td)
c.OnError(err)
}
return
} }
c.OnTableDelta(td)
} }
// HeartBeat handler // HeartBeat handler
func (c *Client) hbHandler(payload []byte) { func (c *Client) hbHandler(payload []byte) {
var hb message.HeartBeat var hb message.HeartBeat
if err := c.decoder(payload, &hb); err != nil { if ok := c.decode(payload, &hb); ok {
if c.OnError != nil { c.OnHeartbeat(hb)
c.OnError(err)
}
return
} }
c.OnHeartbeat(hb)
} }
func (c *Client) Subscribe(channel Channel) error { func (c *Client) Subscribe(channel Channel) error {