124 lines
5.6 KiB
Groovy
124 lines
5.6 KiB
Groovy
/*
|
|
* Copyright 2020, 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.
|
|
*/
|
|
import io.netty.bootstrap.ServerBootstrap
|
|
import io.netty.buffer.ByteBuf
|
|
import io.netty.buffer.Unpooled
|
|
import io.netty.channel.ChannelHandlerContext
|
|
import io.netty.channel.ChannelInitializer
|
|
import io.netty.channel.ChannelPipeline
|
|
import io.netty.channel.EventLoopGroup
|
|
import io.netty.channel.SimpleChannelInboundHandler
|
|
import io.netty.channel.nio.NioEventLoopGroup
|
|
import io.netty.channel.socket.nio.NioServerSocketChannel
|
|
import io.netty.handler.codec.http.DefaultFullHttpResponse
|
|
import io.netty.handler.codec.http.FullHttpResponse
|
|
import io.netty.handler.codec.http.HttpHeaderNames
|
|
import io.netty.handler.codec.http.HttpRequest
|
|
import io.netty.handler.codec.http.HttpResponseStatus
|
|
import io.netty.handler.codec.http.HttpServerCodec
|
|
import io.netty.handler.logging.LogLevel
|
|
import io.netty.handler.logging.LoggingHandler
|
|
import io.netty.util.CharsetUtil
|
|
import io.opentelemetry.auto.test.base.HttpServerTest
|
|
|
|
import static io.netty.handler.codec.http.HttpHeaderNames.CONTENT_LENGTH
|
|
import static io.netty.handler.codec.http.HttpHeaderNames.CONTENT_TYPE
|
|
import static io.netty.handler.codec.http.HttpResponseStatus.INTERNAL_SERVER_ERROR
|
|
import static io.netty.handler.codec.http.HttpVersion.HTTP_1_1
|
|
import static io.opentelemetry.auto.test.base.HttpServerTest.ServerEndpoint.ERROR
|
|
import static io.opentelemetry.auto.test.base.HttpServerTest.ServerEndpoint.EXCEPTION
|
|
import static io.opentelemetry.auto.test.base.HttpServerTest.ServerEndpoint.NOT_FOUND
|
|
import static io.opentelemetry.auto.test.base.HttpServerTest.ServerEndpoint.QUERY_PARAM
|
|
import static io.opentelemetry.auto.test.base.HttpServerTest.ServerEndpoint.REDIRECT
|
|
import static io.opentelemetry.auto.test.base.HttpServerTest.ServerEndpoint.SUCCESS
|
|
|
|
class Netty41ServerTest extends HttpServerTest<EventLoopGroup> {
|
|
|
|
static final LoggingHandler LOGGING_HANDLER = new LoggingHandler(SERVER_LOGGER.name, LogLevel.DEBUG)
|
|
|
|
@Override
|
|
EventLoopGroup startServer(int port) {
|
|
def eventLoopGroup = new NioEventLoopGroup()
|
|
|
|
ServerBootstrap bootstrap = new ServerBootstrap()
|
|
.group(eventLoopGroup)
|
|
.handler(LOGGING_HANDLER)
|
|
.childHandler([
|
|
initChannel: { ch ->
|
|
ChannelPipeline pipeline = ch.pipeline()
|
|
pipeline.addFirst("logger", LOGGING_HANDLER)
|
|
|
|
def handlers = [new HttpServerCodec()]
|
|
handlers.each { pipeline.addLast(it) }
|
|
pipeline.addLast([
|
|
channelRead0 : { ctx, msg ->
|
|
if (msg instanceof HttpRequest) {
|
|
def uri = URI.create((msg as HttpRequest).uri)
|
|
ServerEndpoint endpoint = ServerEndpoint.forPath(uri.path)
|
|
ctx.write controller(endpoint) {
|
|
ByteBuf content = null
|
|
FullHttpResponse response = null
|
|
switch (endpoint) {
|
|
case SUCCESS:
|
|
case ERROR:
|
|
content = Unpooled.copiedBuffer(endpoint.body, CharsetUtil.UTF_8)
|
|
response = new DefaultFullHttpResponse(HTTP_1_1, HttpResponseStatus.valueOf(endpoint.status), content)
|
|
break
|
|
case QUERY_PARAM:
|
|
content = Unpooled.copiedBuffer(uri.query, CharsetUtil.UTF_8)
|
|
response = new DefaultFullHttpResponse(HTTP_1_1, HttpResponseStatus.valueOf(endpoint.status), content)
|
|
break
|
|
case REDIRECT:
|
|
response = new DefaultFullHttpResponse(HTTP_1_1, HttpResponseStatus.valueOf(endpoint.status))
|
|
response.headers().set(HttpHeaderNames.LOCATION, endpoint.body)
|
|
break
|
|
case EXCEPTION:
|
|
throw new Exception(endpoint.body)
|
|
default:
|
|
content = Unpooled.copiedBuffer(NOT_FOUND.body, CharsetUtil.UTF_8)
|
|
response = new DefaultFullHttpResponse(HTTP_1_1, HttpResponseStatus.valueOf(NOT_FOUND.status), content)
|
|
break
|
|
}
|
|
response.headers().set(CONTENT_TYPE, "text/plain")
|
|
if (content) {
|
|
response.headers().set(CONTENT_LENGTH, content.readableBytes())
|
|
}
|
|
return response
|
|
}
|
|
}
|
|
},
|
|
exceptionCaught : { ChannelHandlerContext ctx, Throwable cause ->
|
|
ByteBuf content = Unpooled.copiedBuffer(cause.message, CharsetUtil.UTF_8)
|
|
FullHttpResponse response = new DefaultFullHttpResponse(HTTP_1_1, INTERNAL_SERVER_ERROR, content)
|
|
response.headers().set(CONTENT_TYPE, "text/plain")
|
|
response.headers().set(CONTENT_LENGTH, content.readableBytes())
|
|
ctx.write(response)
|
|
},
|
|
channelReadComplete: { it.flush() }
|
|
] as SimpleChannelInboundHandler)
|
|
}
|
|
] as ChannelInitializer).channel(NioServerSocketChannel)
|
|
bootstrap.bind(port).sync()
|
|
|
|
return eventLoopGroup
|
|
}
|
|
|
|
@Override
|
|
void stopServer(EventLoopGroup server) {
|
|
server?.shutdownGracefully()
|
|
}
|
|
}
|