scouter.jdbc.Generator Maven / Gradle / Ivy
/*
* Copyright 2015 the original author or authors.
* @https://github.com/scouter-project/scouter
*
* 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 scouter.jdbc;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
public class Generator {
public static void main(String[] args) throws Exception {
// String className = "ScouterConnection";
// String superName =null;
// String innerName = "java.sql.Connection";
// String className = "ScouterStatement";
// String superName = null;
// String innerName = "java.sql.Statement";
// String className = "ScouterPreparedStatement";
// String superName = "ScouterStatement";
// String innerName = "java.sql.PreparedStatement";
// String className = "ScouterCallableStatement";
// String superName = "ScouterPreparedStatement";
// String innerName = "java.sql.CallableStatement";
String className = "ScouterResultSet";
String superName = null;
String innerName = "java.sql.ResultSet";
Class target = Class.forName(innerName);
if (superName == null) {
System.out.println("public class " + className + " implements " + innerName + " {");
} else {
System.out.println("public class " + className + " extends " + superName + " implements " + innerName
+ " {");
}
System.out.println(" " + innerName + " inner;");
System.out.println(" public " + className + "(" + innerName + " inner){");
if (superName != null) {
System.out.println(" super(inner);");
}
System.out.println(" this.inner=inner;");
System.out.println(" }");
Method[] m = target.getMethods();
for (int i = 0; i < m.length; i++) {
boolean isVoid = m[i].getReturnType().toString().endsWith("void");
System.out.println(" " + getMethodString(m[i]) + "{");
System.out.println(" " + getInnerString(m[i]));
System.out.println(" }");
}
System.out.println("}");
}
public static String getMethodString(Method m) {
try {
StringBuilder sb = new StringBuilder();
int mod = m.getModifiers();
if (mod != 0) {
if (Modifier.isAbstract(mod)) {
mod ^= Modifier.ABSTRACT;
}
sb.append("final " + Modifier.toString(mod)).append(" ");
}
sb.append(getTypeName(m.getReturnType())).append(' ');
sb.append(m.getName()).append('(');
Class[] params = m.getParameterTypes(); // avoid clone
for (int j = 0; j < params.length; j++) {
sb.append(getTypeName(params[j]) + " a" + j);
if (j < (params.length - 1))
sb.append(',');
}
sb.append(')');
Class[] exceptions = m.getExceptionTypes(); // avoid clone
if (exceptions.length > 0) {
sb.append(" throws ");
for (int k = 0; k < exceptions.length; k++) {
sb.append(exceptions[k].getName());
if (k < (exceptions.length - 1))
sb.append(',');
}
}
return sb.toString();
} catch (Exception e) {
return "<" + e + ">";
}
}
public static String getInnerString(Method m) {
try {
boolean isVoid = getTypeName(m.getReturnType()).equals("void");
StringBuilder sb = new StringBuilder();
if (isVoid == false)
sb.append("return ");
sb.append("this.inner.");
sb.append(m.getName()).append('(');
Class[] params = m.getParameterTypes(); // avoid clone
for (int j = 0; j < params.length; j++) {
sb.append("a" + j);
if (j < (params.length - 1))
sb.append(", ");
}
sb.append(");");
return sb.toString();
} catch (Exception e) {
return "<" + e + ">";
}
}
static String getTypeName(Class type) {
if (type.isArray()) {
try {
Class cl = type;
int dimensions = 0;
while (cl.isArray()) {
dimensions++;
cl = cl.getComponentType();
}
StringBuffer sb = new StringBuffer();
sb.append(cl.getName());
for (int i = 0; i < dimensions; i++) {
sb.append("[]");
}
return sb.toString();
} catch (Throwable e) { /* FALLTHRU */
}
}
return type.getName();
}
}