com.taobao.api.internal.toplink.netcat.NetCatCommandServerChannelHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of alibaba-dingtalk-service-sdk Show documentation

dingtalk service sdk

The newest version!

package com.taobao.api.internal.toplink.netcat;

import java.util.HashMap;
import java.util.Map;

import com.taobao.api.internal.toplink.Logger;
import com.taobao.api.internal.toplink.LoggerFactory;
import com.taobao.api.internal.toplink.channel.ChannelContext;
import com.taobao.api.internal.toplink.channel.ChannelException;
import com.taobao.api.internal.toplink.channel.SimpleChannelHandler;
import com.taobao.api.internal.toplink.logging.LogUtil;

public class NetCatCommandServerChannelHandler extends SimpleChannelHandler {
	private Logger logger;
	private Map processors;

	public NetCatCommandServerChannelHandler() {
		this(LogUtil.getLoggerFactory(new Object()));
	}

	public NetCatCommandServerChannelHandler(LoggerFactory loggerFactory) {
		this.logger = loggerFactory.create(this);
		this.processors = new HashMap();
	}

	public void addProcessor(CommandProcessor processor) {
		this.processors.put(processor.getName(), processor);
	}

	public void onMessage(final ChannelContext context) {
		String line = (String) context.getMessage();

		if (this.logger.isDebugEnabled())
			this.logger.debug("command: ", line);

		if (line == null)
			return;

		String[] arr = line.trim().split(" ");

		if (arr.length == 0)
			return;
		CommandProcessor processor = this.processors.get(arr[0]);
		if (processor == null) {
			this.logger.warn("processor not found for command: %s", arr[0]);
			return;
		}

		NetCatOuputWriter writer = new NetCatOuputWriter() {
			public void write(String value) {
				byte[] data = (value + "\n").getBytes();
				try {
					context.reply(data, 0, data.length);
				} catch (ChannelException e) {
					logger.error(e);
				}
			}
		};
		
		try {
			processor.process(this.parseInput(arr, 1), writer);
		} catch (Exception e) {
			writer.write(e.getMessage());
		}
	}

	protected Map parseInput(String[] input, int from) {
		Map map = new HashMap();
		for (int i = from; i < input.length; i += 2) {
			if (i >= input.length)
				break;
			map.put(input[i].startsWith("-") ? input[i].substring(1) : input[i],
					i + 1 < input.length ? input[i + 1] : null);
		}
		return map;
	}
}