Function BenchmarkWebSocketFanOut

websocket_test.go:1865–1904

(b *testing.B)

}

func BenchmarkWebSocketFanOut(b *testing.B) {
	ensureServer()
	token := signUpAndGetToken(b)

	const numSubs = 10

	topicName := fmt.Sprintf("bench-fo-%d", time.Now().UnixNano())
	wsPub := dialWS(b, token)
	defer wsPub.Close()

	sendJSON(b, wsPub, wsPayload{Id: nextReqId(), Method: "create-topicName", Payload: map[string]interface{}{"name": topicName}})
	recvJSON(b, wsPub, 5*time.Second) // create response
	time.Sleep(300 * time.Millisecond)

	subs := make([]*websocket.Conn, numSubs)
	for i := 0; i < numSubs; i++ {
		subs[i] = dialWS(b, token)
		sendJSON(b, subs[i], wsPayload{Id: nextReqId(), Method: "subscribe", Payload: map[string]interface{}{"topicName": topicName}})
		recvJSON(b, subs[i], 5*time.Second) // subscribe ack
	}

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		sendJSON(b, wsPub, wsPayload{
			Method: "new-message",
			Payload: map[string]interface{}{
				"topicName": topicName,
				"message":   map[string]interface{}{"seq": i},
			},
		})
		for _, sub := range subs {
			recvJSON(b, sub, 10*time.Second)
		}
	}
	b.StopTimer()

	for _, s := range subs {
		s.Close()
	}
}

func BenchmarkWebSocketConnect(b *testing.B) {
	ensureServer()