org.jfaster.mango.parser.ASTJDBCIterableParameter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mango Show documentation
mango is a dao framework.
The newest version!
/*
 * Copyright 2014 mango.jfaster.org
 *
 * The Mango Project licenses this file to you 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 org.jfaster.mango.parser;

import org.jfaster.mango.binding.BindingParameter;
import org.jfaster.mango.binding.BindingParameterInvoker;
import org.jfaster.mango.binding.InvocationContext;
import org.jfaster.mango.type.TypeHandler;
import org.jfaster.mango.util.IterObj;
import org.jfaster.mango.util.Strings;
import org.jfaster.mango.util.jdbc.JdbcType;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * 可迭代参数
 *
 * @author ash
 */
public class ASTJDBCIterableParameter extends AbstractRenderableNode implements ParameterBean {

  private BindingParameter bindingParameter;
  private BindingParameterInvoker bindingParameterInvoker;
  private TypeHandler typeHandler;

  private String propertyOfMapper; // "msg_id in (:1)"中的msg_id

  public ASTJDBCIterableParameter(int i) {
    super(i);
  }

  public ASTJDBCIterableParameter(Parser p, int i) {
    super(p, i);
  }

  public void init(String str) {
    Pattern p = Pattern.compile("in\\s*\\(\\s*:(\\w+)((\\.\\w+)*)(\\@\\w+)?\\s*\\)", Pattern.CASE_INSENSITIVE);
    Matcher m = p.matcher(str);
    if (!m.matches()) {
      throw new IllegalStateException("Can't compile string '" + str + "'");
    }
    String group1 = m.group(1);
    String group2 = m.group(2);
    String group4 = m.group(4);
    String parameterName = group1;
    String propertyPath = Strings.isNotEmpty(group2) ?
        group2.substring(1) :
        "";
    JdbcType jdbcType = null;
    String jdbcTypeStr = null;
    if (group4 != null) {
      try {
        jdbcTypeStr = group4.substring(1).toUpperCase();
        jdbcType = JdbcType.valueOf(jdbcTypeStr);
      } catch (Exception e) {
        throw new IllegalStateException("error jdbc type '" + jdbcTypeStr + "'");
      }
    }
    bindingParameter = BindingParameter.create(parameterName, propertyPath, jdbcType);
  }

  @Override
  public BindingParameter getBindingParameter() {
    return bindingParameter;
  }

  @Override
  public void setBindingParameter(BindingParameter bindingParameter) {
    this.bindingParameter = bindingParameter;
  }

  @Override
  public boolean render(InvocationContext context) {
    if (bindingParameterInvoker == null) {
      throw new NullPointerException("invoker must set");
    }
    Object objs = context.getNullableBindingValue(bindingParameterInvoker);
    if (objs == null) {
      throw new NullPointerException("value of " +
          bindingParameter.getFullName() + " can't be null");
    }
    IterObj iterObj = new IterObj(objs);
    if (iterObj.isEmpty()) {
      if (iterObj.isCollection()) {
        throw new EmptyCollectionException("value of " +
            bindingParameter.getFullName() + " can't be empty");
      } else {
        throw new EmptyArrayException("value of " +
            bindingParameter.getFullName() + " can't be empty");
      }
    }
    context.writeToSqlBuffer("in (");
    int t = 0;
    for (Object obj : iterObj) {
      context.appendToArgs(obj, typeHandler);
      if (t == 0) {
        context.writeToSqlBuffer("?");
      } else {
        context.writeToSqlBuffer(",?");
      }
      t++;
    }
    context.writeToSqlBuffer(")");
    return true;
  }

  @Override
  public String toString() {
    return super.toString() + "{" +
        "fullName=" + getFullName() + ", " +
        "parameterName=" + bindingParameter.getParameterName() + ", " +
        "propertyName=" + bindingParameter.getPropertyName() +
        "}";
  }

  @Override
  public Object jjtAccept(ParserVisitor visitor, Object data) {
    return visitor.visit(this, data);
  }

  @Override
  public String getFullName() {
    return bindingParameter.getFullName();
  }

  @Override
  public void setBindingParameterInvoker(BindingParameterInvoker bindingParameterInvoker) {
    this.bindingParameterInvoker = bindingParameterInvoker;
  }

  public void setTypeHandler(TypeHandler typeHandler) {
    this.typeHandler = typeHandler;
  }

  public String getPropertyOfMapper() {
    return propertyOfMapper;
  }

  public void setPropertyOfMapper(String propertyOfMapper) {
    this.propertyOfMapper = propertyOfMapper;
  }

}