rpad.java

来自「mysql集群」· Java 代码 · 共 85 行

/*****************************************************************************
      SQLJEP - Java SQL Expression Parser 0.2
      November 1 2006
         (c) Copyright 2006, Alexey Gaidukov
      SQLJEP Author: Alexey Gaidukov

      SQLJEP is based on JEP 2.24 (http://www.singularsys.com/jep/)
           (c) Copyright 2002, Nathan Funk
 
      See LICENSE.txt for license information.
*****************************************************************************/

package com.meidusa.amoeba.sqljep.function;


import com.meidusa.amoeba.sqljep.function.PostfixCommand;
import com.meidusa.amoeba.sqljep.ASTFunNode;
import com.meidusa.amoeba.sqljep.JepRuntime;
import com.meidusa.amoeba.sqljep.ParseException;

public class Rpad extends PostfixCommand {
	private static final String PARAM_EXCEPTION = "Length in rpad shoud be integer";

	final public int getNumberOfParameters() {
		return -1;
	}
	
	public Comparable<?>[] evaluate(ASTFunNode node, JepRuntime runtime) throws ParseException {
		node.childrenAccept(runtime.ev, null);
		int num = node.jjtGetNumChildren();
		if (num == 2) {
			Comparable<?>  param2 = runtime.stack.pop();
			Comparable<?>  param1 = runtime.stack.pop();
			return new Comparable<?>[]{param1,param2," "};
		}
		else if (num == 3) {
			Comparable<?>  param3 = runtime.stack.pop();
			Comparable<?>  param2 = runtime.stack.pop();
			Comparable<?>  param1 = runtime.stack.pop();
			return new Comparable<?>[]{param1,param2,param3};
		} else {
			// remove all parameters from stack and push null
			removeParams(runtime.stack, num);
			throw new ParseException(PARAMS_NUMBER+" for rpad");
		}
	}

	public static String rpad(Comparable<?>  param1, Comparable<?>  param2, Comparable<?>  param3) throws ParseException {
		if (param1 == null || param2 == null || param3 == null) {
			return null;
		}
		String inputStr = param1.toString();
		String chars = param3.toString();
		int length;
		try {
			length = getInteger(param2);
		} catch (ParseException e) {
			throw new ParseException(PARAM_EXCEPTION);
		}
		if (length == inputStr.length()) {
			return inputStr;
		}
		else if (length < inputStr.length()) {
			return inputStr.substring(0, length);
		} else {
			length -= inputStr.length();
			int count = length / chars.length();
			int remainder = length % chars.length();
			StringBuilder output = new StringBuilder();
			output.append(inputStr);
			for (int i = 0; i < count; i++) {
				output.append(chars);
			}
			output.append(chars.substring(0, remainder));
			return output.toString();
		}
	}

	public Comparable<?> getResult(Comparable<?>... comparables)
			throws ParseException {
		return rpad(comparables[0],comparables[1],comparables[2]);
	}
}