深入JAVA注解之属性注解

项目目录结构

实体类：

package org.guangsoft.annotation.entity;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(value = {ElementType.FIELD, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface DBInfo {
    //属性名参考com.mchange.v2.c3p0.DriverManagerDataSource
    String driverClass() default "com.mysql.jdbc.Driver";
    String jdbcUrl() default "jdbc:mysql://192.168.50.30:3306/guanghe";
    String user() default "root";
    String password() default "root";
}

DAO层：

package org.guangsoft.annotation.dao;

import org.guangsoft.annotation.entity.DBInfo;

import com.mchange.v2.c3p0.ComboPooledDataSource;

public class CommDAO {

    @DBInfo
    private ComboPooledDataSource dataSource;
    
    public void setDataSource(ComboPooledDataSource dataSource) {
        this.dataSource = dataSource;
    }
    
    public ComboPooledDataSource getDataSource() {
        return dataSource;
    }
    
}

service层：

package org.guangsoft.annotation.service;

import java.sql.Connection;

import javax.sql.DataSource;

import org.guangsoft.annotation.dao.CommDAO;
import org.guangsoft.annotation.utils.JDBCUtil3;

public class CommService {
    
    public static void main(String args[]) throws Exception {
        CommDAO commDAO = JDBCUtil3.createCommDAO();
        DataSource dataSource = commDAO.getDataSource();
        Connection connection = dataSource.getConnection();
        System.out.println(connection);
    }
    
}

util类：

package org.guangsoft.annotation.utils;

import java.beans.PropertyDescriptor;
import java.lang.reflect.Field;
import java.lang.reflect.Method;

import javax.sql.DataSource;

import org.guangsoft.annotation.dao.CommDAO;
import org.guangsoft.annotation.entity.DBInfo;

public class JDBCUtil3 {
    
    //将注解注入到数据源类
    private static DataSource getDataSourceByDBInfo (DBInfo dbInfo, DataSource dataSource) {
        Method[] methods = DBInfo.class.getMethods();
        for(Method method : methods) {
            String name = method.getName();
            try {                
                //注解类没有属性，反射注解类的方法名，内省出数据源类设置参数的方法, 找不到会抛异常，进入下次循环
                PropertyDescriptor propertyDescriptor = new PropertyDescriptor(name, dataSource.getClass());
                //注解类的方法能得到实际注解的值
                Object value = method.invoke(dbInfo, null);
                //用数据源的方法将注解类的值注入
                propertyDescriptor.getWriteMethod().invoke(dataSource, value);
            } catch(Exception e) {                
                continue;
            }
        }
        return dataSource;
    }
    
    //工厂模式下的创建DAO
    public static CommDAO createCommDAO() {
        CommDAO commDAO = new CommDAO();
        try {
            Field[] fields = commDAO.getClass().getDeclaredFields();
            if(fields != null) {
                for(Field field : fields) {
                    field.setAccessible(true);
                    DBInfo dbInfo = field.getAnnotation(DBInfo.class);
                    if(dbInfo != null) {
                        //获取dao中dataSource的实体类ComboPooledDataSource
                        DataSource dataSource = (DataSource) field.getType().newInstance(); 
                        dataSource = getDataSourceByDBInfo(dbInfo, dataSource);
                        field.set(commDAO, dataSource);
                    } 
                }
            }
        }catch(Exception e) {
            e.printStackTrace();
        }
        return commDAO;
    }
    
}

结果（成功）：