cn.sanenen.sunutils.chain.SimpleMsgChain Maven / Gradle / Ivy

Go to download
package cn.sanenen.sunutils.chain;

import cn.hutool.core.lang.TypeReference;
import cn.hutool.core.map.MapUtil;
import cn.hutool.log.Log;
import com.alibaba.fastjson.JSON;
import org.apache.commons.chain.Chain;
import org.apache.commons.chain.Command;
import org.apache.commons.chain.Context;
import org.apache.commons.chain.impl.ChainBase;
import org.apache.commons.chain.impl.ContextBase;

import java.util.List;

/**
 * @author sun
 * @date 2020-12-21
 **/
public class SimpleMsgChain implements IMsgChain {
	private static final Log log = Log.get();
	private final Chain chain = new ChainBase();

	@Override
	public void setHandlers(List> iMsgHandlers) {
		for (IMsgHandler handler : iMsgHandlers) {
			chain.addCommand(new SimpleCommand<>(handler));
		}
	}

	@Override
	public void execute(T t) {
		ContextBase contextBase = new ContextBase();
		contextBase.put(IMsgChain.DATA_KEY, t);
		try {
			chain.execute(contextBase);
		} catch (Exception e) {
			log.error(e, "chain execute error. data:{}", JSON.toJSONString(t));
		} finally {
			contextBase.clear();
		}
	}

	private static class SimpleCommand extends TypeReference implements Command {
		private final IMsgHandler handler;

		public SimpleCommand(IMsgHandler handler) {
			this.handler = handler;
		}

		@Override
		public boolean execute(Context context) {
			return handler.process(MapUtil.get(context, IMsgChain.DATA_KEY, this));
		}
	}
}