Java注解（三）

上一篇了解了自定义注解的使用，不过里面的例子没有多大使用价值，这一回来个有用点的Demo。

目标：将实体bean保存到数据库

先来定义一个实体注解

import java.lang.annotation.ElementType;

import java.lang.annotation.Retention;

import java.lang.annotation.RetentionPolicy;

import java.lang.annotation.Target;



@Target(ElementType.TYPE)  

@Retention(RetentionPolicy.RUNTIME)

public @interface Entity {

    String getTableName();

}

这个注解可用在类上，它有一个变量参数getTableName。

其实意义很明显，就是一个实体类对应一张数据库的表，通过Entity注解将类和数据库表名关联起来

 

那么，通过什么将类的参数和数据库表中的列关联起来呢？再来定义一个注解

import java.lang.annotation.ElementType;

import java.lang.annotation.Retention;

import java.lang.annotation.RetentionPolicy;

import java.lang.annotation.Target;



@Target(ElementType.FIELD) 

@Retention(RetentionPolicy.RUNTIME)

public @interface Column {

    String getName();

}

有了Column注解，类里面的属性就和表中的列关联起来了。

 

下面来看看POJO中怎么用这两个注解：

@Entity(getTableName = "user")

public class User {

    @Column(getName = "user_id")

    private String id;

    @Column(getName = "user_name")

    private String name;

    @Column(getName = "user_age")

    private int age;



    public String getId() {

        return id;

    }



    public void setId(String id) {

        this.id = id;

    }



    public String getName() {

        return name;

    }



    public void setName(String name) {

        this.name = name;

    }



    public int getAge() {

        return age;

    }



    public void setAge(int age) {

        this.age = age;

    }

}

通过Entity和Column注解，就将一个实体bean和一张数据库表连接起来了。很多ORM映射就是采取这种方式实现的。

 

最后，来感受一下注解给我们带来的便利，来个方法见证下~~

import java.lang.reflect.Field;

import java.sql.Connection;

import java.sql.PreparedStatement;



public class Session {

    public static void main(String[] args) {

        Session session = new Session();

        System.out.println(session.getInsertSql(new User()));

    }



    //在实际项目中，你可以save(obj)方法来保存一个bean

    public void save(Object obj) {

        // get a connection

        //PreparedStatement pstmt = getStatement(con, obj);

        //pstmt.execute();

    }



    //得到PreparedStatement

    public PreparedStatement getStatement(Connection con, Object obj)

            throws Exception {

        PreparedStatement pstmt = con.prepareStatement(getInsertSql(obj));

        Class<?> c = obj.getClass();

        Field[] fs = c.getDeclaredFields();

        for (int i = 0; i < fs.length; i++) {

            fs[i].setAccessible(true);

            pstmt.setObject(i + 1, fs[i].get(obj));

        }

        return pstmt;

    }



    //insert into tableName(ziduan1,ziduan2...) values(?,?...)

    public String getInsertSql(Object obj) {

        StringBuilder s = new StringBuilder();

        s.append("insert into ");



        Class<?> c = obj.getClass();

        String tableName = c.getSimpleName();//类名，不包含包名 User

        Entity entity = (Entity) c.getAnnotation(Entity.class);

        if (entity != null) {

            tableName = entity.getTableName();

        }

        s.append(tableName).append("(");

        Field[] fs = c.getDeclaredFields();



        for (int i = 0; i < fs.length; i++) {

            String fieldName = fs[i].getName();

            Column column = fs[i].getAnnotation(Column.class);

            if (column != null) {

                fieldName = column.getName();

            }

            s = i == 0 ? s.append(fieldName) : s.append(",").append(fieldName);

        }

        s.append(") values").append(getString(fs.length));

        return s.toString();



    }



    //得到(?,?,?,...?,?)

    private String getString(int length) {

        StringBuilder s = new StringBuilder();

        s.append("(");

        for (int i = 0; i < length; i++) {

            s = i == 0 ? s.append("?") : s.append(",?");

        }

        s.append(")");

        return s.toString();

    }

}

 

 OK，That's all！