/* | |
* Copyright 1999-2011 Alibaba Group. | |
* | |
* 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 com.alibaba.dubbo.remoting.telnet.support; | |
import com.alibaba.dubbo.common.Constants; | |
import com.alibaba.dubbo.common.extension.ExtensionLoader; | |
import com.alibaba.dubbo.remoting.Channel; | |
import com.alibaba.dubbo.remoting.RemotingException; | |
import com.alibaba.dubbo.remoting.telnet.TelnetHandler; | |
import com.alibaba.dubbo.remoting.transport.ChannelHandlerAdapter; | |
/** | |
* TelnetHandlerDispather | |
* | |
* @author william.liangf | |
*/ | |
public class TelnetHandlerAdapter extends ChannelHandlerAdapter implements TelnetHandler { | |
private final ExtensionLoader<TelnetHandler> extensionLoader = ExtensionLoader.getExtensionLoader(TelnetHandler.class); | |
public String telnet(Channel channel, String message) throws RemotingException { | |
String prompt = channel.getUrl().getParameterAndDecoded(Constants.PROMPT_KEY, Constants.DEFAULT_PROMPT); | |
boolean noprompt = message.contains("--no-prompt"); | |
message = message.replace("--no-prompt", ""); | |
StringBuilder buf = new StringBuilder(); | |
message = message.trim(); | |
String command; | |
if (message.length() > 0) { | |
int i = message.indexOf(' '); | |
if (i > 0) { | |
command = message.substring(0, i).trim(); | |
message = message.substring(i + 1).trim(); | |
} else { | |
command = message; | |
message = ""; | |
} | |
} else { | |
command = ""; | |
} | |
if (command.length() > 0) { | |
if (extensionLoader.hasExtension(command)) { | |
try { | |
String result = extensionLoader.getExtension(command).telnet(channel, message); | |
if (result == null) { | |
return null; | |
} | |
buf.append(result); | |
} catch (Throwable t) { | |
buf.append(t.getMessage()); | |
} | |
} else { | |
buf.append("Unsupported command: "); | |
buf.append(command); | |
} | |
} | |
if (buf.length() > 0) { | |
buf.append("\r\n"); | |
} | |
if (prompt != null && prompt.length() > 0 && ! noprompt) { | |
buf.append(prompt); | |
} | |
return buf.toString(); | |
} | |
} |