func main() {
var iaddr string
flag.StringVar(&iaddr, "iaddr", "tcp://localhost:5555", "input data port")
var oaddr string
flag.StringVar(&oaddr, "oaddr", "tcp://localhost:5556", "output data port")
flag.Parse()
isck, err := pull.NewSocket()
if err != nil {
log.Fatalf("error creating a nanomsg socket: %v\n", err)
}
defer isck.Close()
isck.AddTransport(ipc.NewTransport())
isck.AddTransport(tcp.NewTransport())
osck, err := push.NewSocket()
if err != nil {
log.Fatalf("error creating output port: %v\n", err)
}
defer osck.Close()
osck.AddTransport(ipc.NewTransport())
osck.AddTransport(tcp.NewTransport())
log.Printf("dialing %s ...\n", iaddr)
err = isck.Dial(iaddr)
if err != nil {
log.Fatalf("error dialing: %v\n", err)
}
log.Printf("dialing %s ... [done]\n", iaddr)
log.Printf("dialing %s ...\n", oaddr)
err = osck.Dial(oaddr)
if err != nil {
log.Fatalf("error dialing: %v\n", err)
}
log.Printf("dialing %s ... [done]\n", oaddr)
for {
msg, err := isck.Recv()
if err != nil {
if err == io.EOF || err == mangos.ErrClosed {
log.Printf("received EOF: %v\n", err)
break
}
log.Fatalf("error receiving data: %v\n", err)
}
log.Printf("recv: %v\n", string(msg))
omsg := bytes.Repeat(msg, 2)
err = osck.Send(omsg)
if err != nil {
log.Fatalf("error sending data: %v\n", err)
}
}
}
// NewSocket allocates a new Socket. The Socket is the handle used to
// access the underlying library.
func NewSocket(d Domain, p Protocol) (*Socket, error) {
var s Socket
var err error
s.proto = p
s.dom = d
switch p {
case PUB:
s.sock, err = pub.NewSocket()
case SUB:
s.sock, err = sub.NewSocket()
case PUSH:
s.sock, err = push.NewSocket()
case PULL:
s.sock, err = pull.NewSocket()
case REQ:
s.sock, err = req.NewSocket()
case REP:
s.sock, err = rep.NewSocket()
case SURVEYOR:
s.sock, err = surveyor.NewSocket()
case RESPONDENT:
s.sock, err = respondent.NewSocket()
case PAIR:
s.sock, err = pair.NewSocket()
case BUS:
s.sock, err = bus.NewSocket()
default:
err = mangos.ErrBadProto
}
if err != nil {
return nil, err
}
switch d {
case AF_SP:
case AF_SP_RAW:
err = s.sock.SetOption(mangos.OptionRaw, true)
default:
err = errBadDomain
}
if err != nil {
s.sock.Close()
return nil, err
}
// Compat mode sockets should timeout on send if we don't have any pipes
if err = s.sock.SetOption(mangos.OptionWriteQLen, 0); err != nil {
s.sock.Close()
return nil, err
}
s.rto = -1
s.sto = -1
all.AddTransports(s.sock)
return &s, nil
}
func (pt *PushTest) Init(t *testing.T, addr string) bool {
var err error
if pt.Sock, err = push.NewSocket(); err != nil {
pt.Errorf("NewSocket(): %v", err)
return false
}
return pt.T.Init(t, addr)
}
func newSocket(url, protocol string) (sock mangos.Socket, err error) {
var str string
switch protocol {
case "bus":
sock, err = bus.NewSocket()
str = "bus.NewSocket: "
case "pair":
sock, err = pair.NewSocket()
str = "pair.NewSocket: "
case "push":
sock, err = push.NewSocket()
str = "push.NewSocket: "
case "pull":
sock, err = pull.NewSocket()
str = "pull.NewSocket: "
case "pub":
sock, err = pub.NewSocket()
str = "pub.NewSocket: "
case "sub":
sock, err = sub.NewSocket()
str = "sub.NewSocket: "
case "request":
sock, err = req.NewSocket()
str = "request.NewSocket: "
case "reply":
sock, err = rep.NewSocket()
str = "reply.NewSocket: "
case "surveyor":
sock, err = surveyor.NewSocket()
str = "surveyor.NewSocket: "
case "respondent":
sock, err = respondent.NewSocket()
str = "respondent.NewSocket: "
}
if err != nil {
s := fmt.Sprintf(str+" %s", err)
return nil, errors.New(s)
}
sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
return sock, nil
}
func node1(url string, msg string) {
var sock mangos.Socket
var err error
if sock, err = push.NewSocket(); err != nil {
die("can't get new push socket: %s", err.Error())
}
sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
if err = sock.Dial(url); err != nil {
die("can't dial on push socket: %s", err.Error())
}
fmt.Printf("NODE1: SENDING \"%s\"\n", msg)
if err = sock.Send([]byte(msg)); err != nil {
die("can't send message on push socket: %s", err.Error())
}
sock.Close()
}