113 lines
3.6 KiB
C#
113 lines
3.6 KiB
C#
// <copyright file="TestHttpServer.cs" company="OpenTelemetry Authors">
|
|
// Copyright The OpenTelemetry 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.
|
|
// </copyright>
|
|
using System;
|
|
using System.Net;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace OpenTelemetry.Internal.Test
|
|
{
|
|
internal class TestHttpServer
|
|
{
|
|
private static readonly Random GlobalRandom = new Random();
|
|
|
|
public static IDisposable RunServer(Action<HttpListenerContext> action, out string host, out int port)
|
|
{
|
|
host = "localhost";
|
|
port = 0;
|
|
RunningServer server = null;
|
|
|
|
var retryCount = 5;
|
|
while (retryCount > 0)
|
|
{
|
|
try
|
|
{
|
|
port = GlobalRandom.Next(2000, 5000);
|
|
server = new RunningServer(action, host, port);
|
|
server.Start();
|
|
break;
|
|
}
|
|
catch (HttpListenerException)
|
|
{
|
|
retryCount--;
|
|
}
|
|
}
|
|
|
|
return server;
|
|
}
|
|
|
|
private class RunningServer : IDisposable
|
|
{
|
|
private readonly Task httpListenerTask;
|
|
private readonly HttpListener listener;
|
|
private readonly AutoResetEvent initialized = new AutoResetEvent(false);
|
|
|
|
public RunningServer(Action<HttpListenerContext> action, string host, int port)
|
|
{
|
|
this.listener = new HttpListener();
|
|
|
|
this.listener.Prefixes.Add($"http://{host}:{port}/");
|
|
this.listener.Start();
|
|
|
|
this.httpListenerTask = new Task(async () =>
|
|
{
|
|
while (true)
|
|
{
|
|
try
|
|
{
|
|
var ctxTask = this.listener.GetContextAsync();
|
|
|
|
this.initialized.Set();
|
|
|
|
action(await ctxTask.ConfigureAwait(false));
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
if (ex is ObjectDisposedException
|
|
|| (ex is HttpListenerException httpEx && httpEx.ErrorCode == 995))
|
|
{
|
|
// Listener was closed before we got into GetContextAsync or
|
|
// Listener was closed while we were in GetContextAsync.
|
|
break;
|
|
}
|
|
|
|
throw;
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
public void Start()
|
|
{
|
|
this.httpListenerTask.Start();
|
|
this.initialized.WaitOne();
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
try
|
|
{
|
|
this.listener?.Stop();
|
|
}
|
|
catch (ObjectDisposedException)
|
|
{
|
|
// swallow this exception just in case
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|