Java tutorial
/* * Copyright 2008-2009 MOPAS(Ministry of Public Administration and Security). * * 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 egovframework.rte.itl.webservice.service.impl; import static javax.jws.WebParam.Mode.INOUT; import static javax.jws.WebParam.Mode.OUT; import static org.objectweb.asm.Opcodes.ACC_ABSTRACT; import static org.objectweb.asm.Opcodes.ACC_INTERFACE; import static org.objectweb.asm.Opcodes.ACC_PUBLIC; import static org.objectweb.asm.Opcodes.ALOAD; import static org.objectweb.asm.Opcodes.ARETURN; import static org.objectweb.asm.Opcodes.ASTORE; import static org.objectweb.asm.Opcodes.CHECKCAST; import static org.objectweb.asm.Opcodes.DUP; import static org.objectweb.asm.Opcodes.GETFIELD; import static org.objectweb.asm.Opcodes.INVOKEINTERFACE; import static org.objectweb.asm.Opcodes.INVOKESPECIAL; import static org.objectweb.asm.Opcodes.NEW; import static org.objectweb.asm.Opcodes.POP; import static org.objectweb.asm.Opcodes.RETURN; import static org.objectweb.asm.Opcodes.V1_5; import java.lang.reflect.Array; import java.math.BigDecimal; import java.math.BigInteger; import java.util.Calendar; import java.util.Collection; import java.util.HashMap; import java.util.Map; import java.util.Map.Entry; import javax.xml.ws.Holder; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.objectweb.asm.AnnotationVisitor; import org.objectweb.asm.ClassWriter; import org.objectweb.asm.FieldVisitor; import org.objectweb.asm.MethodVisitor; import egovframework.rte.itl.integration.type.ListType; import egovframework.rte.itl.integration.type.PrimitiveType; import egovframework.rte.itl.integration.type.RecordType; import egovframework.rte.itl.integration.type.Type; import egovframework.rte.itl.webservice.EgovWebServiceMessageHeader; import egovframework.rte.itl.webservice.service.EgovWebServiceClassLoader; import egovframework.rte.itl.webservice.service.ServiceEndpointInfo; import egovframework.rte.itl.webservice.service.ServiceEndpointInterfaceInfo; import egovframework.rte.itl.webservice.service.ServiceParamInfo; /** * ? ? ? ClassLoader ? * <p> * <b>NOTE:</b> ? Type, * ServiceEndpointInterface, ServiceEndpoint ? class * ? ClassLoader?. * @author ? * @since 2009.06.01 * @version 1.0 * @see <pre> * == ?(Modification Information) == * * ? ? * ------- -------- --------------------------- * 2009.06.01 ? ? * * </pre> */ public class EgovWebServiceClassLoaderImpl extends ClassLoader implements EgovWebServiceClassLoader { private Log LOG = LogFactory.getLog(this.getClass()); public static final String BASE_PACKAGE_NAME = "egovframework.rte.itl.webservice.generated"; public static final String CLASS_NAME_PREFIX = BASE_PACKAGE_NAME + "." + "EgovType"; public static final String SERVICE_ENDPOINT_CLASS_NAME_POSTFIX = "ServiceImpl"; public static final String SERVICE_ENDPOINT_INTERFACE_CLASS_NAME_POSTFIX = "Service"; public static final String NAME_OF_SERVICE_BRIDGE_CLASS = "egovframework/rte/itl/webservice/service/ServiceBridge"; public static final String DESC_OF_SERVICE_BRIDGE_CLASS = "L" + NAME_OF_SERVICE_BRIDGE_CLASS + ";"; public static final String FIELD_NAME_OF_SERVICE_BRIDGE = "serviceBridge"; @SuppressWarnings("serial") private static final Map<PrimitiveType, Class<?>> primitiveClasses = new HashMap<PrimitiveType, Class<?>>() { { put(PrimitiveType.BOOLEAN, boolean.class); put(PrimitiveType.STRING, String.class); put(PrimitiveType.BYTE, byte.class); put(PrimitiveType.SHORT, short.class); put(PrimitiveType.INTEGER, int.class); put(PrimitiveType.LONG, long.class); put(PrimitiveType.BIGINTEGER, BigInteger.class); put(PrimitiveType.FLOAT, float.class); put(PrimitiveType.DOUBLE, double.class); put(PrimitiveType.BIGDECIMAL, BigDecimal.class); put(PrimitiveType.CALENDAR, Calendar.class); } }; @SuppressWarnings("serial") private static final Map<Class<?>, Class<?>> wrapperClasses = new HashMap<Class<?>, Class<?>>() { { put(boolean.class, Boolean.class); put(byte.class, Byte.class); put(short.class, Short.class); put(int.class, Integer.class); put(long.class, Long.class); put(float.class, Float.class); put(double.class, Double.class); } }; /** * Default Constructor */ protected EgovWebServiceClassLoaderImpl() { super(); } /** * Constructor * @param parent * parent */ protected EgovWebServiceClassLoaderImpl(ClassLoader parent) { super(parent); } public String getFieldNameOfServiceBridge() { return FIELD_NAME_OF_SERVICE_BRIDGE; } public Class<?> loadClass(final Type type) throws ClassNotFoundException { LOG.debug("loadClass of Type (" + type + ")"); if (type == EgovWebServiceMessageHeader.TYPE) { LOG.debug("Type is EgovWebServiceMessageHeader."); return EgovWebServiceMessageHeader.class; } if (type instanceof PrimitiveType) { LOG.debug("Type is a Primitive Type"); Class<?> clazz = primitiveClasses.get((PrimitiveType) type); if (clazz == null) { LOG.error("No such primitive type"); throw new ClassNotFoundException(); } return clazz; } else if (type instanceof ListType) { LOG.debug("Type is a List Type"); ListType listType = (ListType) type; Class<?> elementClass = loadClass(listType.getElementType()); return Array.newInstance(elementClass, 0).getClass(); } else if (type instanceof RecordType) { LOG.debug("Type is a Record Type"); RecordType recordType = (RecordType) type; String className = getRecordTypeClassName(recordType.getName()); try { LOG.debug("Check the class \"" + className + "\" is already loaded."); return loadClass(className); } catch (ClassNotFoundException e) { LOG.debug("Create a new class \"" + className + "\""); byte[] byteCode = createRecordClass(className, recordType); return defineClass(className, byteCode, 0, byteCode.length); } } LOG.error("Type is invalid"); throw new ClassNotFoundException(); } public Class<?> loadClass(final ServiceEndpointInfo serviceEndpointInfo) throws ClassNotFoundException { LOG.debug("loadClass of ServiceEndpointInfo (" + serviceEndpointInfo + ")"); // load interface class LOG.debug("load Interface Class"); loadClass((ServiceEndpointInterfaceInfo) serviceEndpointInfo); String className = getServiceEndpointClassName(serviceEndpointInfo.getServiceName()); try { LOG.debug("Check the class \"" + className + "\" is already loaded"); return loadClass(className); } catch (ClassNotFoundException e) { LOG.debug("Create a new class \"" + className + "\""); byte[] byteCode = createServiceEndpointClass(serviceEndpointInfo); return defineClass(className, byteCode, 0, byteCode.length); } } public Class<?> loadClass(final ServiceEndpointInterfaceInfo serviceEndpointInterfaceInfo) throws ClassNotFoundException { LOG.debug("loadClass of ServiceEndpointInterfaceInfo (" + serviceEndpointInterfaceInfo + ")"); String className = getServiceEndpointInterfaceClassName(serviceEndpointInterfaceInfo.getServiceName()); try { LOG.debug("Check the class \"" + className + "\" is already loaded."); return loadClass(className); } catch (ClassNotFoundException e) { LOG.debug("Create new ServiceEndpointInterface class"); // load message type class ServiceParamInfo returnInfo = serviceEndpointInterfaceInfo.getReturnInfo(); if (returnInfo != null) { LOG.debug("Load return type (" + returnInfo.getType() + ")"); loadClass(returnInfo.getType()); } for (ServiceParamInfo info : serviceEndpointInterfaceInfo.getParamInfos()) { LOG.debug("Load param type (" + info.getType() + ")"); loadClass(info.getType()); } LOG.debug("Create a new class \"" + className + "\""); byte[] byteCode = createServiceEndpointInterfaceClass(serviceEndpointInterfaceInfo); return defineClass(className, byteCode, 0, byteCode.length); } } private static final String DESC_OF_XML_ACCESSOR_TYPE = // org.objectweb.asm.Type.getDescriptor(XmlAccessorType.class); "Ljavax/xml/bind/annotation/XmlAccessorType;"; private static final String DESC_OF_XML_ACCESS_TYPE = // org.objectweb.asm.Type.getDescriptor(XmlAccessType.class); "Ljavax/xml/bind/annotation/XmlAccessType;"; private static final String DESC_OF_WEB_SERVICE = // org.objectweb.asm.Type.getDescriptor(WebService.class); "Ljavax/jws/WebService;"; private static final String DESC_OF_SOAP_BINDING = // org.objectweb.asm.Type.getDescriptor(SOAPBinding.class); "Ljavax/jws/soap/SOAPBinding;"; private static final String DESC_OF_SOAP_BINDING_PARAMETER_STYLE = // org.objectweb.asm.Type.getDescriptor(SOAPBinding.ParameterStyle.class); "Ljavax/jws/soap/SOAPBinding$ParameterStyle;"; private static final String DESC_OF_WEB_METHOD = // org.objectweb.asm.Type.getDescriptor(WebMethod.class); "Ljavax/jws/WebMethod;"; private static final String DESC_OF_WEB_PARAM = // org.objectweb.asm.Type.getDescriptor(WebParam.class); "Ljavax/jws/WebParam;"; private static final String DESC_OF_WEB_PARAM_MODE = // org.objectweb.asm.Type.getDescriptor(WebParam.Mode.class); "Ljavax/jws/WebParam$Mode;"; private static final String DESC_OF_WEB_RESULT = // org.objectweb.asm.Type.getDescriptor(WebResult.class); "Ljavax/jws/WebResult;"; private static final org.objectweb.asm.Type TYPE_OF_HOLDER = org.objectweb.asm.Type.getType(Holder.class); private byte[] createRecordClass(final String className, final RecordType recordType) throws ClassNotFoundException { String asmClassName = className.replace('.', '/'); // ClassWriter classWriter = new // ClassWriter(0); ClassWriter classWriter = new ClassWriter(0); classWriter.visit(V1_5, // version ACC_PUBLIC, // access asmClassName, // name null, // signature "java/lang/Object", // superName null); // interfaces // Create Annotation AnnotationVisitor annotationVisitor = classWriter.visitAnnotation(DESC_OF_XML_ACCESSOR_TYPE, true); annotationVisitor.visitEnum("value", // name DESC_OF_XML_ACCESS_TYPE, // desc "FIELD"); // value annotationVisitor.visitEnd(); // Create Fields for (Entry<String, Type> entry : recordType.getFieldTypes().entrySet()) { String fieldName = entry.getKey(); Type fieldType = entry.getValue(); Class<?> fieldTypeClass = loadClass(fieldType); String desc = org.objectweb.asm.Type.getDescriptor(fieldTypeClass); classWriter.visitField(ACC_PUBLIC, // access fieldName, // name desc, // desc null, // signature null); // value } // Create Constructor MethodVisitor methodVisitor = classWriter.visitMethod(ACC_PUBLIC, // access "<init>", // name "()V", // desc null, // signature null); // exceptions methodVisitor.visitCode(); methodVisitor.visitVarInsn(ALOAD, 0); methodVisitor.visitMethodInsn(INVOKESPECIAL, // opcode "java/lang/Object", // owner "<init>", // name "()V"); // desc methodVisitor.visitInsn(RETURN); methodVisitor.visitMaxs(1, 1); methodVisitor.visitEnd(); // Class finalize classWriter.visitEnd(); // try // { // DataOutputStream dos = new DataOutputStream( // new FileOutputStream("EgovType" + // recordType.getId() + ".class")); // dos.write(classWriter.toByteArray()); // dos.close(); // } // catch (IOException e) // { // e.printStackTrace(); // } return classWriter.toByteArray(); } private byte[] createServiceEndpointInterfaceClass( final ServiceEndpointInterfaceInfo serviceEndpointInterfaceInfo) throws ClassNotFoundException { String serviceEndpointInterfaceClassName = getServiceEndpointInterfaceClassName( serviceEndpointInterfaceInfo.getServiceName()); String asmServiceEndpointInterfaceClassName = serviceEndpointInterfaceClassName.replace('.', '/'); // ClassWriter classWriter = new ClassWriter(false); ClassWriter classWriter = new ClassWriter(0); classWriter.visit(V1_5, // version ACC_PUBLIC | ACC_ABSTRACT | ACC_INTERFACE, // access asmServiceEndpointInterfaceClassName, // name null, // signature "java/lang/Object", // superName null); // interfaces // Create Annotation AnnotationVisitor annotationVisitor = classWriter.visitAnnotation(DESC_OF_WEB_SERVICE, true); annotationVisitor.visit("targetNamespace", serviceEndpointInterfaceInfo.getNamespace()); annotationVisitor.visitEnd(); annotationVisitor = classWriter.visitAnnotation(DESC_OF_SOAP_BINDING, true); annotationVisitor.visitEnum("parameterStyle", DESC_OF_SOAP_BINDING_PARAMETER_STYLE, "BARE"); // Create Method ServiceParamInfo returnInfo = serviceEndpointInterfaceInfo.getReturnInfo(); Collection<ServiceParamInfo> paramInfos = serviceEndpointInterfaceInfo.getParamInfos(); StringBuffer desc = new StringBuffer("("); StringBuffer signature = new StringBuffer("("); for (ServiceParamInfo info : paramInfos) { Class<?> paramClass = loadClass(info.getType()); org.objectweb.asm.Type paramType = org.objectweb.asm.Type.getType(paramClass); String paramSign = paramType.getDescriptor(); if (info.getMode().equals(OUT) || info.getMode().equals(INOUT)) { if (paramClass.isPrimitive()) { paramClass = wrapperClasses.get(paramClass); paramType = org.objectweb.asm.Type.getType(paramClass); paramSign = paramType.getDescriptor(); } paramClass = Holder.class; paramType = TYPE_OF_HOLDER; paramSign = "Ljavax/xml/ws/Holder<" + paramSign + ">;"; } desc.append(paramType.getDescriptor()); signature.append(paramSign); } desc.append(")"); signature.append(")"); org.objectweb.asm.Type returnType = (returnInfo == null ? returnType = org.objectweb.asm.Type.VOID_TYPE : org.objectweb.asm.Type.getType(loadClass(returnInfo.getType()))); desc.append(returnType.getDescriptor()); signature.append(returnType.getDescriptor()); MethodVisitor methodVisitor = classWriter.visitMethod(ACC_PUBLIC | ACC_ABSTRACT, // access serviceEndpointInterfaceInfo.getOperationName(), // name desc.toString(), // desc signature.toString(), // signature null); // exceptions // @WebMethod annotationVisitor = methodVisitor.visitAnnotation(DESC_OF_WEB_METHOD, true); annotationVisitor.visit("operationName", serviceEndpointInterfaceInfo.getOperationName()); annotationVisitor.visitEnd(); // @WebResult if (returnInfo != null) { annotationVisitor = methodVisitor.visitAnnotation(DESC_OF_WEB_RESULT, true); annotationVisitor.visit("name", returnInfo.getName()); // annotationVisitor.visit("partName", // returnInfo.getName()); annotationVisitor.visit("header", returnInfo.isHeader()); annotationVisitor.visit("targetNamespace", serviceEndpointInterfaceInfo.getNamespace()); annotationVisitor.visitEnd(); } // @WebParam int index = 0; for (ServiceParamInfo info : serviceEndpointInterfaceInfo.getParamInfos()) { annotationVisitor = methodVisitor.visitParameterAnnotation(index, DESC_OF_WEB_PARAM, true); annotationVisitor.visit("name", info.getName()); // annotationVisitor.visit("partName", // info.getName()); annotationVisitor.visitEnum("mode", DESC_OF_WEB_PARAM_MODE, info.getMode().toString()); annotationVisitor.visit("header", info.isHeader()); annotationVisitor.visit("targetNamespace", serviceEndpointInterfaceInfo.getNamespace()); annotationVisitor.visitEnd(); index++; } methodVisitor.visitEnd(); // Class finalize classWriter.visitEnd(); // try // { // DataOutputStream dos = new DataOutputStream( // new FileOutputStream("EgovType" + // serviceInfo.getServiceName() + ".class")); // dos.write(classWriter.toByteArray()); // dos.close(); // } // catch (IOException e) // { // e.printStackTrace(); // } return classWriter.toByteArray(); } private byte[] createServiceEndpointClass(final ServiceEndpointInfo serviceEndpointInfo) throws ClassNotFoundException { String serviceEndpointInterfaceClassName = getServiceEndpointInterfaceClassName( serviceEndpointInfo.getServiceName()); String serviceEndpointClassName = getServiceEndpointClassName(serviceEndpointInfo.getServiceName()); String asmServiceEndpointInterfaceClassName = serviceEndpointInterfaceClassName.replace('.', '/'); String asmServiceEndpointClassName = serviceEndpointClassName.replace('.', '/'); // ClassWriter classWriter = new ClassWriter(false); ClassWriter classWriter = new ClassWriter(0); classWriter.visit(V1_5, // version ACC_PUBLIC, // access asmServiceEndpointClassName, // name null, // signature "java/lang/Object", // superName new String[] { asmServiceEndpointInterfaceClassName }); // interfaces // Create Annotation AnnotationVisitor annotationVisitor = classWriter.visitAnnotation(DESC_OF_WEB_SERVICE, true); annotationVisitor.visit("endpointInterface", serviceEndpointInterfaceClassName); annotationVisitor.visit("targetNamespace", serviceEndpointInfo.getNamespace()); // annotationVisitor.visit("name", // serviceProviderInfo.getServiceName()); annotationVisitor.visit("serviceName", serviceEndpointInfo.getServiceName()); annotationVisitor.visit("portName", serviceEndpointInfo.getPortName()); annotationVisitor.visitEnd(); // Create Attribute FieldVisitor fieldVisitor = classWriter.visitField(ACC_PUBLIC, // access FIELD_NAME_OF_SERVICE_BRIDGE, // name DESC_OF_SERVICE_BRIDGE_CLASS, // desc null, // signature null); // value fieldVisitor.visitEnd(); // Create Constructor MethodVisitor methodVisitor = classWriter.visitMethod(ACC_PUBLIC, // access "<init>", // name "()V", // desc null, // signature null); // exceptions methodVisitor.visitCode(); methodVisitor.visitVarInsn(ALOAD, 0); methodVisitor.visitMethodInsn(INVOKESPECIAL, // opcode "java/lang/Object", // owner "<init>", // name "()V"); // desc methodVisitor.visitInsn(RETURN); methodVisitor.visitMaxs(1, 1); methodVisitor.visitEnd(); // Create Method ServiceParamInfo returnInfo = serviceEndpointInfo.getReturnInfo(); Collection<ServiceParamInfo> paramInfos = serviceEndpointInfo.getParamInfos(); StringBuffer desc = new StringBuffer("("); StringBuffer signature = new StringBuffer("("); for (ServiceParamInfo info : paramInfos) { Class<?> paramClass = loadClass(info.getType()); org.objectweb.asm.Type paramType = org.objectweb.asm.Type.getType(paramClass); String paramSign = paramType.getDescriptor(); if (info.getMode().equals(OUT) || info.getMode().equals(INOUT)) { if (paramClass.isPrimitive()) { paramClass = wrapperClasses.get(paramClass); paramType = org.objectweb.asm.Type.getType(paramClass); paramSign = paramType.getDescriptor(); } paramClass = Holder.class; paramType = TYPE_OF_HOLDER; paramSign = "Ljavax/xml/ws/Holder<" + paramSign + ">;"; } desc.append(paramType.getDescriptor()); signature.append(paramSign); } desc.append(")"); signature.append(")"); org.objectweb.asm.Type returnType = (returnInfo == null ? returnType = org.objectweb.asm.Type.VOID_TYPE : org.objectweb.asm.Type.getType(loadClass(returnInfo.getType()))); desc.append(returnType.getDescriptor()); signature.append(returnType.getDescriptor()); methodVisitor = classWriter.visitMethod(ACC_PUBLIC, // access serviceEndpointInfo.getOperationName(), // name desc.toString(), // desc signature.toString(), // signature null); // exceptions int mapPosition = paramInfos.size() + 1; methodVisitor.visitCode(); methodVisitor.visitTypeInsn(NEW, "java/util/HashMap"); methodVisitor.visitInsn(DUP); methodVisitor.visitMethodInsn(INVOKESPECIAL, // opcode "java/util/HashMap", // owner "<init>", // name "()V"); // desc methodVisitor.visitVarInsn(ASTORE, mapPosition); int i = 1; for (ServiceParamInfo info : paramInfos) { methodVisitor.visitVarInsn(ALOAD, mapPosition); methodVisitor.visitLdcInsn(info.getName()); methodVisitor.visitVarInsn(ALOAD, i); methodVisitor.visitMethodInsn(INVOKEINTERFACE, // opcode "java/util/Map", // owner "put", // name "(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;"); // desc methodVisitor.visitInsn(POP); i++; } methodVisitor.visitVarInsn(ALOAD, 0); methodVisitor.visitFieldInsn(GETFIELD, // opcode asmServiceEndpointClassName, // owner FIELD_NAME_OF_SERVICE_BRIDGE, // name DESC_OF_SERVICE_BRIDGE_CLASS); // desc methodVisitor.visitVarInsn(ALOAD, mapPosition); methodVisitor.visitMethodInsn(INVOKEINTERFACE, // opcode NAME_OF_SERVICE_BRIDGE_CLASS, // owner "doService", // name "(Ljava/util/Map;)Ljava/lang/Object;"); // desc if (returnInfo != null) { methodVisitor.visitTypeInsn(CHECKCAST, // opcode returnType.getInternalName()); // type methodVisitor.visitInsn(ARETURN); } else { methodVisitor.visitInsn(POP); methodVisitor.visitInsn(RETURN); } methodVisitor.visitMaxs(paramInfos.size(), paramInfos.size() + 2); methodVisitor.visitEnd(); // Class finalize classWriter.visitEnd(); // try // { // DataOutputStream dos = new DataOutputStream( // new FileOutputStream("EgovType" + // serviceProviderInfo.getServiceName() + // "Impl.class")); // dos.write(classWriter.toByteArray()); // dos.close(); // } // catch (IOException e) // { // e.printStackTrace(); // } return classWriter.toByteArray(); } private static String getRecordTypeClassName(final String recordTypeName) { return CLASS_NAME_PREFIX + recordTypeName; } private static String getServiceEndpointClassName(final String serviceName) { return CLASS_NAME_PREFIX + serviceName + SERVICE_ENDPOINT_CLASS_NAME_POSTFIX; } private static String getServiceEndpointInterfaceClassName(final String serviceName) { return CLASS_NAME_PREFIX + serviceName + SERVICE_ENDPOINT_INTERFACE_CLASS_NAME_POSTFIX; } }