org.yx.rpc.log.PlainRpcLogHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sumk Show documentation
A quick developing framewort for internet company
There is a newer version: 4.0.2
/**
 * Copyright (C) 2016 - 2030 youtongluan.
 *
 * 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 org.yx.rpc.log;

import static org.yx.conf.AppInfo.LN;

import org.slf4j.Logger;
import org.yx.exception.BizException;
import org.yx.log.Log;
import org.yx.rpc.RpcSettings;
import org.yx.rpc.client.Req;
import org.yx.rpc.client.RpcResult;
import org.yx.rpc.codec.ReqParamType;
import org.yx.rpc.codec.Request;
import org.yx.rpc.server.Response;
import org.yx.util.S;

public class PlainRpcLogHandler implements RpcLogHandler {

	protected boolean isLogEnable(Logger logger, long totalTime, Exception e) {
		if (e != null) {

			return logger.isErrorEnabled();
		}

		if (logger.isDebugEnabled()) {
			return true;
		}

		return (logger.isInfoEnabled() && totalTime >= RpcSettings.infoTime())
				|| (logger.isWarnEnabled() && totalTime >= RpcSettings.warnTime());
	}

	protected void appendParam(StringBuilder sb, Req req) {
		if (req.hasFeature(ReqParamType.REQ_PARAM_JSON)) {
			sb.append("   param(json): ").append(req.getJsonedParam());
		} else {
			sb.append("   param(array): ").append(S.json().toJson(req.getParamArray()));
		}
	}

	@Override
	public void clientLog(RpcLog rpcLog) {
		if (RpcSettings.isClientLogDisable() || rpcLog == null) {
			return;
		}
		Req req = rpcLog.getReq();
		String api = req == null ? null : req.getApi();
		if (api == null || api.isEmpty() || api.startsWith("$")) {
			return;
		}
		Logger logger = Log.get("sumk.rpc.log.client");
		long totalTime = rpcLog.getReceiveTime() - req.getStart();
		RpcResult result = rpcLog.getResult();
		Exception e = result != null ? result.exception() : null;
		if (!this.isLogEnable(logger, totalTime, e)) {
			return;
		}
		StringBuilder sb = new StringBuilder(64);
		sb.append(api).append("   server:").append(rpcLog.getServer()).append("   totalTime:").append(totalTime)
				.append(LN);
		this.appendParam(sb, req);
		if (result != null) {
			if (e == null) {
				sb.append(LN).append("   result: ").append(result.json());
			}
		}

		if (e != null) {
			logger.error(sb.toString(), e);
		} else if (totalTime >= RpcSettings.warnTime()) {
			logger.warn(sb.toString());
		} else if (totalTime >= RpcSettings.infoTime()) {
			logger.info(sb.toString());
		} else {
			logger.debug(sb.toString());
		}
	}

	@Override
	public void serverLog(Request req, Response resp) {
		if (RpcSettings.isServerLogDisable()) {
			return;
		}
		Logger logger = Log.get("sumk.rpc.log.server");
		long totalTime = -1;
		Exception e = null;
		if (resp != null) {
			totalTime = resp.serviceInvokeMilTime();
			e = resp.exception();
		}
		if (!this.isLogEnable(logger, totalTime, e)) {
			return;
		}
		StringBuilder sb = new StringBuilder(64);
		if (req != null) {
			sb.append(req.getApi()).append("   serverTime:").append(totalTime).append(LN);
			this.appendParam(sb, req);
		}
		String json = resp != null ? resp.json() : null;
		if (e == null) {
			sb.append(LN).append("   result: ").append(json);
		}

		if (resp != null && resp.isSuccess()) {
			if (totalTime >= RpcSettings.warnTime()) {
				logger.warn(sb.toString());
			} else if (totalTime > RpcSettings.infoTime()) {
				logger.info(sb.toString());
			} else {
				logger.debug(sb.toString());
			}
			return;
		}
		if (BizException.class.isInstance(e)) {
			logger.warn(sb.toString(), e.toString());
		} else {
			logger.error(sb.toString(), e);
		}
	}

}