Class ProxyServer

ZezeJava/ZezeJava/src/main/java/Zeze/Raft/ProxyServer.java:17–135

import Zeze.Util.OutObject;

public class ProxyServer extends Service {
	public static final String eProxyServerName = "Zeze.Raft.ProxyServer";
	private final int rpcTimeout;

	public ProxyServer(Config config, int rpcTimeout) {
		super(eProxyServerName, config);
		this.rpcTimeout = rpcTimeout;
		getSocketOptions().setOutputBufferMaxSize(100 * 1024 * 1024);
		getSocketOptions().setInputBufferMaxProtocolSize(100 * 1024 * 1024);
		RegisterProtocols();
	}

	public ProxyServer(Application zeze, int rpcTimeout) {
		super(eProxyServerName, zeze);
		this.rpcTimeout = rpcTimeout;
		getSocketOptions().setOutputBufferMaxSize(100 * 1024 * 1024);
		getSocketOptions().setInputBufferMaxProtocolSize(100 * 1024 * 1024);
		RegisterProtocols();
	}

	private void RegisterProtocols() {
		AddFactoryHandle(ProxyRequest.TypeId_, new ProtocolFactoryHandle<>(
				ProxyRequest::new,
				this::ProcessProxyRequest,
				TransactionLevel.None
		));
	}

	/**
	 * 把代理请求派发到指定的raft中执行。
	 *
	 * @param r ProxyRequest
	 */
	private long ProcessProxyRequest(ProxyRequest r) throws Exception {
		var raft = rafts.get(r.Argument.getRaftName());
		if (null == raft) {
			logger.warn("ProxyRequest: not found raftName={}, available={}", r.Argument.getRaftName(), rafts.keySet());
			return Procedure.ProviderNotExist;
		}
		var server = raft.getServer();
		var outFactoryHandle = new OutObject<ProtocolFactoryHandle<?>>();
		var p = Protocol.decode(server::findProtocolFactoryHandle, ByteBuffer.Wrap(r.Argument.getRpcBinary()), outFactoryHandle);
		if (null == p)
			return Procedure.NotImplement;
		var raftRpc = (RaftRpc<?, ?>)p;
		raftRpc.setProxyRequest(r);

		// 下面的流程从Raft.Server.dispatchProtocol的部分代码拷贝出，请参考原来的地方，进行比较。。
		if (raft.isWorkingLeader()) {
			if (raftRpc.getUnique().getRequestId() <= 0) {
				p.SendResultCode(Procedure.ErrorRequestId);
				return 0;
			}

			server.dispatchRaftRequest(p,
					() -> server.processRequest(p, outFactoryHandle.value),
					p.getClass().getName(),
					() -> p.SendResultCode(Procedure.RaftRetry),