com.qq.tars.client.rpc.ServantProtocolInvoker Maven / Gradle / Ivy
/**
* Tencent is pleased to support the open source community by making Tars available.
*
* Copyright (C) 2016 THL A29 Limited, a Tencent company. All rights reserved.
*
* Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
* in compliance with the License. You may obtain a copy of the License at
*
* https://opensource.org/licenses/BSD-3-Clause
*
* 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.qq.tars.client.rpc;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.ThreadPoolExecutor;
import com.qq.tars.client.ServantProxyConfig;
import com.qq.tars.client.support.ClientPoolManager;
import com.qq.tars.client.util.ClientLogger;
import com.qq.tars.client.util.ParseTools;
import com.qq.tars.common.util.Constants;
import com.qq.tars.net.core.nio.SelectorManager;
import com.qq.tars.net.protocol.ProtocolFactory;
import com.qq.tars.rpc.common.Invoker;
import com.qq.tars.rpc.common.ProtocolInvoker;
import com.qq.tars.rpc.common.Url;
import com.qq.tars.rpc.common.util.concurrent.ConcurrentHashSet;
import com.qq.tars.rpc.exc.ClientException;
public abstract class ServantProtocolInvoker implements ProtocolInvoker {
protected final Class api;
protected final ServantProxyConfig servantProxyConfig;
protected final ThreadPoolExecutor threadPoolExecutor;
protected final ProtocolFactory protocolFactory;
protected volatile SelectorManager selectorManager = null;
protected final ConcurrentHashSet> allInvoker = new ConcurrentHashSet>();
public ServantProtocolInvoker(Class api, ServantProxyConfig config, ProtocolFactory protocolFactory,
ThreadPoolExecutor threadPoolExecutor) {
this.api = api;
this.servantProxyConfig = config;
this.threadPoolExecutor = threadPoolExecutor;
this.protocolFactory = protocolFactory;
this.initInvoker();
}
public abstract Invoker create(Class api, Url url) throws Exception;
public Collection> getInvokers() {
return Collections.unmodifiableCollection(allInvoker);
}
public void destroy() {
destroy(allInvoker);
this.selectorManager.stop();
}
public void refresh() {
ClientLogger.getLogger().info("try to refresh " + servantProxyConfig.getSimpleObjectName());
List> invokers = new ArrayList>();
for (Invoker invoker : allInvoker) {
invokers.add(invoker);
}
this.initInvoker();
destroy(invokers);
}
protected ServantClient[] getClients(Url url) throws IOException {
int connections = url.getParameter(Constants.TARS_CLIENT_CONNECTIONS, Constants.default_connections);
ServantClient[] clients = new ServantClient[connections];
for (int i = 0; i < clients.length; i++) {
clients[i] = initClient(url);
}
return clients;
}
protected ServantClient initClient(Url url) {
ServantClient client = null;
try {
boolean tcpNoDelay = url.getParameter(Constants.TARS_CLIENT_TCPNODELAY, false);
long connectTimeout = url.getParameter(Constants.TARS_CLIENT_CONNECTTIMEOUT, Constants.default_connect_timeout);
long syncTimeout = url.getParameter(Constants.TARS_CLIENT_SYNCTIMEOUT, Constants.default_sync_timeout);
long asyncTimeout = url.getParameter(Constants.TARS_CLIENT_ASYNCTIMEOUT, Constants.default_async_timeout);
boolean udpMode = url.getParameter(Constants.TARS_CLIENT_UDPMODE, false);
if (this.selectorManager == null) {
this.selectorManager = ClientPoolManager.getSelectorManager(this.protocolFactory, this.threadPoolExecutor, true, udpMode, this.servantProxyConfig);
}
client = new ServantClient(url.getHost(), url.getPort(), this.selectorManager, udpMode);
client.setConnectTimeout(connectTimeout);
client.setSyncTimeout(syncTimeout);
client.setAsyncTimeout(asyncTimeout);
client.setTcpNoDelay(tcpNoDelay);
} catch (Throwable e) {
throw new ClientException(servantProxyConfig.getSimpleObjectName(), "Fail to create client|" + url.toIdentityString() + "|" + e.getLocalizedMessage(), e);
}
return client;
}
protected void initInvoker() {
try {
ClientLogger.getLogger().info("try to init invoker|conf=" + servantProxyConfig.toString());
List list = ParseTools.parse(servantProxyConfig);
for (Url url : list) {
try {
boolean active = url.getParameter(Constants.TARS_CLIENT_ACTIVE, false);
if (active) {
ClientLogger.getLogger().info("try to init invoker|active=" + active + "|" + url.toIdentityString());
allInvoker.add(create(api, url));
} else {
ClientLogger.getLogger().info("inactive invoker can`t to init|active=" + active + "|" + url.toIdentityString());
}
} catch (Throwable e) {
ClientLogger.getLogger().error("error occurred on init invoker|" + url.toIdentityString(), e);
}
}
} catch (Throwable t) {
ClientLogger.getLogger().error("error occurred on init invoker|" + servantProxyConfig.getObjectName(), t);
}
}
private void destroy(Collection> invokers) {
for (Invoker invoker : invokers) {
if (invoker != null) {
ClientLogger.getLogger().info("destroy reference|" + invoker);
try {
allInvoker.remove(invoker);
invoker.destroy();
} catch (Throwable t) {
ClientLogger.getLogger().error("error occurred on destroy invoker|" + invoker, t);
}
}
}
}
}