uwaru/go-xmpp
1
0
Fork 0
forked from jshiffer/go-xmpp
Code Pull Requests Activity

Fix timeout when server doesn't reply with closing stream element.

Browse Source
This commit is contained in:
Martin Dosch 2024-03-28 17:22:02 +01:00
parent da17a46e6f
commit aef1257ed1
1 changed files with 14 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
xmpp.go
View File

@ -339,12 +339,12 @@ func NewClientNoTLS(host, user, passwd string, debug bool) (*Client, error) {
func (c *Client) Close() error { func (c *Client) Close() error {
if c.conn != (*tls.Conn)(nil) { if c.conn != (*tls.Conn)(nil) {
fmt.Fprintf(c.stanzaWriter, "</stream:stream>\n") fmt.Fprintf(c.stanzaWriter, "</stream:stream>\n")
go func() {
<-time.After(10 * time.Second)
c.conn.Close()
}()
// Wait for the server also closing the stream. // Wait for the server also closing the stream.
for { for {
select {
case <-time.After(10 * time.Second):
break
default:
ee, err := c.nextEnd() ee, err := c.nextEnd()
// If the server already closed the stream it is // If the server already closed the stream it is
// likely to receive an error when trying to parse // likely to receive an error when trying to parse
@ -358,7 +358,6 @@ func (c *Client) Close() error {
} }
} }
} }
}
return nil return nil
} }