Example #1
0
func startQueue(name string) {
	for {
		select {
		case body := <-channels[name]:
			stomp_message := stomp.NewStompMessage("MESSAGE", map[string]string{"destination": name}, body)

			for i := 0; i < len(connections[name]); i++ {
				connections[name][i].Write([]byte(stomp_message.ToString()))
			}
		}
	}
}
Example #2
0
func handleMessage(msg *stomp.StompMessage, conn net.Conn) {
	switch msg.Command {
	case "CONNECT":
		//Respond with a connected frame
		//Add hostname to the connection pool
		hosts = append(hosts, conn.RemoteAddr().String())
		response := stomp.NewStompMessage("CONNECTED", map[string]string{"session": "0"}, "").ToString()
		conn.Write([]byte(response))

	case "SUBSCRIBE":
		// Check if destination exists
		// create the destination if it doesnt
		// Add connection to list
		// TODO Deal with ack headers
		// TODO Deal with multiple subscribes to the same topic
		if _, present := msg.Headers["destination"]; present {

			if _, present = channels[msg.Headers["destination"]]; !present {
				fmt.Println("Creating Queue : " + msg.Headers["destination"])
				channels[msg.Headers["destination"]] = make(chan string)

				go startQueue(msg.Headers["destination"])
			}

			if _, present = connections[msg.Headers["destination"]]; !present {
				connections[msg.Headers["destination"]] = make([]net.Conn, 0)
				connections[msg.Headers["destination"]] = append(connections[msg.Headers["destination"]], conn)
			} else {
				connections[msg.Headers["destination"]] = append(connections[msg.Headers["destination"]], conn)
			}
		}

	case "UNSUBSCRIBE":
		// Remove the connection from the connection pool
		if _, present := connections[msg.Headers["destination"]]; present {
			for i := 0; i < len(connections[msg.Headers["destination"]]); i++ {
				if connections[msg.Headers["destination"]][i].RemoteAddr().String() == conn.RemoteAddr().String() {
					//TODO FIX THIS
					//copy(connections[msg.Headers["destination"]][i-1:], connections[msg.Headers["destination"]][i+1:])
					break
				}
			}
		}

	case "SEND":
		if _, present := channels[msg.Headers["destination"]]; present {
			channels[msg.Headers["destination"]] <- msg.Body
		}

	case "DISCONNECT":
		if _, present := connections[msg.Headers["destination"]]; present {
			for i := 0; i < len(connections[msg.Headers["destination"]]); i++ {
				if connections[msg.Headers["destination"]][i].RemoteAddr().String() == conn.RemoteAddr().String() {
					connections[msg.Headers["destination"]][i].Close()
					//TODO FIX THIS
					//connections[msg.Headers["destination"]][i] = append(connections[msg.Headers["destination"]][i-1:], connections[msg.Headers["destination"]][i+1:])
				}
			}
		}

	case "BEGIN":
	//TODO: Implement

	case "COMMIT":
	//TODO: Implement

	case "ABORT":
	//TODO: Implement

	case "ACK":
		//TODO: Implement
	}
}