func NewPubsubClient(config PubsubConfig) (*PubsubClient, error) {
context, err := zmq4.NewContext()
var pub *zmq4.Socket
var sub *zmq4.Socket
if err != nil {
return nil, err
}
if sub, err = context.NewSocket(zmq4.SUB); err != nil {
return nil, err
}
if pub, err = context.NewSocket(zmq4.PUSH); err != nil {
return nil, err
}
if err = pub.Connect(config.PubAddress); err != nil {
return nil, err
}
if err = sub.Connect(config.SubAddress); err != nil {
return nil, err
}
return &PubsubClient{pub, sub}, nil
}
func Example_multiple_contexts() {
chQuit := make(chan interface{})
chReactor := make(chan bool)
addr1 := "tcp://127.0.0.1:9997"
addr2 := "tcp://127.0.0.1:9998"
serv_ctx1, err := zmq.NewContext()
if checkErr(err) {
return
}
serv1, err := serv_ctx1.NewSocket(zmq.REP)
if checkErr(err) {
return
}
err = serv1.Bind(addr1)
if checkErr(err) {
return
}
defer func() {
serv1.Close()
serv_ctx1.Term()
}()
serv_ctx2, err := zmq.NewContext()
if checkErr(err) {
return
}
serv2, err := serv_ctx2.NewSocket(zmq.REP)
if checkErr(err) {
return
}
err = serv2.Bind(addr2)
if checkErr(err) {
return
}
defer func() {
serv2.Close()
serv_ctx2.Term()
}()
new_service := func(sock *zmq.Socket, addr string) {
socket_handler := func(state zmq.State) error {
msg, err := sock.RecvMessage(0)
if checkErr(err) {
return err
}
_, err = sock.SendMessage(addr, msg)
if checkErr(err) {
return err
}
return nil
}
quit_handler := func(interface{}) error {
return errors.New("quit")
}
defer func() {
chReactor <- true
}()
reactor := zmq.NewReactor()
reactor.AddSocket(sock, zmq.POLLIN, socket_handler)
reactor.AddChannel(chQuit, 1, quit_handler)
err = reactor.Run(100 * time.Millisecond)
fmt.Println(err)
}
go new_service(serv1, addr1)
go new_service(serv2, addr2)
time.Sleep(time.Second)
// default context
sock1, err := zmq.NewSocket(zmq.REQ)
if checkErr(err) {
return
}
sock2, err := zmq.NewSocket(zmq.REQ)
if checkErr(err) {
return
}
err = sock1.Connect(addr1)
if checkErr(err) {
return
}
err = sock2.Connect(addr2)
if checkErr(err) {
return
}
_, err = sock1.SendMessage(addr1)
if checkErr(err) {
return
}
_, err = sock2.SendMessage(addr2)
if checkErr(err) {
return
}
msg, err := sock1.RecvMessage(0)
fmt.Println(err, msg)
msg, err = sock2.RecvMessage(0)
fmt.Println(err, msg)
err = sock1.Close()
if checkErr(err) {
return
}
err = sock2.Close()
if checkErr(err) {
return
}
// non-default contexts
ctx1, err := zmq.NewContext()
if checkErr(err) {
return
}
ctx2, err := zmq.NewContext()
if checkErr(err) {
return
}
sock1, err = ctx1.NewSocket(zmq.REQ)
if checkErr(err) {
return
}
sock2, err = ctx2.NewSocket(zmq.REQ)
if checkErr(err) {
return
}
err = sock1.Connect(addr1)
if checkErr(err) {
return
}
err = sock2.Connect(addr2)
if checkErr(err) {
return
}
_, err = sock1.SendMessage(addr1)
if checkErr(err) {
return
}
_, err = sock2.SendMessage(addr2)
if checkErr(err) {
return
}
msg, err = sock1.RecvMessage(0)
fmt.Println(err, msg)
msg, err = sock2.RecvMessage(0)
fmt.Println(err, msg)
err = sock1.Close()
if checkErr(err) {
return
}
err = sock2.Close()
if checkErr(err) {
return
}
err = ctx1.Term()
if checkErr(err) {
return
}
err = ctx2.Term()
if checkErr(err) {
return
}
// close(chQuit) doesn't work because the reactor removes closed channels, instead of acting on them
chQuit <- true
<-chReactor
chQuit <- true
<-chReactor
fmt.Println("Done")
// Output:
// <nil> [tcp://127.0.0.1:9997 tcp://127.0.0.1:9997]
// <nil> [tcp://127.0.0.1:9998 tcp://127.0.0.1:9998]
// <nil> [tcp://127.0.0.1:9997 tcp://127.0.0.1:9997]
// <nil> [tcp://127.0.0.1:9998 tcp://127.0.0.1:9998]
// quit
// quit
// Done
}
