com.couchbase.client.core.endpoint.QueryEndpoint Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of core-io Show documentation
Show all versions of core-io Show documentation
The official Couchbase JVM Core IO Library
/*
* Copyright (c) 2018 Couchbase, Inc.
*
* 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.
*/
package com.couchbase.client.core.endpoint;
import com.couchbase.client.core.annotation.Stability;
import com.couchbase.client.core.cnc.events.endpoint.EndpointDisconnectDelayedEvent;
import com.couchbase.client.core.cnc.events.endpoint.EndpointDisconnectResumedEvent;
import com.couchbase.client.core.deps.io.netty.channel.ChannelPipeline;
import com.couchbase.client.core.deps.io.netty.handler.codec.http.HttpClientCodec;
import com.couchbase.client.core.io.netty.query.QueryHandlerSwitcher;
import com.couchbase.client.core.service.ServiceContext;
import com.couchbase.client.core.service.ServiceType;
public class QueryEndpoint extends BaseEndpoint {
/**
* should close chanel when outstanding requests are done.
*/
private volatile boolean closeWhenDone = false;
public QueryEndpoint(final ServiceContext ctx, final String hostname, final int port) {
super(hostname, port, ctx.environment().ioEnvironment().queryEventLoopGroup().get(),
ctx, ctx.environment().ioConfig().queryCircuitBreakerConfig(), ServiceType.QUERY, false);
}
@Override
protected PipelineInitializer pipelineInitializer() {
return new QueryPipelineInitializer(context());
}
public static class QueryPipelineInitializer implements PipelineInitializer {
private final EndpointContext endpointContext;
QueryPipelineInitializer(EndpointContext endpointContext) {
this.endpointContext = endpointContext;
}
@Override
public void init(BaseEndpoint endpoint, ChannelPipeline pipeline) {
pipeline.addLast(new HttpClientCodec());
pipeline.addLast(QueryHandlerSwitcher.SWITCHER_IDENTIFIER, new QueryHandlerSwitcher(endpoint, endpointContext));
}
}
@Override
public synchronized void disconnect() {
if (this.outstandingRequests() > 0) {
closeWhenDone();
} else {
super.disconnect();
}
}
private void closeWhenDone() {
closeWhenDone = true;
endpointContext.get().environment().eventBus().publish(new EndpointDisconnectDelayedEvent(endpointContext.get()));
}
@Stability.Internal
@Override
public synchronized void markRequestCompletion() {
super.markRequestCompletion();
if (closeWhenDone && outstandingRequests() <= 0) {
endpointContext.get().environment().eventBus().publish(new EndpointDisconnectResumedEvent(endpointContext.get()));
closeChannel(this.channel);
closeWhenDone = false;
}
}
}