实现xml方式 查询方式

Author: Aa\aAa

.gitignore

@@ -0,0 +1 @@
+target

MoodBatis.iml

@@ -0,0 +1,19 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module org.jetbrains.idea.maven.project.MavenProjectsManager.isMavenModule="true" type="JAVA_MODULE" version="4">
+  <component name="NewModuleRootManager" LANGUAGE_LEVEL="JDK_1_8">
+    <output url="file://$MODULE_DIR$/target/classes" />
+    <output-test url="file://$MODULE_DIR$/target/test-classes" />
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/src/main/java" isTestSource="false" />
+      <sourceFolder url="file://$MODULE_DIR$/src/main/resources" type="java-resource" />
+      <sourceFolder url="file://$MODULE_DIR$/src/test/java" isTestSource="true" />
+      <excludeFolder url="file://$MODULE_DIR$/target" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+    <orderEntry type="library" name="Maven: dom4j:dom4j:1.6.1" level="project" />
+    <orderEntry type="library" name="Maven: xml-apis:xml-apis:1.0.b2" level="project" />
+    <orderEntry type="library" name="Maven: mysql:mysql-connector-java:5.1.44" level="project" />
+    <orderEntry type="library" name="Maven: org.projectlombok:lombok:1.18.4" level="project" />
+  </component>
+</module>

pom.xml

@@ -0,0 +1,32 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <groupId>MoodBatis</groupId>
+    <artifactId>MoodBatis</artifactId>
+    <version>1.0-SNAPSHOT</version>
+
+    <dependencies>
+        <dependency>
+            <groupId>dom4j</groupId>
+            <artifactId>dom4j</artifactId>
+            <version>1.6.1</version>
+        </dependency>
+        <!-- MySQL -->
+        <dependency>
+            <groupId>mysql</groupId>
+            <artifactId>mysql-connector-java</artifactId>
+            <version>5.1.44</version>
+        </dependency>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <version>1.18.4</version>
+
+        </dependency>
+
+    </dependencies>
+
+</project>

src/main/java/com/mood/SqlSession/Excutor.java

@@ -0,0 +1,7 @@
+package com.mood.SqlSession;
+
+import com.mood.config.Function;
+
+public interface Excutor {
+    public <T> T query(Function statement, Object parameter);
+}

src/main/java/com/mood/SqlSession/MoodExcutor.java

@@ -0,0 +1,22 @@
+package com.mood.SqlSession;
+
+import com.mood.bean.User;
+import com.mood.config.Function;
+import com.mood.config.MapperBean;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+public class MoodExcutor implements Excutor {
+    private MoodSqlSessionFactory xmlConfiguration = new MoodSqlSessionFactory();
+
+    @Override
+    public <T> T query(Function sql, Object parameter) {
+        StatementHandler handler = new StatementHandler(xmlConfiguration);
+        return  handler.query(sql, parameter);
+    }
+
+
+}

src/main/java/com/mood/SqlSession/MoodMapperProxy.java

@@ -0,0 +1,37 @@
+package com.mood.SqlSession;
+
+import com.mood.config.Function;
+import com.mood.config.MapperBean;
+
+import java.lang.reflect.InvocationHandler;
+import java.lang.reflect.Method;
+import java.util.List;
+
+public class MoodMapperProxy implements InvocationHandler {
+    private MoodSession session;
+    private MoodSqlSessionFactory sessionFactory;
+
+    public MoodMapperProxy(MoodSqlSessionFactory sessionFactory, MoodSession session) {
+        this.session = session;
+        this.sessionFactory = sessionFactory;
+    }
+
+    @Override
+    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
+        MapperBean readMapper = sessionFactory.getXMLMapperBean("UserMapper.xml");
+        //是否是xml文件对应的接口
+        if(!method.getDeclaringClass().getName().equals(readMapper.getInterfaceName())){
+            return null;
+        }
+        List<Function> list = readMapper.getList();
+        if(list != null && list.size() > 0){
+            for (Function function : list) {
+                //id是否和接口方法名一样
+                if(method.getName().equals(function.getFuncName())){
+                    return session.selectOne(function, String.valueOf(args[0]));
+                }
+            }
+        }
+        return null;
+    }
+}

src/main/java/com/mood/SqlSession/MoodSession.java

@@ -0,0 +1,23 @@
+package com.mood.SqlSession;
+
+import com.mood.config.Function;
+import com.mood.config.MapperBean;
+
+import java.lang.reflect.Proxy;
+import java.util.HashMap;
+import java.util.Map;
+
+public class MoodSession {
+    private Excutor excutor = new MoodExcutor();
+
+    static private MoodSqlSessionFactory myConfiguration = new MoodSqlSessionFactory();
+
+
+    public <T> T selectOne(Function statement, Object parameter) {
+        return excutor.query(statement, parameter);
+    }
+
+    public <T> T getMapper(Class cla) {
+        return (T) Proxy.newProxyInstance(cla.getClassLoader(), new Class[]{cla}, new MoodMapperProxy(myConfiguration, this));
+    }
+}

src/main/java/com/mood/SqlSession/MoodSqlSessionFactory.java

@@ -0,0 +1,139 @@
+package com.mood.SqlSession;
+
+import com.mood.config.Function;
+import com.mood.config.MapperBean;
+import org.dom4j.Document;
+import org.dom4j.DocumentException;
+import org.dom4j.Element;
+import org.dom4j.io.SAXReader;
+
+import java.io.InputStream;
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+
+public class MoodSqlSessionFactory {
+
+    //加载
+    private static final ClassLoader classLoader = ClassLoader.getSystemClassLoader();
+
+    //初始化
+    public Connection bulider(String path) {
+        try {
+            InputStream stream = classLoader.getResourceAsStream(path);
+
+            SAXReader reader = new SAXReader();
+            Document document = reader.read(stream);
+            Element root = document.getRootElement();
+            return evalDataSource(root);
+        } catch (DocumentException e) {
+            throw new RuntimeException("error occured while evaling xml " + path);
+        }
+
+    }
+
+    private Connection evalDataSource(Element node) {
+        if (!"database".equals(node.getName())) {
+            throw new RuntimeException("root should be <database>");
+        }
+        String driverClassName = null;
+        String url = null;
+        String username = null;
+        String password = null;
+        //获取属性节点
+        for (Object item : node.elements("property")) {
+            Element i = (Element) item;
+            String value = getValue(i);
+            String name = i.attributeValue("name");
+            if (name == null || value == null) {
+                throw new RuntimeException("[database]: <property> should contain name and value");
+            }
+            //赋值
+            switch (name) {
+                case "url":
+                    url = value;
+                    break;
+                case "username":
+                    username = value;
+                    break;
+                case "password":
+                    password = value;
+                    break;
+                case "driverClassName":
+                    driverClassName = value;
+                    break;
+                default:
+                    throw new RuntimeException("[database]: <property> unknown name");
+            }
+        }
+
+        Connection connection = null;
+        try {
+            Class.forName(driverClassName);
+            //建立数据库链接
+            connection = DriverManager.getConnection(url, username, password);
+        } catch (ClassNotFoundException e) {
+            e.printStackTrace();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        return connection;
+    }
+
+    //获取property属性的值,如果有value值,则读取 没有设置value,则读取内容
+    private String getValue(Element node) {
+        return node.hasContent() ? node.getText() : node.attributeValue("value");
+    }
+
+    public MapperBean getXMLMapperBean(String path) {
+        MapperBean bean = new MapperBean();
+        InputStream resourceAsStream = classLoader.getResourceAsStream(path);
+        SAXReader saxReader = new SAXReader();
+        Document document = null;
+        try {
+            document = saxReader.read(resourceAsStream);
+            Element rootElement = document.getRootElement();
+            bean.setInterfaceName(rootElement.attributeValue("nameSpace").trim());
+            List<Function> list = new ArrayList<Function>(); //用来存储方法的List
+            for(Iterator rootIter = rootElement.elementIterator(); rootIter.hasNext();) {//遍历根节点下所有子节点
+                Function fun = new Function();    //用来存储一条方法的信息
+                Element e = (Element) rootIter.next();
+                String sqltype = e.getName().trim();
+                String funcName = e.attributeValue("id").trim();
+                String sql = e.getText().trim();
+                String resultType = e.attributeValue("resultType").trim();
+                fun.setSqltype(sqltype);
+                fun.setFuncName(funcName);
+                Object newInstance=null;
+                try {
+                    newInstance = Class.forName(resultType).newInstance();
+                } catch (InstantiationException e1) {
+                    e1.printStackTrace();
+                } catch (IllegalAccessException e1) {
+                    e1.printStackTrace();
+                } catch (ClassNotFoundException e1) {
+                    e1.printStackTrace();
+                }
+                fun.setResultType(newInstance);
+                fun.setSql(sql);
+                list.add(fun);
+            }
+            bean.setList(list);
+        } catch (DocumentException e) {
+            e.printStackTrace();
+        }
+
+
+        return bean;
+    }
+
+    public static void main(String[] args) {
+        MoodSqlSessionFactory myConfiguration = new MoodSqlSessionFactory();
+       // Connection con = myConfiguration.bulider("config.xml");
+        MapperBean xmlMapperBean = myConfiguration.getXMLMapperBean("UserMapper.xml");
+        System.out.println(xmlMapperBean);
+    }
+}

src/main/java/com/mood/SqlSession/ResultSetHandler.java

@@ -0,0 +1,62 @@
+package com.mood.SqlSession;
+
+import java.lang.reflect.Field;
+import java.lang.reflect.Method;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+public class ResultSetHandler {
+    //解析结果。生成实体
+    public <T> T handler(ResultSet resultSet, Object obj) {
+
+        //
+        Class clazz = obj.getClass();
+        //  Object obj = null;
+        try {
+             if (resultSet.next()){
+                int i = 0;
+                for (Field field: clazz.getDeclaredFields()){
+                    setValue(obj,field,resultSet,i);
+                }
+
+            }
+
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+
+        return (T)obj;
+
+
+    }
+
+    private void setValue(Object obj, Field field, ResultSet resultSet, int i) throws Exception {
+
+        Method m =  obj.getClass().getMethod("set"+upperCapital(field.getName()),field.getType());
+
+        m.invoke(obj,getResult(field,resultSet));
+
+    }
+
+    private Object getResult(Field field, ResultSet rs) throws SQLException {
+        //bean属性的名字必须要和数据库column的名字一样
+        Class<?> type = field.getType();
+        if("int".equals(type.getName()) ||Integer.class == type ){
+            return rs.getInt(field.getName());
+        }
+        if(String.class == type){
+            return rs.getString(field.getName());
+        }
+        return rs.getString(field.getName());
+
+
+
+    }
+
+    private String upperCapital(String name) {
+
+        char[] nameAry = name.toCharArray();
+        nameAry[0] -= 32;
+        return new String(nameAry);
+    }
+}

src/main/java/com/mood/SqlSession/StatementHandler.java

@@ -0,0 +1,43 @@
+package com.mood.SqlSession;
+
+import com.mood.config.Function;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.SQLException;
+
+public class StatementHandler {
+    private MoodSqlSessionFactory sqlSessionFactory;
+    private ResultSetHandler resultSetHandler = new ResultSetHandler();
+
+    public StatementHandler(MoodSqlSessionFactory sqlSessionFactory) {
+        this.sqlSessionFactory = sqlSessionFactory;
+    }
+
+    public <T> T query(Function function, Object parameter) {
+        Connection conn = this.getConnection();
+        try {
+            String format = String.format(function.getSql(),
+                    Integer.parseInt(String.valueOf(parameter)));
+            PreparedStatement p = conn.prepareStatement(
+                    format
+            );
+            p.setObject(1, parameter);
+            p.execute();
+            return resultSetHandler.handler(p.getResultSet(), function.getResultType());
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        return null;
+    }
+
+    private Connection getConnection() {
+        try {
+            Connection connection = sqlSessionFactory.bulider("config.xml");
+            return connection;
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return null;
+    }
+}