自定义mybatis(注解和XML)
包结构
配置文件
<?xml version="1.0" encoding="UTF-8" ?>
<!--mybaits的主配置文件-->
<configuration>
<!--配置环境-->
<environments default="mysql">
<!--配置mysql的环境-->
<environment id="mysql">
<!--配置事务的类型-->
<transactionManager type="JDBC"></transactionManager>
<!--配置数据源(连接池)-->
<dataSource type="POOLED">
<!--配置连接数据库的4个基本信息-->
<property name="driver" value="com.mysql.jdbc.Driver"/>
<property name="url" value="jdbc:mysql://localhost:3306/eesy"/>
<property name="username" value="root"/>
<property name="password" value="root"/>
</dataSource>
</environment>
</environments>
<!--指定映射配置文件的位置,映射配置文件指的是每个独立的配置文件-->
<mappers>
<!--<mapper resource="com/hu/dao/UserDao.xml"/>-->
<mapper class="com.hu.dao.UserDao"/>
</mappers>
</configuration>
<?xml version="1.0" encoding="UTF-8" ?>
<mapper namespace="com.hu.dao.UserDao">
<!--配置查询所有-->
<select id="findAll" resultType="com.hu.domain.User">
select * from user;
</select>
</mapper>
utils包
package com.hu.utils;
import com.hu.mybatis.cfg.Configuration;
import java.sql.Connection;
import java.sql.DriverManager;
public class DataSourceUtil {
public static Connection getConnection(Configuration cfg){
try {
Class.forName(cfg.getDriver());
return DriverManager.getConnection(cfg.getUrl(), cfg.getUsername(), cfg.getPassword());
}catch(Exception e){
throw new RuntimeException(e);
}
}
}
package com.hu.utils;
import com.hu.mybatis.cfg.Mapper;
import java.beans.PropertyDescriptor;
import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.util.ArrayList;
import java.util.List;
public class Executor {
public <E> List<E> selectList(Mapper mapper, Connection conn) {
PreparedStatement pstm = null;
ResultSet rs = null;
try {
String queryString = mapper.getQueryString();
String resultType = mapper.getResultType();
Class domainClass = Class.forName(resultType);
pstm = conn.prepareStatement(queryString);
rs = pstm.executeQuery();
List<E> list = new ArrayList<E>();
while(rs.next()) {
E obj = (E)domainClass.newInstance();
ResultSetMetaData rsmd = rs.getMetaData();
int columnCount = rsmd.getColumnCount();
for (int i = 1; i <= columnCount; i++) {
String columnName = rsmd.getColumnName(i);
Object columnValue = rs.getObject(columnName);
PropertyDescriptor pd = new PropertyDescriptor(columnName,domainClass);
Method writeMethod = pd.getWriteMethod();
writeMethod.invoke(obj,columnValue);
}
list.add(obj);
}
return list;
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
release(pstm,rs);
}
}
private void release(PreparedStatement pstm,ResultSet rs){
if(rs != null){
try {
rs.close();
}catch(Exception e){
e.printStackTrace();
}
}
if(pstm != null){
try {
pstm.close();
}catch(Exception e){
e.printStackTrace();
}
}
}
}
package com.hu.utils;
import com.hu.mybatis.annotation.Select;
import com.hu.mybatis.cfg.Configuration;
import com.hu.mybatis.cfg.Mapper;
import com.hu.mybatis.io.Resources;
import org.dom4j.Attribute;
import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class XMLConfigBuilder {
public static Configuration loadConfiguration(InputStream config){
try{
Configuration cfg = new Configuration();
SAXReader reader = new SAXReader();
Document document = reader.read(config);
Element root = document.getRootElement();
List<Element> propertyElements = root.selectNodes("//property");
for(Element propertyElement : propertyElements){
String name = propertyElement.attributeValue("name");
if("driver".equals(name)){
String driver = propertyElement.attributeValue("value");
cfg.setDriver(driver);
}
if("url".equals(name)){
String url = propertyElement.attributeValue("value");
cfg.setUrl(url);
}
if("username".equals(name)){
String username = propertyElement.attributeValue("value");
cfg.setUsername(username);
}
if("password".equals(name)){
String password = propertyElement.attributeValue("value");
cfg.setPassword(password);
}
}
List<Element> mapperElements = root.selectNodes("//mappers/mapper");
for(Element mapperElement : mapperElements){
Attribute attribute = mapperElement.attribute("resource");
if(attribute != null){
System.out.println("使用的是XML");
String mapperPath = attribute.getValue();
Map<String, Mapper> mappers = loadMapperConfiguration(mapperPath);
cfg.setMappers(mappers);
}else{
System.out.println("使用的是注解");
String daoClassPath = mapperElement.attributeValue("class");
Map<String,Mapper> mappers = loadMapperAnnotation(daoClassPath);
cfg.setMappers(mappers);
}
}
return cfg;
}catch(Exception e){
throw new RuntimeException(e);
}finally{
try {
config.close();
}catch(Exception e){
e.printStackTrace();
}
}
}
private static Map<String,Mapper> loadMapperConfiguration(String mapperPath)throws IOException {
InputStream in = null;
try{
Map<String,Mapper> mappers = new HashMap<String,Mapper>();
in = Resources.getResourceAsStream(mapperPath);
SAXReader reader = new SAXReader();
Document document = reader.read(in);
Element root = document.getRootElement();
String namespace = root.attributeValue("namespace");
List<Element> selectElements = root.selectNodes("//select");
for(Element selectElement : selectElements){
String id = selectElement.attributeValue("id");
String resultType = selectElement.attributeValue("resultType");
String queryString = selectElement.getText();
String key = namespace+"."+id;
Mapper mapper = new Mapper();
mapper.setQueryString(queryString);
mapper.setResultType(resultType);
mappers.put(key,mapper);
}
return mappers;
}catch(Exception e){
throw new RuntimeException(e);
}finally{
in.close();
}
}
private static Map<String,Mapper> loadMapperAnnotation(String daoClassPath)throws Exception{
Map<String,Mapper> mappers = new HashMap<String, Mapper>();
Class daoClass = Class.forName(daoClassPath);
Method[] methods = daoClass.getMethods();
for(Method method : methods){
boolean isAnnotated = method.isAnnotationPresent(Select.class);
if(isAnnotated){
Mapper mapper = new Mapper();
Select selectAnno = method.getAnnotation(Select.class);
String queryString = selectAnno.value();
mapper.setQueryString(queryString);
Type type = method.getGenericReturnType();
if(type instanceof ParameterizedType){
ParameterizedType ptype = (ParameterizedType)type;
Type[] types = ptype.getActualTypeArguments();
Class domainClass = (Class)types[0];
String resultType = domainClass.getName();
mapper.setResultType(resultType);
}
String methodName = method.getName();
String className = method.getDeclaringClass().getName();
String key = className+"."+methodName;
mappers.put(key,mapper);
}
}
return mappers;
}
}
dao包
package com.hu.dao;
import com.hu.domain.User;
import com.hu.mybatis.annotation.Select;
import java.util.List;
public interface UserDao {
@Select("select * from user")
List<User> findAll();
}
domain包
package com.hu.domain;
import java.io.Serializable;
import java.util.Date;
public class User implements Serializable {
private int id;
private String username;
private Date birthday;
private String sex;
private String address;
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getUsername() {
return username;
}
public void setUsername(String username) {
this.username = username;
}
public Date getBirthday() {
return birthday;
}
public void setBirthday(Date birthday) {
this.birthday = birthday;
}
public String getSex() {
return sex;
}
public void setSex(String sex) {
this.sex = sex;
}
public String getAddress() {
return address;
}
public void setAddress(String address) {
this.address = address;
}
@Override
public String toString() {
return "User{" +
"id=" + id +
", username='" + username + '\'' +
", birthday=" + birthday +
", sex='" + sex + '\'' +
", address='" + address + '\'' +
'}';
}
}
mybatis.annotation包
package com.hu.mybatis.annotation;
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 Select {
String value();
}
mybatis.cfg包
package com.hu.mybatis.cfg;
import java.util.HashMap;
import java.util.Map;
public class Configuration {
private String driver;
private String url;
private String username;
private String password;
private Map<String,Mapper> mappers;
public Map<String, Mapper> getMappers() {
return mappers;
}
public void setMappers(Map<String, Mapper> mappers1) {
this.mappers=new HashMap<String, Mapper>();
this.mappers.putAll(mappers1);
}
public String getDriver() {
return driver;
}
public void setDriver(String driver) {
this.driver = driver;
}
public String getUrl() {
return url;
}
public void setUrl(String url) {
this.url = url;
}
public String getUsername() {
return username;
}
public void setUsername(String username) {
this.username = username;
}
public String getPassword() {
return password;
}
public void setPassword(String password) {
this.password = password;
}
}
package com.hu.mybatis.cfg;
public class Mapper {
private String queryString;
private String resultType;
public String getQueryString() {
return queryString;
}
public void setQueryString(String queryString) {
this.queryString = queryString;
}
public String getResultType() {
return resultType;
}
public void setResultType(String resultType) {
this.resultType = resultType;
}
}
mybatis.io包
package com.hu.mybatis.io;
import java.io.InputStream;
public class Resources {
public static InputStream getResourceAsStream(String filePath){
return Resources.class.getClassLoader().getResourceAsStream(filePath);
}
}
sqlsession包
package com.hu.mybatis.sqlsession;
public interface SqlSession {
<T> T getMapper(Class<T> daoInterfaceClass);
void close();
}
package com.hu.mybatis.sqlsession;
public interface SqlSessionFactory {
SqlSession openSession();
}
package com.hu.mybatis.sqlsession;
import com.hu.mybatis.cfg.Configuration;
import com.hu.mybatis.sqlsession.defaults.DefaultSqlSessionFactory;
import com.hu.utils.XMLConfigBuilder;
import java.io.InputStream;
public class SqlSessionFactoryBuilder {
public SqlSessionFactory build(InputStream config){
Configuration cfg= XMLConfigBuilder.loadConfiguration(config);
return new DefaultSqlSessionFactory(cfg);
}
}
sqlsession.proxy包
package com.hu.mybatis.sqlsession.proxy;
import com.hu.mybatis.cfg.Mapper;
import com.hu.utils.Executor;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.sql.Connection;
import java.util.Map;
public class MapperProxy implements InvocationHandler {
private Map<String, Mapper> mappers;
private Connection coon;
public MapperProxy(Map<String, Mapper> mappers,Connection coon){
this.mappers=mappers;
this.coon=coon;
}
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
String methodName = method.getName();
String className=method.getDeclaringClass().getName();
String key= className+"."+methodName;
Mapper mapper = mappers.get(key);
if(mapper==null){
throw new IllegalArgumentException("传入的参数有误");
}
return new Executor().selectList(mapper,coon);
}
}
sqlsession.defaults包
package com.hu.mybatis.sqlsession.defaults;
import com.hu.mybatis.cfg.Configuration;
import com.hu.mybatis.sqlsession.SqlSession;
import com.hu.mybatis.sqlsession.proxy.MapperProxy;
import com.hu.utils.DataSourceUtil;
import java.lang.reflect.Proxy;
import java.sql.Connection;
import java.sql.SQLException;
public class DefaultSqlSession implements SqlSession {
private Configuration cfg;
private Connection connection;
public DefaultSqlSession(Configuration cfg){
this.cfg=cfg;
connection= DataSourceUtil.getConnection(cfg);
}
public <T> T getMapper(Class<T> daoInterfaceClass) {
return (T) Proxy.newProxyInstance(daoInterfaceClass.getClassLoader(), new Class[]{daoInterfaceClass}, new MapperProxy(cfg.getMappers(), connection));
}
public void close() {
if(connection!=null){
try {
connection.close();
} catch (SQLException e) {
e.printStackTrace();
}
}
}
}
package com.hu.mybatis.sqlsession.defaults;
import com.hu.mybatis.cfg.Configuration;
import com.hu.mybatis.sqlsession.SqlSession;
import com.hu.mybatis.sqlsession.SqlSessionFactory;
public class DefaultSqlSessionFactory implements SqlSessionFactory {
Configuration cfg;
public DefaultSqlSessionFactory(Configuration cfg){
this.cfg=cfg;
}
public SqlSession openSession() {
return new DefaultSqlSession(cfg);
}
}
test
package com.hu.test;
import com.hu.dao.UserDao;
import com.hu.domain.User;
import com.hu.mybatis.io.Resources;
import com.hu.mybatis.sqlsession.SqlSession;
import com.hu.mybatis.sqlsession.SqlSessionFactory;
import com.hu.mybatis.sqlsession.SqlSessionFactoryBuilder;
import java.io.InputStream;
import java.util.List;
public class MybatisTest {
public static void main(String[] args)throws Exception {
InputStream in = Resources.getResourceAsStream("SqlMapConfig.xml");
SqlSessionFactoryBuilder builder=new SqlSessionFactoryBuilder();
SqlSessionFactory factory=builder.build(in);
SqlSession session = factory.openSession();
UserDao userDao = session.getMapper(UserDao.class);
List<User> users = userDao .findAll();
for (User user : users) {
System.out.println(user);
}
session.close();
in.close();
}
}