Shell

1. 代码

package com.yuan.common.shell;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicBoolean;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.yuan.common.util.AssertUtil;
import com.yuan.common.util.StringUtil;

public class Shell {

    private static final Logger log = LoggerFactory.getLogger(Shell.class);
   
    public static String COMMAND_SEPARATOR = " ";
    public static String ARGS_SEPARATOR = ",";
   
    private Map<String, ShellCommand> commandMap = new HashMap<String, ShellCommand>();
    private AtomicBoolean running = new AtomicBoolean(true);
    private String shellName = "shell";
   
    public Shell(){
        addCommand(new ExitCommand());
        addCommand(new HelpCommand());
    }
   
    public Shell(String shellName){
        this();
        this.shellName = shellName;
    }
   
    public void addCommand(ShellCommand command){
        commandMap.put(command.getCommandText(), command);
    }
   
    public void start(){
       
        cmdLoop();
    }
   
    public void shutdown(){
        running.set(false);
       
    }
   
    /**
     * 命令循环，等待用户输入命令
     */
    protected void cmdLoop(){
        while(running.get()){
            System.out.print(shellName + ">");
            String line = readLine();
            if(StringUtil.hasText(line)){
                String[] commandArr = line.split(COMMAND_SEPARATOR);
                if(commandMap.containsKey(commandArr[0])){
                    String args = null;
                    if(commandArr.length > 1){
                        args = commandArr[1];
                    }
                    try {
                        commandMap.get(commandArr[0]).exec(processArgs(args));
                    } catch (Exception e) {
                        show(e.getMessage());
                        e.printStackTrace();
                    }
                }else{
                    show("命令" + commandArr[0] + "不存在！");
                }
            }else{
                help();
            }
        }
    }
   
    protected List<String> processArgs(String args){
        if(!StringUtil.hasText(args)){
            return null;
        }
        String[] argArr = args.split(ARGS_SEPARATOR);
       
        return Arrays.asList(argArr);
    }
   
    /**
     * TAB命令智能补全
     */
    protected void tabBash(){
       
    }
   
    /**
     * 读取控制台输入，阻塞方法
     * @return String
     */
    protected String readLine(){
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
       
        try {
            return br.readLine();
        } catch (IOException e) {
            log.warn(e.getMessage(), e);
        }
        return null;
    }
   
    /**
     * 向控制台打印消息
     * @param message String
     */
    protected void show(String message){
       
        System.out.println(message);
    }
   
    public void help(){
        for(String command : commandMap.keySet()){
            show(command);
        }
    }
   
    private class ExitCommand implements ShellCommand{

        @Override
        public void exec(List<String> args) throws Exception {
            shutdown();
           
        }

        @Override
        public String getCommandText() {
           
            return "exit";
        }

        @Override
        public String getHelp() {
           
            return "";
        }
       
    }
    private class HelpCommand implements ShellCommand{

        @Override
        public void exec(List<String> args) throws Exception {
            if(AssertUtil.notEmpty(args)){
                String command = args.get(0);
                if(commandMap.containsKey(command)){
                    show(commandMap.get(command).getHelp());
                }else{
                    show("命令" + command + "不存在！");
                }
            }else{
                help();
            }
           
        }

        @Override
        public String getCommandText() {
           
            return "help";
        }

        @Override
        public String getHelp() {
           
            return "";
        }
       
    }
}

 

 

package com.yuan.common.shell;

import java.util.List;

public interface ShellCommand {

    public void exec(List<String> args)throws Exception;
    public String getCommandText();
    public String getHelp();
   
}

 

2. 测试

 

public static void main(String[] args) {
        Shell shell = new Shell("yuan");
        shell.addCommand(new ShellCommand() {
            public String getHelp() {
                return null;
            }
            public String getCommandText() {
                return "sum";
            }
            public void exec(List<String> args) throws Exception {
                Integer sum = 0;
                for(String arg : args){
                    sum += Integer.parseInt(arg);
                }
                System.out.println("sum = " + sum);
            }
        });
        shell.start();
    }