forked from gmallard/stompngo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
disconnect.go
75 lines (68 loc) · 1.95 KB
/
disconnect.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
//
// Copyright © 2011-2014 Guy M. Allard
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
package stompngo
/*
Disconnect from a STOMP broker.
Shut down heart beats if necessary.
Set connection status to false to disable further actions with this
connection.
Obtain a receipt. If the client asks for a receipt, use the supplied receipt
id. Otherwise generate a uniqueue receipt id and add that to the DISCONNECT
headers.
Example:
h := stompngo.Headers{"receipt", "receipt-id1"} // Ask for a receipt
e := c.Disconnect(h)
if e != nil {
// Do something sane ...
}
fmt.Printf("%q\n", c.DisconnectReceipt)
*/
func (c *Connection) Disconnect(h Headers) error {
c.log(DISCONNECT, "start", h)
if !c.connected {
return ECONBAD
}
e := checkHeaders(h, c.Protocol())
if e != nil {
return e
}
ch := h.Clone()
// Add a receipt request if caller did not ask for one. This is in the spirit
// of the specification, and allows reasonable resource cleanup in both the
// client and the message broker.
if _, ok := ch.Contains("receipt"); !ok {
ch = append(ch, "receipt", Uuid())
}
//
c.connected = false
c.rsd <- true
f := Frame{DISCONNECT, ch, NULLBUFF}
//
r := make(chan error)
c.output <- wiredata{f, r}
e = <-r
//
if e != nil {
return e
}
// Drive shutdown logic
c.shutdown()
// Receipt
c.DisconnectReceipt = <-c.input
c.log(DISCONNECT, "end", ch, c.DisconnectReceipt)
close(c.input) // close message data channel
return nil
}