// Tell the instance to initialize the information.
func (ep *Endpoint) initial(context *zmq.Context,
act action, tp zmq.SocketType) error {
if ep.sock != nil {
return errors.New("Instance has already been initialized.")
}
// Check the pointer's validation.
if context == nil {
return errors.New("Specified context is null.")
}
var err error
ep.sock, err = context.NewSocket(tp)
if err != nil {
return err
}
// Check the addrs' validation.
for _, addr := range ep.Addr {
if !addr.valid() {
return ep.handleError(
errors.New("Specified addr not valid."))
}
if err = act(addr); err != nil {
return ep.handleError(err)
}
}
return nil
}
func runZmqStream() {
var context zmq.Context
var socket zmq.Socket
// connect to zmq
var err error
if context, err = zmq.NewContext(); err != nil {
panic("No ZMQ Context?")
}
defer context.Close()
if socket, err = context.NewSocket(zmq.SUB); err != nil {
panic("No ZMQ Socket Outbound??")
}
defer socket.Close()
socket.Connect("tcp://localhost:5558")
socket.SetSockOptString(zmq.SUBSCRIBE, "")
for {
// block here, waiting for inbound requests
msg, _ := socket.Recv(0)
if len(msg) > 0 {
parts := strings.Split(string(msg), "\n\n")
process(parts[0])
}
}
}
func WorkerSocket(context *zmq.Context) *zmq.Socket {
worker, _ := context.NewSocket(zmq.DEALER)
worker.Connect("tcp://localhost:5556")
// Tell queue we're ready for work
fmt.Println("I: worker ready")
worker.Send([]byte(PPP_READY), 0)
return worker
}
// Helper functions {{{
func openSocket(ctx zmq.Context, t zmq.SocketType, url string) zmq.Socket {
socket, _ := ctx.NewSocket(t)
if t == zmq.PUB || t == zmq.REP {
socket.Bind(url)
} else {
socket.Connect(url)
}
return socket
}
func newBoundSocket(context zmq.Context, address string, kind zmq.SocketType) zmq.Socket {
socket, _ := context.NewSocket(kind)
socket.Bind(address)
return socket
}
// the listen and server for mongrel, expects an address like this
// @addr = string config parameter like this:
// m2go.ListenAndServe("tcp://127.0.0.1:9555|tcp://127.0.0.1:9556|54c6755b-9628-40a4-9a2d-cc82a816345e", handler)
func ListenAndServe(addr string, handler http.Handler) {
var Context zmq.Context
var SocketIn zmq.Socket
var SocketOut zmq.Socket
var hasExited bool
var err error
m2addr := strings.Split(addr, "|") //
log.Printf("m2go serving %s\n", addr)
/*
Connection to ZMQ setup
*/
connect := func() {
if Context, err = zmq.NewContext(); err != nil {
panic("No ZMQ Context?")
}
// listen for incoming requests
if SocketIn, err = Context.NewSocket(zmq.PULL); err != nil {
panic("No ZMQ Socket?")
}
SocketIn.Connect(m2addr[0])
if SocketOut, err = Context.NewSocket(zmq.PUB); err != nil {
panic("No ZMQ Socket Outbound??")
}
// outbound response on a different channel
SocketOut.SetSockOptString(zmq.IDENTITY, m2addr[2])
//socket.SetSockOptString(zmq.SUBSCRIBE, filter)
SocketOut.Connect(m2addr[1])
}
connect()
handleResponse := func(response []byte) {
SocketOut.Send(response, 0)
}
stopper := func() {
if !hasExited {
hasExited = true
SocketOut.Close()
SocketIn.Close()
Context.Close()
}
}
defer stopper()
for {
// each inbound request
m2data, err := SocketIn.Recv(0)
//log.Println(string(m2data))
if err != nil {
log.Println("ZMQ Socket Input accept error ", err.Error())
} else {
go HandleM2Request(m2data, handleResponse, handler)
}
}
log.Print("after close of runner")
}
