com.baidu.jprotobuf.pbrpc.management.ServerStatus Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jprotobuf-rpc-core Show documentation
Show all versions of jprotobuf-rpc-core Show documentation
The core module of jprotobuf rpc socket implementation
/*
* Copyright 2002-2007 the original author or 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.
*/
package com.baidu.jprotobuf.pbrpc.management;
import static com.baidu.jprotobuf.pbrpc.management.HttpConstants.BOLD_FONT;
import static com.baidu.jprotobuf.pbrpc.management.HttpConstants.BOLD_FONT_END;
import static com.baidu.jprotobuf.pbrpc.management.HttpConstants.HTML_HEAD;
import static com.baidu.jprotobuf.pbrpc.management.HttpConstants.HTML_TAIL;
import static com.baidu.jprotobuf.pbrpc.management.HttpConstants.LINE_BREAK;
import static com.baidu.jprotobuf.pbrpc.management.HttpConstants.PRE_ENDS;
import static com.baidu.jprotobuf.pbrpc.management.HttpConstants.PRE_STARTS;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.baidu.jprotobuf.pbrpc.meta.MetaExportHelper;
import com.baidu.jprotobuf.pbrpc.meta.RpcServiceMeta;
import com.baidu.jprotobuf.pbrpc.meta.RpcServiceMetaList;
import com.baidu.jprotobuf.pbrpc.transport.RpcServer;
import com.baidu.jprotobuf.pbrpc.transport.RpcServerOptions;
/**
* Server status.
*
* @author xiemalin
* @since 3.1.0
*/
public class ServerStatus {
/** The Constant LOGGER. */
private static final Logger LOGGER = LoggerFactory.getLogger(ServerStatus.class);
/** The Constant SECONDS_IN_HOUR. */
private static final int SECONDS_IN_HOUR = 3600;
/** The Constant SECONDS_IN_DAY. */
private static final int SECONDS_IN_DAY = 86400;
/** The port. */
private int port;
/** The http port. */
private int httpPort;
/** The start time. */
private long startTime;
/** The rpc server. */
private RpcServer rpcServer;
/** The close. */
private transient boolean close = false;
private static transient boolean enabled = false;
/** The Constant REQUEST_COUNTS. */
public static final ConcurrentHashMap REQUEST_COUNTS =
new ConcurrentHashMap();
/** The Constant ASYNC_REQUEST. */
public static final BlockingQueue ASYNC_REQUEST = new LinkedBlockingQueue();
/** The es. */
private static ExecutorService es;
public static void Enabled() {
enabled = true;
}
/**
* Incr.
*
* @param serviceSignature the service signature
* @param timetook the timetook
*/
public static void incr(String serviceSignature, long timetook) {
if (!enabled) {
return;
}
ASYNC_REQUEST.add(new RequestInfo(timetook, serviceSignature));
}
/**
* Do incr.
*
* @param requestInfo the request info
*/
private static void doIncr(RequestInfo requestInfo) {
AtomicLong newOne = new AtomicLong();
AtomicLong oldOne = REQUEST_COUNTS.putIfAbsent(requestInfo.signature, newOne);
if (oldOne != null) {
newOne = oldOne;
}
newOne.incrementAndGet();
}
/**
* Instantiates a new server status.
*
* @param rpcServer the rpc server
*/
public ServerStatus(RpcServer rpcServer) {
super();
this.rpcServer = rpcServer;
startTime = rpcServer.getStartTime();
port = rpcServer.getInetSocketAddress().getPort();
httpPort = rpcServer.getRpcServerOptions().getHttpServerPort();
es = Executors.newSingleThreadExecutor();
es.execute(new Runnable() {
@Override
public void run() {
while (!close) {
RequestInfo requestInfo;
try {
requestInfo = ASYNC_REQUEST.poll(1, TimeUnit.SECONDS);
if (requestInfo != null) {
doIncr(requestInfo);
}
} catch (InterruptedException e) {
if (LOGGER.isDebugEnabled()) {
LOGGER.debug(e.getMessage(), e);
}
}
}
}
});
}
/*
* (non-Javadoc)
*
* @see java.lang.Object#toString()
*/
@Override
public String toString() {
StringBuilder ret = new StringBuilder();
ret.append(HTML_HEAD);
ret.append("Server online: ").append(getOnlineDuration(startTime)).append(LINE_BREAK);
ret.append("RPC port:").append(port).append(LINE_BREAK);
ret.append("Http management port:").append(httpPort).append(LINE_BREAK);
ret.append("Chunk enabled").append(LINE_BREAK);
ret.append("Compress enabled(Gzip Snappy)").append(LINE_BREAK);
ret.append("Attachment enabled").append(LINE_BREAK);
if (rpcServer.getEs() != null) {
ret.append("--------------Thread status----------------").append(LINE_BREAK);
ret.append("Max task count:").append(rpcServer.getEs().getMaximumPoolSize()).append(LINE_BREAK);
ret.append("Running task count:").append(rpcServer.getEs().getActiveCount()).append(LINE_BREAK);
ret.append("Waiting task count:").append(rpcServer.getEs().getQueue().size()).append(LINE_BREAK);
ret.append(LINE_BREAK).append(LINE_BREAK);
}
ret.append(PRE_STARTS);
ret.append("--------------properties info(").append(RpcServerOptions.class.getDeclaredFields().length)
.append(")----------------").append(LINE_BREAK);
ret.append(rpcServer.getRpcServerOptions());
ret.append(LINE_BREAK).append(LINE_BREAK);
RpcServiceMetaList exportRPCMeta = MetaExportHelper.exportRPCMeta(port);
List metaList = exportRPCMeta.getRpcServiceMetas();
if (metaList != null) {
ret.append("--------------RPC service list(").append(metaList.size()).append(") ----------------")
.append(LINE_BREAK);
for (RpcServiceMeta rpcServiceMeta : metaList) {
ret.append(BOLD_FONT).append("Service name:").append(rpcServiceMeta.getServiceName())
.append(LINE_BREAK);
ret.append("Method name:").append(rpcServiceMeta.getMethodName()).append(BOLD_FONT_END)
.append(LINE_BREAK);
ret.append("Request IDL:").append(LINE_BREAK).append(rpcServiceMeta.getInputProto()).append(LINE_BREAK);
ret.append("Response IDL:").append(LINE_BREAK).append(rpcServiceMeta.getOutputProto())
.append(LINE_BREAK);
ret.append(LINE_BREAK);
}
}
ret.append(LINE_BREAK).append(LINE_BREAK);
ret.append("--------------protobuf idl info ----------------").append(LINE_BREAK);
ret.append(exportRPCMeta.getTypesIDL());
ret.append(exportRPCMeta.getRpcsIDL());
ret.append(PRE_ENDS);
ret.append(HTML_TAIL);
ret.append("--------------Request Info ----------------").append(LINE_BREAK);
Map copy = new HashMap(REQUEST_COUNTS);
Iterator> iterator = copy.entrySet().iterator();
ret.append("service request count next = iterator.next();
ret.append("");
ret.append("").append(next.getKey()).append(" ");
ret.append("").append(next.getValue()).append(" ");
ret.append(" ");
}
ret.append("
");
return ret.toString();
}
/**
* Gets the online duration.
*
* @param startTime the start time
* @return the online duration
*/
private String getOnlineDuration(long startTime) {
StringBuilder ret = new StringBuilder();
long ms = (System.currentTimeMillis() - startTime) / 1000;
long days = ms / SECONDS_IN_DAY;
long hours = (ms % SECONDS_IN_DAY) / SECONDS_IN_HOUR;
long seconds = ((ms % SECONDS_IN_DAY) % SECONDS_IN_HOUR);
ret.append(days).append(" days ").append(hours).append(" hours ").append(seconds).append(" seconds");
return ret.toString();
}
/**
* The Class RequestInfo.
*/
private static class RequestInfo {
/** The took. */
private long took;
/** The signature. */
private String signature;
/**
* Instantiates a new request info.
*
* @param took the took
* @param signature the signature
*/
public RequestInfo(long took, String signature) {
super();
this.took = took;
this.signature = signature;
}
}
/**
* Close.
*/
public void close() {
close = true;
if (es != null) {
es.shutdown();
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy