TransactionalInstruction.java :  » UnTagged » androidannotations » com » googlecode » androidannotations » generation » Android Open Source

/*
 * Copyright 2010-2011 Pierre-Yves Ricau (py.ricau at gmail.com)
 * 
 * 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.googlecode.androidannotations.generation;

import java.util.List;

public class TransactionalInstruction extends AbstractInstruction {

  private static final String FORMAT = //
  "" + //
      "    @Override\n" + //
      "    public %s %s(%s) {\n" + //
      "        %s.beginTransaction();\n" + //
      "        try {\n" + //
      "            %s%s(%s);\n" + //
      "            %s.setTransactionSuccessful();\n" + //
      "            return%s;\n" + //
      "        } catch (RuntimeException e) {\n" + //
      "          Log.e(\"%s\", \"Error in transaction\", e);\n" + //
      "          throw e;\n" + //
      "        } finally {\n" + //
      "          %s.endTransaction();\n" + //
      "        }\n" + //
      "    }\n" + //
      "\n";

  private final String methodName;

  private final String className;

  private final List<String> methodArguments;

  private final List<String> methodParameters;

  private final String returnType;

  public TransactionalInstruction(String methodName, String className, List<String> methodArguments, List<String> methodParameters, String returnType) {
    this.methodName = methodName;
    this.className = className;
    this.methodArguments = methodArguments;
    this.methodParameters = methodParameters;
    this.returnType = returnType;
    addImports("android.util.Log");
  }

  @Override
  public String generate() {

    StringBuilder arguments = new StringBuilder();
    boolean first = true;
    for (String argument : methodArguments) {
      if (first) {
        first = false;
      } else {
        arguments.append(", ");
      }
      arguments.append(argument);
    }

    first = true;
    StringBuilder parameters = new StringBuilder();
    for (String parameter : methodParameters) {
      if (first) {
        first = false;
      } else {
        parameters.append(", ");
      }
      parameters.append(parameter);
    }

    String dbParameter = methodParameters.get(0);

    String returnedValue = returnType.equals("void") ? "" : returnType + " result = ";
    String returnStatement = returnType.equals("void") ? "" : " result";

    return String.format(FORMAT, returnType, methodName, arguments.toString(), dbParameter, returnedValue, methodName, parameters.toString(), dbParameter, returnStatement, className, dbParameter);
  }

}