// 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 *pubTest) Init(t *testing.T, addr string) bool { pt.pubidx = 0 var err error if pt.Sock, err = pub.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 newEvtPublisher(addr string) (mangos.Socket, error) { log.Info("sensor: creating event publisher...") socket, err := pub.NewSocket() if err != nil { return nil, err } if err := socket.SetOption(mangos.OptionSendDeadline, time.Second*3); err != nil { socket.Close() return nil, err } //socket.AddTransport(ipc.NewTransport()) socket.AddTransport(tcp.NewTransport()) if err = socket.Listen(addr); err != nil { socket.Close() return nil, err } return socket, nil }
func server(url string) { var sock mangos.Socket var err error if sock, err = pub.NewSocket(); err != nil { die("can't get new pub socket: %s", err) } sock.AddTransport(ipc.NewTransport()) sock.AddTransport(tcp.NewTransport()) if err = sock.Listen(url); err != nil { die("can't listen on pub socket: %s", err.Error()) } for { // Could also use sock.RecvMsg to get header d := date() fmt.Printf("SERVER: PUBLISHING DATE %s\n", d) if err = sock.Send([]byte(d)); err != nil { die("Failed publishing: %s", err.Error()) } time.Sleep(time.Second) } }
func testPubNonBlock(t *testing.T, addr string, tran mangos.Transport) { maxqlen := 2 timeout := time.Second / 2 rp, err := pub.NewSocket() if err != nil { t.Errorf("Failed to make PUB: %v", err) return } defer rp.Close() rp.AddTransport(tran) // Now try setting the option err = rp.SetOption(mangos.OptionWriteQLen, maxqlen) if err != nil { t.Errorf("Failed set WriteQLen: %v", err) return } // At this point, we can issue requests on rq, and read them from rp. if err = rp.SetOption(mangos.OptionSendDeadline, timeout); err != nil { t.Errorf("Failed set recv deadline") return } if err = rp.Listen(addr); err != nil { t.Errorf("Failed listen: %v", err) return } msg := []byte{'A', 'B', 'C'} for i := 0; i < maxqlen*2; i++ { t.Logf("Sending #%d", i) if err := rp.Send(msg); err != nil { t.Errorf("Failed to send: %v", err) } } }