Java tutorial
/** * Copyright 2011-2017 Asakusa Framework Team. * * 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.asakusafw.dag.compiler.jdbc.windgate; import static com.asakusafw.dag.compiler.codegen.AsmUtil.*; import java.util.List; import java.util.function.Function; import java.util.function.Supplier; import org.objectweb.asm.ClassWriter; import org.objectweb.asm.MethodVisitor; import org.objectweb.asm.Opcodes; import org.objectweb.asm.Type; import com.asakusafw.dag.compiler.codegen.AsmUtil.LocalVarRef; import com.asakusafw.dag.compiler.codegen.ClassGeneratorContext; import com.asakusafw.dag.compiler.codegen.SupplierGenerator; import com.asakusafw.dag.compiler.jdbc.PreparedStatementAdapterGenerator; import com.asakusafw.dag.compiler.model.ClassData; import com.asakusafw.dag.runtime.jdbc.operation.JdbcOutputProcessor; import com.asakusafw.dag.runtime.jdbc.util.WindGateJdbcDirect; import com.asakusafw.lang.compiler.model.description.ClassDescription; import com.asakusafw.lang.utils.common.Arguments; import com.asakusafw.lang.utils.common.Lang; import com.asakusafw.lang.utils.common.Optionals; import com.asakusafw.lang.utils.common.Tuple; /** * Generates {@link JdbcOutputProcessor} using {@link WindGateJdbcDirect} API. * @since 0.4.0 */ public final class WindGateJdbcOutputProcessorGenerator { private static final Type TRUNCATE_BUILDER = typeOf(WindGateJdbcDirect.TruncateBuilder.class); private static final Type OUTPUT_BUILDER = typeOf(WindGateJdbcDirect.OutputBuilder.class); private static final String CATEGORY = "jdbc.windgate"; //$NON-NLS-1$ private static final String HINT = "OutputProcessor"; //$NON-NLS-1$ private WindGateJdbcOutputProcessorGenerator() { return; } /** * Generates {@link JdbcOutputProcessor} class. * @param context the current context * @param specs the target operation specs * @return the generated class data */ public static ClassData generate(ClassGeneratorContext context, List<Spec> specs) { Arguments.requireNonNull(context); Arguments.requireNonNull(specs); return generate(context, specs, context.getClassName(CATEGORY, HINT)); } /** * Generates {@link JdbcOutputProcessor} class. * @param context the current context * @param specs the target operation specs * @param target the target class * @return the generated class data */ public static ClassData generate(ClassGeneratorContext context, List<Spec> specs, ClassDescription target) { Arguments.requireNonNull(context); Arguments.requireNonNull(specs); ClassWriter writer = newWriter(target, JdbcOutputProcessor.class); defineEmptyConstructor(writer, JdbcOutputProcessor.class, v -> { LocalVarRef self = new LocalVarRef(Opcodes.ALOAD, 0); self.load(v); specs.forEach(spec -> { truncate(context, v, spec); output(context, v, spec); }); v.visitInsn(Opcodes.POP); }); writer.visitEnd(); return new ClassData(target, writer::toByteArray); } private static void truncate(ClassGeneratorContext context, MethodVisitor v, Spec spec) { if (spec.initialize == false) { return; } getConst(v, spec.id); getConst(v, spec.model.getProfileName()); getConst(v, spec.model.getProfileName()); getConst(v, spec.model.getTableName()); getList(v, Lang.project(spec.model.getColumnMappings(), Tuple::left)); v.visitMethodInsn(Opcodes.INVOKESTATIC, typeOf(WindGateJdbcDirect.class).getInternalName(), "truncate", Type.getMethodDescriptor(TRUNCATE_BUILDER, typeOf(String.class), // profileName typeOf(String.class), // tableName typeOf(List.class)), // columnNames false); Lang.forEach(Optionals.of(spec.model.getCustomTruncate()), s -> { getConst(v, s); v.visitMethodInsn(Opcodes.INVOKEVIRTUAL, TRUNCATE_BUILDER.getInternalName(), "withCustomTruncate", //$NON-NLS-1$ Type.getMethodDescriptor(TRUNCATE_BUILDER, typeOf(String.class)), false); }); Lang.forEach(spec.model.getOptions(), s -> { getConst(v, s); v.visitMethodInsn(Opcodes.INVOKEVIRTUAL, TRUNCATE_BUILDER.getInternalName(), "withOption", //$NON-NLS-1$ Type.getMethodDescriptor(TRUNCATE_BUILDER, typeOf(String.class)), false); }); v.visitMethodInsn(Opcodes.INVOKEVIRTUAL, TRUNCATE_BUILDER.getInternalName(), "build", //$NON-NLS-1$ Type.getMethodDescriptor(typeOf(Function.class)), false); v.visitMethodInsn(Opcodes.INVOKEVIRTUAL, typeOf(JdbcOutputProcessor.class).getInternalName(), "initialize", //$NON-NLS-1$ Type.getMethodDescriptor(typeOf(JdbcOutputProcessor.class), typeOf(String.class), typeOf(String.class), typeOf(Function.class)), false); } private static void output(ClassGeneratorContext context, MethodVisitor v, Spec spec) { if (spec.output == false) { return; } getConst(v, spec.id); getConst(v, spec.model.getProfileName()); getConst(v, spec.model.getProfileName()); getConst(v, spec.model.getTableName()); getList(v, Lang.project(spec.model.getColumnMappings(), Tuple::left)); getNew(v, getAdapter(context, spec)); v.visitMethodInsn(Opcodes.INVOKESTATIC, typeOf(WindGateJdbcDirect.class).getInternalName(), "output", Type.getMethodDescriptor(OUTPUT_BUILDER, typeOf(String.class), // profileName typeOf(String.class), // tableName typeOf(List.class), // columnNames typeOf(Supplier.class)), // adapter false); Lang.forEach(spec.model.getOptions(), s -> { getConst(v, s); v.visitMethodInsn(Opcodes.INVOKEVIRTUAL, OUTPUT_BUILDER.getInternalName(), "withOption", //$NON-NLS-1$ Type.getMethodDescriptor(OUTPUT_BUILDER, typeOf(String.class)), false); }); v.visitMethodInsn(Opcodes.INVOKEVIRTUAL, OUTPUT_BUILDER.getInternalName(), "build", //$NON-NLS-1$ Type.getMethodDescriptor(typeOf(Function.class)), false); v.visitMethodInsn(Opcodes.INVOKEVIRTUAL, typeOf(JdbcOutputProcessor.class).getInternalName(), "output", //$NON-NLS-1$ Type.getMethodDescriptor(typeOf(JdbcOutputProcessor.class), typeOf(String.class), typeOf(String.class), typeOf(Function.class)), false); } private static ClassDescription getAdapter(ClassGeneratorContext context, Spec spec) { return SupplierGenerator.get(context, PreparedStatementAdapterGenerator.get(context, new PreparedStatementAdapterGenerator.Spec( spec.model.getDataType(), Lang.project(spec.model.getColumnMappings(), Tuple::right)))); } /** * Represents an operation spec for {@link JdbcOutputProcessor}. * @since 0.4.0 */ public static class Spec { final String id; final WindGateJdbcOutputModel model; final boolean initialize; final boolean output; /** * Creates a new instance. * @param id the output ID * @param model the output model */ public Spec(String id, WindGateJdbcOutputModel model) { this(id, model, true, true); } private Spec(String id, WindGateJdbcOutputModel model, boolean initialize, boolean output) { Arguments.requireNonNull(id); Arguments.requireNonNull(model); this.id = id; this.model = model; this.initialize = initialize; this.output = output; } /** * Returns a copy of this. * @param doOutput {@code true} if the operation can output data, otherwise {@code false} * @return the copy */ public Spec withOutput(boolean doOutput) { return new Spec(id, model, initialize, doOutput); } } }