コード例 #1
0
ファイル: chansocket_test.go プロジェクト: abligh/ghostfish
func runWrite(t *testing.T, num int, c zmqchan.ChanSocket) {
	tx := 0
	rx := 0
	srcMsg := [][]byte{[]byte("Hello"), []byte("World")}
	txchan := c.GetTxChan()
	for {
		select {
		case msg, ok := <-c.GetRxChan():
			if !ok {
				t.Fatal("Cannot read from main channel")
			}
			rx++
			log.Printf("MAIN: got message %d", rx)
			if !msgEqual(msg, srcMsg) {
				t.Fatal("Messages do not match")
			}
			if rx >= num {
				log.Println("MAIN: done")
				return
			}
		case txchan <- srcMsg:
			log.Println("MAIN: sent message")
			tx++
			if tx >= num {
				txchan = nil
			}
		case <-time.After(1 * time.Second):
			t.Fatal("Timeout in runWrite")
		}
	}
}
コード例 #2
0
ファイル: chansocket_test.go プロジェクト: abligh/ghostfish
func runEcho(t *testing.T, num int, c zmqchan.ChanSocket) {
	for {
		select {
		case msg, ok := <-c.GetRxChan():
			if !ok {
				t.Fatal("Cannot read from echo channel")
			}
			log.Println("ECHO: got message")
			c.GetTxChan() <- msg
			log.Println("ECHO: sent message")
			num--
			if num <= 0 {
				log.Println("ECHO: done")
				return
			}
		case <-time.After(1 * time.Second):
			t.Fatal("Timeout in runEcho")
		}
	}
}
コード例 #3
0
ファイル: chansocket_test.go プロジェクト: abligh/ghostfish
func TestZmqChanSocket(t *testing.T) {

	var sb, sc *zmq.Socket
	var cb, cc zmqchan.ChanSocket
	var err error
	num := 10

	defer func() {
		if sb != nil {
			sb.SetLinger(0)
			sb.Close()
		}
		if sc != nil {
			sc.SetLinger(0)
			sc.Close()
		}
		if cb != nil {
			log.Println("MAIN: Close")
			cb.Close()
		}
		if cc != nil {
			log.Println("ECHO: Close")
			cc.Close()
		}
		log.Println("BOTH: Exit")
	}()

	if sb, err = zmq.NewSocket(zmq.PAIR); err != nil {
		t.Fatal("NewSocket:", err)
	}

	if sc, err = zmq.NewSocket(zmq.PAIR); err != nil {
		t.Fatal("NewSocket:", err)
	}

	if err = sb.Bind("tcp://127.0.0.1:9737"); err != nil {
		t.Fatal("sb.Bind:", err)
	}

	if err = sc.Connect("tcp://127.0.0.1:9737"); err != nil {
		t.Fatal("sc.Connect:", err)
	}

	if cb, err = zmqchan.NewZmqChanSocket(sb, 0, 0); err != nil {
		t.Fatal("sb.NewZmqChanSocket:", err)
	}
	sb = nil // don't access this or close it on defer

	if cc, err = zmqchan.NewZmqChanSocket(sc, 0, 0); err != nil {
		t.Fatal("sb.NewZmqChanSocket:", err)
	}
	sc = nil // don't access this or close it on defer

	var wg sync.WaitGroup

	wg.Add(1)
	go func() {
		runEcho(t, num, cc)
		wg.Done()
	}()
	wg.Add(1)
	go func() {
		runWrite(t, num, cb)
		wg.Done()
	}()
	wg.Wait()
	log.Println("BOTH: done")
}