com.ikasoa.rpc.handler.impl.KryoProtocolHandlerImpl Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ikasoa-rpc Show documentation

ikasoa-rpc is a high performance RPC（Remote Procedure Call) framework.

There is a newer version: 0.3.3-BETA3

package com.ikasoa.rpc.handler.impl;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Optional;

import com.esotericsoftware.kryo.Kryo;
import com.esotericsoftware.kryo.io.Input;
import com.esotericsoftware.kryo.io.Output;
import com.ikasoa.rpc.handler.ProtocolHandler;
import com.ikasoa.rpc.handler.ReturnData;
import com.ikasoa.rpc.utils.Base64Util;

/**
 * 转换协议处理器Kryo实现
 * 
 * @author Larry
 * @version 0.1
 */
public class KryoProtocolHandlerImpl implements ProtocolHandler {

	private ReturnData resultData;

	private Kryo kryo;

	public KryoProtocolHandlerImpl(ReturnData resultData) {
		this.resultData = resultData;
		this.kryo = new Kryo();
	}

	@Override
	@SuppressWarnings("unchecked")
	public T strToArg(String str) {
		return (T) kryo.readObject(new Input(Base64Util.decode(str)), kryo.register(Object[].class).getType());
	}

	@Override
	@SuppressWarnings("unchecked")
	public String argToStr(T arg) {
		arg = Optional.ofNullable(arg).orElse((T) new Object[0]);
		Output output = new Output(1, 4096);
		kryo.writeObject(output, arg);
		byte[] bb = output.toBytes();
		output.flush();
		return Base64Util.encode(bb);
	}

	@Override
	public String resultToStr(R result) {
		if (result == null)
			return String.valueOf(V);
		Output output = new Output(1, 4096);
		kryo.writeObject(output, result);
		byte[] bb = output.toBytes();
		output.flush();
		return Base64Util.encode(bb);
	}

	@Override
	@SuppressWarnings("unchecked")
	public R strToResult(String str) {
		if (String.valueOf(V).equals(str))
			return null;
		return resultData.isArray()
				? (R) kryo.readObject(new Input(Base64Util.decode(str)),
						kryo.register((new ArrayList<>()).getClass()).getType())
				: resultData.isMap()
						? (R) kryo.readObject(new Input(Base64Util.decode(str)),
								kryo.register((new HashMap<>()).getClass()).getType())
						: (R) kryo.readObject(new Input(Base64Util.decode(str)),
								kryo.register(resultData.getClassType()).getType());
	}

	@Override
	public Throwable strToThrowable(String str) {
		// TODO: 该实现暂不对异常返回做处理
		return null;
	}

	@Override
	public ReturnData getReturnData() {
		return resultData;
	}

}