jshiffer/go-xmpp
1
0
Fork 1
mirror of https://github.com/FluuxIO/go-xmpp.git synced 2024-12-03 07:42:00 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Renamed Hooks

Browse Source
This commit is contained in:
rcorniere 2020-03-10 16:31:27 +01:00
parent 477a2b114c
commit 7850d07d37
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
client.go
View File

@ -139,10 +139,10 @@ type Client struct {
ErrorHandler func(error)
// Post connection hook. This will be executed on first connection
PostFirstConnHook func() error
PostConnectHook func() error
// Post resume hook. This will be executed after the client resumes a lost connection using StreamManagement (XEP-0198)
PostReconnectHook func() error
PostResumeHook func() error
}
/*
@ -213,7 +213,7 @@ func NewClient(config *Config, r *Router, errorHandler func(error)) (c *Client,
}
// Connect establishes a first time connection to a XMPP server.
// It calls the PostFirstConnHook
// It calls the PostConnectHook
func (c *Client) Connect() error {
err := c.connect()
if err != nil {
@ -223,8 +223,8 @@ func (c *Client) Connect() error {
// Do we need an option to avoid that or do we rely on client to send the presence itself ?
err = c.sendWithWriter(c.transport, []byte(InitialPresence))
// Execute the post first connection hook. Typically this holds "ask for roster" and this type of actions.
if c.PostFirstConnHook != nil {
err = c.PostFirstConnHook()
if c.PostConnectHook != nil {
err = c.PostConnectHook()
if err != nil {
return err
}
@ -287,8 +287,8 @@ func (c *Client) Resume() error {
}
// Execute post reconnect hook. This can be different from the first connection hook, and not trigger roster retrival
// for example.
if c.PostReconnectHook != nil {
err = c.PostReconnectHook()
if c.PostResumeHook != nil {
err = c.PostResumeHook()
}
return err
}

4
client_test.go
View File

@ -417,7 +417,7 @@ func Test_ClientPostConnectHook(t *testing.T) {
}
// The post connection client hook should just write to a channel that we will read later.
client.PostFirstConnHook = func() error {
client.PostConnectHook = func() error {
go func() {
hookChan <- struct{}{}
}()
@ -486,7 +486,7 @@ func Test_ClientPostReconnectHook(t *testing.T) {
t.Errorf("connect create XMPP client: %s", err)
}
client.PostReconnectHook = func() error {
client.PostResumeHook = func() error {
go func() {
hookChan <- struct{}{}
}()