去年上课老师要求写一个java项目作为作业 今天闲来无事收集起来 保留一下以前写的代码而已
首先直接上代码吧
public class generateSQL {
public static void main(String[] args) {
Teacher teacher = new Teacher();
generateSQL generateSQL= new generateSQL();
teacher.setId("1");
teacher.setSalary("111");
teacher.setProfessional("111");
teacher.setBirth("111");
teacher.setSchool("111");
teacher.setName("111");
String str1 = generateSQL.selectSQL(teacher.getClass());
System.out.println("输出的SELECT的sql:" + str1);
String str2 = generateSQL.deleteSQL(teacher.getClass(), teacher);
System.out.println("输出的DELETE的sq2:" + str2);
String str3 = generateSQL.insertSQL(teacher.getClass(), teacher);
System.out.println("输出的INSERT的sq3:" + str3);
String str4 = generateSQL.updateSQL(teacher.getClass(), teacher);
System.out.println("输出的UPDATE的sq4:" + str4);
}
//生成select的sql语句,
public String selectSQL(Class<?> clazz) {
StringBuilder str = new StringBuilder();
// //获取一个Table的名字
// boolean exists = clazz.isAnnotationPresent(Entity.class);
// if (!exists){
// return null;
// }
// Entity entity = clazz.getAnnotation(Entity.class);
String tableName = getTableName(clazz);
str.append("select * from ").append(tableName);
String resultStr = str.toString();
return resultStr;
}
//生成delete的sql语句
public String deleteSQL(Class<?> clazz, Object object) {
StringBuilder str = new StringBuilder();
String tableName = getTableName(clazz);
str.append("delete from ").append(tableName).append(" where ");
Field[] fields = clazz.getDeclaredFields();
String pk = "";
for (int i = 0; i < fields.length; i++) {
Column column = fields[i].getAnnotation(Column.class);
if (column.isPK()) {
pk = column.label();
break;
}
}
try {
str.append(pk).append(" = ").append(getValue(pk, object));
} catch (Exception e) {
e.printStackTrace();
}
String resultStr = str.toString();
return resultStr;
}
//生成insert的sql语句
public String insertSQL(Class<?> clazz, Object object) {
StringBuilder str = new StringBuilder();
String tableName = getTableName(clazz);
str.append("insert into ").append(tableName).append(" (");
Field[] fields = clazz.getDeclaredFields();
for (int i = 0; i < fields.length - 1; i++) {
Column column = fields[i].getAnnotation(Column.class);
str.append(column.label()).append(" , ");
}
Column columns = fields[fields.length - 1].getAnnotation(Column.class);
str.append(columns.label()).append(") values(");
try {
for (int i = 0; i < fields.length - 1; i++) {
Column column = fields[i].getAnnotation(Column.class);
str.append(getValue(column.label(), object)).append(" , ");
}
str.append(getValue(columns.label(),object)).append(")");
} catch (Exception e) {
e.printStackTrace();
}
String resultStr = str.toString();
return resultStr;
}
//生成update的sql语句
public String updateSQL(Class<?> clazz, Object object) {
StringBuilder str = new StringBuilder();
String tableName = getTableName(clazz);
str.append("update ").append(tableName).append(" set ");
Field[] fields = clazz.getDeclaredFields();
String pk = "";
try {
for (int i = 0; i < fields.length - 1; i++) {
Column column = fields[i].getAnnotation(Column.class);
if (column.isPK()) {
pk = column.label();
} else {
str.append(column.label()).append(" = ").append(getValue(column.label(), object)).append(",");
}
}
Column columns = fields[fields.length - 1].getAnnotation(Column.class);
str.append(columns.label()).append(" = ").append(getValue(columns.label(), object)).append(" ");
str.append(" where ").append(pk).append(" = ").append(getValue(pk, object));
} catch (Exception e) {
e.printStackTrace();
}
String resultStr = str.toString();
return resultStr;
}
//获取表的名称
private String getTableName(Class<?> clazz) {
//判断是否为Table注释类型是方法返回true,否则返回false
if (clazz.isAnnotationPresent(Entity.class)) {
//获取注解信息
Entity table = clazz.getAnnotation(Entity.class);
if (!"".equals(table.tableName())) {
return table.tableName();
}
}
return null;
}
//通过对象的值,来获取实际的值
private <T> String getValue(String columnName, Object object) throws Exception {
//获取相应字段的getXXX方法
String getMethod = columnName.substring(0, 1).toUpperCase() + columnName.substring(1);
Method method = object.getClass().getMethod("get" + getMethod);
return method.invoke(object).toString();
}
}
Teache类:
entity注解:
Column注解:
输出结果的图片