最近看一些ORM框架的源码,自己记录一些笔记.类似于JPA,不用手写Sql,那个实体类要要@Table和@Column注解.(基于JDK1.8).
首先自己定义一个@Column注解.
@Target(ElementType.FIELD) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface Column { String name(); String type() default "string"; int length() default 20; }
实体类(使用了Lombok的依赖,不用在添加Set()和Get()方法,直接在类上面使用注解就可以的)
@Table(name="student") @Getter @Setter public class Student { @Id(name="sid") private int sid; @Column(name="sname") private String sname; @Column(name="sex") private String sex; @Column(name="age") private int age; @Override public String toString() { return "Student{" + "sid=" + sid + ", sname='" + sname + '\'' + ", sex='" + sex + '\'' + ", age=" + age + '}'; } public Student(int sid, String sname, String sex, int age){ this.sid=sid; this.sname=sname; this.sex=sex; this.age=age; } public Student(){} }
接下来就是如何如何获取@Column注解的所有属性值,举个例子就是获取@Column(name="sname")中的sname.
实现的源码如下:
分析一下:首先我们要注意一点的就是我们注解的属性值是私有的实例变量,首先要设置私有变量的属性值是可以访问的,这里就要使用按个setAccessible(true),默认是false.保证注解属性值可以被访问到.
思路分析:
1:由相应类的Class实例对象获取所有声明的变量(包括private).同时设置这些变量值的可访问性.
2:获取相应的变量上面指定类型的注解(这里是@Column注解),同时使用一个String[]数组来保存的.
import com.lx.orm.annotation.Column; import com.lx.orm.entity.Student; import java.lang.reflect.Field; /** * title: com.lx.orm.util * @author: * date: * description:根据注解获取类的指定注解的属性值 */ public class ColumnUtil { public static void main(String[] args) { String[] str=getColumnValue(Student.class); for (int i = 1; i <str.length ; i++) { System.out.println(str[i]); } } public static String[] getColumnValue(Class classz) { Field[] fields=classz.getDeclaredFields(); Field field; String[] value=new String[fields.length]; for (int i = 0; i <fields.length ; i++) { fields[i].setAccessible(true); } for(int i = 1;i<fields.length ; i++){ try { field=classz.getDeclaredField(fields[i].getName()); Column column=field.getAnnotation(Column.class); //获取指定类型注解 if(column!=null){ value[i]=column.name(); } } catch (NoSuchFieldException e) { e.printStackTrace(); } } return value; } }
运行结果如下:
正是自己想要的那三个基于@Column注解的属性值.