1. 程式人生 > >自定義一個註解來獲取資料庫的連結

自定義一個註解來獲取資料庫的連結

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

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface JdbcInfo {
String driverClass() default “com.mysql.jdbc.Driver”;
String url();
String username() default “root”;
String password() default “1012”;
}


import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.DriverManager;

public class JdbcUtils {
@JdbcInfo(url = “jdbc:mysql://localhost:3306/day16”)
public static Connection getConnection() throws Exception{
//獲取位元組碼檔案
Class clazz = JdbcUtils.class;
//獲取getConnection()
Method method = clazz.getMethod(“getConnection”, null);
if(method.isAnnotationPresent(JdbcInfo.class)){//判斷方法上是否有註解
//獲取註解
JdbcInfo jdbcInfo = method.getAnnotation(JdbcInfo.class);
//獲取四大引數
String driverClass = jdbcInfo.driverClass();
String url = jdbcInfo.url();
String username = jdbcInfo.username();
String password = jdbcInfo.password();
//註冊驅動
Class.forName(driverClass);
Connection connection = DriverManager.getConnection(url, username, password);
//獲取連線
return connection;
}
return null;
}
public static void main(String[] args) throws Exception {
System.out.println(getConnection());
}
}