// 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 (bt *busTest) Init(t *testing.T, addr string) bool { var err error bt.resp = make(map[uint32]uint32) bt.start = make(map[uint32]bool) bt.send = 0 bt.nstart = 0 if bt.Sock, err = bus.NewSocket(); err != nil { bt.Errorf("NewSocket(): %v", err) return false } return bt.T.Init(t, addr) }
func node(args []string) { var sock mangos.Socket var err error var msg []byte var x int if sock, err = bus.NewSocket(); err != nil { die("bus.NewSocket: %s", err) } sock.AddTransport(ipc.NewTransport()) sock.AddTransport(tcp.NewTransport()) if err = sock.Listen(args[2]); err != nil { die("sock.Listen: %s", err.Error()) } // wait for everyone to start listening time.Sleep(time.Second) for x = 3; x < len(args); x++ { if err = sock.Dial(args[x]); err != nil { die("socket.Dial: %s", err.Error()) } } // wait for everyone to join time.Sleep(time.Second) fmt.Printf("%s: SENDING '%s' ONTO BUS\n", args[1], args[1]) if err = sock.Send([]byte(args[1])); err != nil { die("sock.Send: %s", err.Error()) } for { if msg, err = sock.Recv(); err != nil { die("sock.Recv: %s", err.Error()) } fmt.Printf("%s: RECEIVED \"%s\" FROM BUS\n", args[1], string(msg)) } }