Add Stomp 1.2 examples, ACK examples, make minor corrections.

gmallard · Nov 9, 2012 · df8c86b · df8c86b
1 parent 4b80197
commit df8c86b
Show file tree

Hide file tree

Showing 10 changed files with 633 additions and 10 deletions.
diff --git a/ack_10/ack_10.go b/ack_10/ack_10.go
@@ -0,0 +1,122 @@
+//
+// Copyright © 2011-2012 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.
+//
+
+/*
+Receive messages from a STOMP 1.0 broker, and ACK them.
+*/
+package main
+
+import (
+	"fmt"
+	"github.com/gmallard/stompngo"
+	"github.com/gmallard/stompngo_examples/sngecomm"
+	"log"
+	"net"
+)
+
+var exampid = "ack_10: "
+
+// Connect to a STOMP 1.0 broker, receive some messages and disconnect.
+func main() {
+	fmt.Println(exampid + "starts ...")
+
+	// Set up the connection.
+	h, p := sngecomm.HostAndPort10()
+	n, e := net.Dial("tcp", net.JoinHostPort(h, p))
+	if e != nil {
+		log.Fatalln(e) // Handle this ......
+	}
+	fmt.Println(exampid + "dial complete ...")
+	eh := stompngo.Headers{}
+	conn, e := stompngo.Connect(n, eh)
+	if e != nil {
+		log.Fatalln(e) // Handle this ......
+	}
+	fmt.Println(exampid + "stomp connect complete ...", conn.Protocol())
+
+	// Setup Headers ...
+	uh := stompngo.Headers{"destination", sngecomm.Dest()} // unsubscribe headers
+	s := stompngo.Headers{"destination", sngecomm.Dest(),  // subscribe headers
+		"ack", "client"}
+
+	// *NOTE* your application functionaltiy goes here!
+	// With Stomp, you must SUBSCRIBE to a destination in order to receive.
+	// Stomp 1.0 allows subscribing without a unique subscription id, and we
+	// do that here.
+	// Subscribe returns:
+	// a) A channel of MessageData struct
+	// b) A possible error.  Always check for errors.  They can be logical
+	// errors detected by the stompngo package, or even hard network errors, for
+	// example the broker just crashed.
+	r, e := conn.Subscribe(s)
+	if e != nil {
+		log.Fatalln(e) // Handle this ...
+	}
+	fmt.Println(exampid + "stomp subscribe complete ...")
+	// Read data from the returned channel
+	for i := 1; i <= sngecomm.Nmsgs(); i++ {
+		m := <-r
+		fmt.Println(exampid + "channel read complete ...")
+		// MessageData has two components:
+		// a) a Message struct
+		// b) an Error value.  Check the error value as usual
+		if m.Error != nil {
+			log.Fatalln(m.Error) // Handle this
+		}
+		//
+		fmt.Printf("Frame Type: %s\n", m.Message.Command) // Will be MESSAGE or ERROR!
+		h := m.Message.Headers
+		for j := 0; j < len(h)-1; j += 2 {
+			fmt.Printf("Header: %s:%s\n", h[j], h[j+1])
+		}
+		fmt.Printf("Payload: %s\n", string(m.Message.Body)) // Data payload
+		// ACK the message just received.
+		ah := stompngo.Headers{"message-id", m.Message.Headers.Value("message-id")}
+		fmt.Println(exampid, "ACK Headers", ah)
+		e := conn.Ack(ah)
+		if e != nil {
+			log.Fatalln(e) // Handle this
+		}
+		// Spurious ERROR frame?
+		select {
+		case m = <-r:
+			log.Fatalln("RECEIVE not expected, got: [%v]\n", m)
+		default:
+		}
+		fmt.Println(exampid + "ACK complete ...")
+	}
+	// It is polite to unsubscribe, although unnecessary if a disconnect follows.
+	e = conn.Unsubscribe(uh)
+	if e != nil {
+		log.Fatalln(e) // Handle this ...
+	}
+	fmt.Println(exampid + "stomp unsubscribe complete ...")
+
+	// Disconnect from the Stomp server
+	e = conn.Disconnect(eh)
+	if e != nil {
+		log.Fatalln(e) // Handle this ......
+	}
+	fmt.Println(exampid + "stomp disconnect complete ...")
+	// Close the network connection
+	e = n.Close()
+	if e != nil {
+		log.Fatalln(e) // Handle this ......
+	}
+	fmt.Println(exampid + "network close complete ...")
+
+	fmt.Println(exampid + "ends ...")
+}
diff --git a/ack_11/ack_11.go b/ack_11/ack_11.go
@@ -0,0 +1,141 @@
+//
+// Copyright © 2011-2012 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.
+//
+
+/*
+Receive messages from a STOMP 1.1 broker, and ACK them.
+*/
+package main
+
+import (
+	"fmt"
+	"github.com/gmallard/stompngo"
+	"github.com/gmallard/stompngo_examples/sngecomm"
+	"log"
+	"net"
+)
+
+var exampid = "ack_11: "
+
+// Connect to a STOMP 1.1 broker, receive some messages and disconnect.
+func main() {
+	fmt.Println(exampid + "starts ...")
+
+	// Set up the connection.
+	h, p := sngecomm.HostAndPort11() // A 1.1 connection
+	n, e := net.Dial("tcp", net.JoinHostPort(h, p))
+	if e != nil {
+		log.Fatalln(e) // Handle this ......
+	}
+	fmt.Println(exampid + "dial complete ...")
+	ch := stompngo.Headers{"accept-version", "1.1",
+		"host", h}
+	conn, e := stompngo.Connect(n, ch)
+	if e != nil {
+		log.Fatalln(e) // Handle this ......
+	}
+	fmt.Println(exampid + "stomp connect complete ...", conn.Protocol())
+
+	// Setup Headers ...
+	u := stompngo.Uuid() // Use package convenience function for unique ID
+	uh := stompngo.Headers{"destination", sngecomm.Dest(),
+		"id", u} // unsubscribe headers
+	s := stompngo.Headers{"destination", sngecomm.Dest(),
+		"id", u, "ack", "client"} // subscribe headers
+
+	// *NOTE* your application functionaltiy goes here!
+	// With Stomp, you must SUBSCRIBE to a destination in order to receive.
+	// Stomp 1.1 demands subscribing with a unique subscription id, and we
+	// do that here.
+	// Subscribe returns:
+	// a) A channel of MessageData struct.  Note that with this package, the
+	// channel is unique (based on the unique subscription id).
+	// b) A possible error.  Always check for errors.  They can be logical
+	// errors detected by the stompngo package, or even hard network errors, for
+	// example the broker just crashed.
+	r, e := conn.Subscribe(s)
+	if e != nil {
+		log.Fatalln(e) // Handle this ...
+	}
+	fmt.Println(exampid + "stomp subscribe complete ...")
+	// Read data from the returned channel
+	for i := 1; i <= sngecomm.Nmsgs(); i++ {
+		m := <-r
+		fmt.Println(exampid + "channel read complete ...")
+		// MessageData has two components:
+		// a) a Message struct
+		// b) an Error value.  Check the error value as usual
+		if m.Error != nil {
+			log.Fatalln(m.Error) // Handle this
+		}
+		//
+		fmt.Printf("Frame Type: %s\n", m.Message.Command) // Will be MESSAGE or ERROR!
+		h := m.Message.Headers
+		for j := 0; j < len(h)-1; j += 2 {
+			fmt.Printf("Header: %s:%s\n", h[j], h[j+1])
+		}
+		fmt.Printf("Payload: %s\n", string(m.Message.Body)) // Data payload
+		// One item to note:  Stomp 1.1 servers _must_ return a 'subscription'
+		// header in each message, where the value of the 'subscription' header
+		// matches the unique subscription id supplied on subscribe.  With _some_
+		// stomp client libraries, this allows you to categorize messages by
+		// 'subscription'.  That is not required with this package!!!  This
+		// because of the unique MessageData channels returned by Subscribe.
+		// But check that this is the case for demonstration purposes.
+		if h.Value("subscription") != u {
+			fmt.Printf("Error condition, expected [%s], received [%s]\n", u, h.Value("subscription"))
+			log.Fatalln("Bad subscription header")
+		}
+		// ACK the message just received.
+		ah := stompngo.Headers{"message-id", m.Message.Headers.Value("message-id"),
+			"subscription", u} // 1.1 ACK headers
+		fmt.Println(exampid, "ACK Headers", ah)
+		e := conn.Ack(ah)
+		if e != nil {
+			log.Fatalln(e) // Handle this
+		}
+		// Spurious ERROR frame?
+		select {
+		case m = <-r:
+			log.Fatalln("RECEIVE not expected, got: [%v]\n", m)
+		default:
+		}
+		fmt.Println(exampid + "ACK complete ...")
+
+	}
+	// It is polite to unsubscribe, although unnecessary if a disconnect follows.
+	// With Stomp 1.1, the same unique ID is required on UNSUBSCRIBE.  Failure
+	// to provide it will result in an error return.
+	e = conn.Unsubscribe(uh)
+	if e != nil {
+		log.Fatalln(e) // Handle this ...
+	}
+	fmt.Println(exampid + "stomp unsubscribe complete ...")
+
+	// Disconnect from the Stomp server
+	e = conn.Disconnect(stompngo.Headers{})
+	if e != nil {
+		log.Fatalln(e) // Handle this ......
+	}
+	fmt.Println(exampid + "stomp disconnect complete ...")
+	// Close the network connection
+	e = n.Close()
+	if e != nil {
+		log.Fatalln(e) // Handle this ......
+	}
+	fmt.Println(exampid + "network close complete ...")
+
+	fmt.Println(exampid + "ends ...")
+}