mirror of
https://github.com/gogrlx/nats-server.git
synced 2026-04-17 03:24:40 -07:00
87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
// Copyright 2018 The NATS Authors
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
// +build !race
|
|
|
|
package test
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/nats-io/gnatsd/server"
|
|
"github.com/nats-io/go-nats"
|
|
)
|
|
|
|
// As we look to improve high fanout situations make sure we
|
|
// have a test that checks ordering for all subscriptions from a single subscriber.
|
|
func TestHighFanoutOrdering(t *testing.T) {
|
|
opts := &server.Options{Host: "127.0.0.1", Port: server.RANDOM_PORT}
|
|
|
|
s := RunServer(opts)
|
|
defer s.Shutdown()
|
|
|
|
url := fmt.Sprintf("nats://%s", s.Addr())
|
|
|
|
const (
|
|
nconns = 100
|
|
nsubs = 100
|
|
npubs = 500
|
|
)
|
|
|
|
// make unique
|
|
subj := nats.NewInbox()
|
|
|
|
var wg sync.WaitGroup
|
|
wg.Add(nconns * nsubs)
|
|
|
|
for i := 0; i < nconns; i++ {
|
|
nc, err := nats.Connect(url)
|
|
if err != nil {
|
|
t.Fatalf("Expected a successful connect on %d, got %v\n", i, err)
|
|
}
|
|
|
|
nc.SetErrorHandler(func(c *nats.Conn, s *nats.Subscription, e error) {
|
|
t.Fatalf("Got an error %v for %+v\n", s, err)
|
|
})
|
|
|
|
ec, _ := nats.NewEncodedConn(nc, nats.DEFAULT_ENCODER)
|
|
|
|
for y := 0; y < nsubs; y++ {
|
|
expected := 0
|
|
ec.Subscribe(subj, func(n int) {
|
|
if n != expected {
|
|
t.Fatalf("Expected %d but received %d\n", expected, n)
|
|
}
|
|
expected++
|
|
if expected >= npubs {
|
|
wg.Done()
|
|
}
|
|
})
|
|
}
|
|
ec.Flush()
|
|
defer ec.Close()
|
|
}
|
|
|
|
nc, _ := nats.Connect(url)
|
|
ec, _ := nats.NewEncodedConn(nc, nats.DEFAULT_ENCODER)
|
|
|
|
for i := 0; i < npubs; i++ {
|
|
ec.Publish(subj, i)
|
|
}
|
|
defer ec.Close()
|
|
|
|
wg.Wait()
|
|
}
|