Function TestAddEndpoint_Concurrency

micro/test/service_test.go:1024–1081

(t *testing.T)

}

func TestAddEndpoint_Concurrency(t *testing.T) {
	s := RunServerOnPort(-1)
	defer s.Shutdown()

	nc, err := nats.Connect(s.ClientURL())
	if err != nil {
		t.Fatalf("Expected to connect to server, got %v", err)
	}
	defer nc.Close()

	ctx := context.Background()

	handler := func(ctx context.Context, req micro.Request) {
		req.RespondJSON(map[string]any{"hello": "world"})
	}
	config := micro.Config{
		Name:    "test_service",
		Version: "0.1.0",
	}

	srv, err := micro.AddService(nc, config)
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}
	defer srv.Stop()

	res := make(chan error, 10)
	wg := sync.WaitGroup{}
	wg.Add(10)

	// now add a few endpoints concurrently
	// and make sure they are added successfully
	// and there is no race
	for i := 0; i < 10; i++ {
		go func(i int) {
			wg.Wait()
			res <- srv.AddEndpoint(fmt.Sprintf("test%d", i), micro.ContextHandler(ctx, handler))
		}(i)
		// after all goroutines are started, release the lock
	}
	wg.Add(-10)

	for i := 0; i < 10; i++ {
		select {
		case err := <-res:
			if err != nil {
				t.Fatalf("Unexpected error: %s", err)
			}
		case <-time.After(1 * time.Second):
			t.Fatalf("Timeout waiting for endpoint to be added")
		}
	}

	if len(srv.Info().Endpoints) != 10 {
		t.Fatalf("Expected 11 endpoints, got: %d", len(srv.Info().Endpoints))
	}

}