模仿hibernate框架，詳解hibernate部分方法設計

王陸寬發布于2019-08-16 15:50 / 2834人閱讀

摘要：導讀源碼地址公司的持久層采用的框架，這也是很多公司使用的一種持久層框架。配置文件本項目以為開發環境和以搭建的，分為包和包。一種是，這種是手動提交事務。返回結果分為兩種，一種是以實體類直接返回，調用方法。

導讀

源碼地址

公司的持久層采用的hibernate框架，這也是很多公司使用的一種持久層框架。它將瞬時態的javabean對象轉化為持久態數據表的字段對象、或將持久態的字段對象轉化為瞬時態javabean對象。我比較喜歡看源碼，看別人的架構思想，因為，筆者想向架構師的方向進發?？戳藙e人的源碼，突然想模擬hibernate框架，自己寫個框架出來。這里去除了hibernate框架晦澀的地方，當做自己學習材料還是不錯的。里面涉及到反射、連接池等等。這個項目中，你可以知道數據庫連接池是怎么建的，又是怎么回收的。使用靜態代碼塊加載配置文件

以下詳細介紹我個人的項目，但肯定沒有人家源碼寫得好，這里僅作為學習使用。

如果不懂的，可以私信我。

配置文件

本項目以idea為開發環境和以maven搭建的，分為java包和test包。java包的配置文件放在resources下，代碼放在com.zby.simulationHibernate包下，如下是配置文件：

連接池

我們在使用hibernate時，一般會配置連接池，比如，初始化連接數是多少，最大連接數是多少？這個連接的是什么？我們在啟動項目時，hibernate根據初始的連接數，來創建多少個數據庫連接對象，也就是jdbc中的Connection對象。

為什么要有這個連接池？因為，每次開啟一個連接和關閉一個連接都是消耗資源的，我們開啟了這些連接對象之后，把它們放在一個容器中，我們何時需要何時從容器中取出來。當不需要的時候，再將踏進放回到容器中。因而，可以減少占用的資源。

如下，是初始化的連接對象：

package com.zby.simulationHibernate.util.factory;

import com.zby.simulationHibernate.util.exception.GenericException;
import org.apache.commons.lang3.StringUtils;

import java.io.IOException;
import java.io.InputStream;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;

/**
 * Created By zby on 21:23 2019/1/23
 * 數據庫的連接
 */
public class Connect {

    /**
     * 連接池的初始值
     */
    private static int initPoolSize = 20;

    /**
     * 創建property的配置文件
     */
    protected static Properties properties;

    /**
     * 連接池的最小值
     */
    protected static int minPoolSize;


    /**
     * 連接池的最大值
     */
    protected static int maxPoolSize;

    //【2】靜態代碼塊
    static {
        //加載配置文件
        properties = new Properties();
        InputStream is = ClassLoader.getSystemClassLoader().getResourceAsStream("db.properties");
        try {
            properties.load(is);
            minPoolSize = Integer.valueOf(properties.getProperty("jdbc.minConnPool"));
            if (minPoolSize <= initPoolSize)
                minPoolSize = initPoolSize;
            maxPoolSize = Integer.valueOf(properties.getProperty("jdbc.maxConnPool"));
            if (minPoolSize > maxPoolSize)
                throw new GenericException("連接池的最小連接數不能大于最大連接數");
        } catch (IOException e) {
            System.out.println("未找到配置文件");
            e.printStackTrace();
        }
    }

    /**
     * Created By zby on 16:50 2019/1/23
     * 獲取數據連接
     */
    protected java.sql.Connection createConnect() {
        String driverName = properties.getProperty("jdbc.driver");
        if (StringUtils.isEmpty(driverName)) {
            driverName = "com.mysql.jdbc.Driver";
        }
        String userName = properties.getProperty("jdbc.username");
        String password = properties.getProperty("jdbc.password");
        String dbUrl = properties.getProperty("jdbc.url");

        try {
            Class.forName(driverName);
            return DriverManager.getConnection(dbUrl, userName, password);
        } catch (ClassNotFoundException e) {
            System.out.println("找不到驅動類");
            e.printStackTrace();
        } catch (SQLException e) {
            System.out.println("加載異常");
            e.printStackTrace();
        }
        return null;
    }
}

創建Session會話

我們在使用hibernate時，不是直接使用連接對象，而是，以會話的方式創建一個連接。創建會話的方式有兩種。一種是openSession，這種是手動提交事務。getCurrentSession是自動提交事務。

如代碼所示：

package com.zby.simulationHibernate.util.factory;

import java.sql.Connection;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

/**
 * Created By zby on 15:43 2019/1/23
 */
public class SqlSessionFactory implements SessionFactory {

    /**
     * 連接池
     */
    private static List connections;

    /**
     * 連接對象
     *
     * @return
     */
    private static Connect connect = new Connect();

    protected static List getConnections() {
        return connections;
    }


    //靜態代碼塊，初始化常量池
    static {
        connections = new ArrayList<>();
        Connection connection;
        for (int i = 0; i < Connect.minPoolSize; i++) {
            connection = connect.createConnect();
            connections.add(connection);
        }
    }

    @Override
    public Session openSession() {
        return getSession(false);
    }

    @Override
    public Session getCurrentSession() {
        return getSession(true);
    }

    /**
     * 獲取session
     *
     * @param autoCommit 是否自動提交事務
     * @return
     */
    private Session getSession(boolean autoCommit) {
        //【1】判斷連接池有可用的連接對象
        boolean hasNoValidConn = hasValidConnction();
        //【2】沒有可用的連接池，使用最大的連接池
        if (!hasNoValidConn) {
            for (int i = 0; i < (Connect.maxPoolSize - Connect.minPoolSize); i++) {
                connections.add(connect.createConnect());
            }
        }
        //【3】有可用的連接
        for (Iterator iterator = connections.iterator(); iterator.hasNext(); ) {
            Connection connection = null;
            try {
                connection = (Connection) iterator.next();
                connection.setAutoCommit(autoCommit);
                Session session = new Session(connection);
                iterator.remove();
                return session;
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }

    /**
     * Created By zby on 21:50 2019/1/23
     * 當我們沒開啟一個連接，連接池的數目減少1，直到連接池的數量為0
     */
    private boolean hasValidConnction() {
        return null != connections && connections.size() != 0;
    }
}

數據查找

我們既然使用這個框架，必然要有數據查找的功能。返回結果分為兩種，一種是以實體類直接返回，調用addEntity方法。但是，多數情況下，是多張表聯合查詢的結果，這種情況下，直接以實體類接肯定不可以的。因而，我們需要自定義接收對象，并將查找結果進行過濾，再封裝成我們想要的對象。

第一種，以實體類返回

/**
 * Created By zby on 23:19 2019/1/23
 * 體檢反射的實體類
 */
public SqlQuery addEntity(Class persistenceClass) {
    this.persistenceClass = persistenceClass;
    return this;
}

第二種，過濾后返回數據

 /**
 * Created By zby on 19:18 2019/1/27
 * 創建類型
 */
public SqlQuery addScalar(String tuple, String alias) {
    if (CommonUtil.isNull(aliasMap)) {
        aliasMap = new HashMap<>();
    }
    for (Map.Entry entry : aliasMap.entrySet()) {
        String key = entry.getKey();
        if (key.equals(tuple))
            throw new GenericException("alias已經存在，即alias=" + key);
        String value = aliasMap.get(key);
        if (value.equals(alias) && key.equals(tuple))
            throw new GenericException("當前alias的type已經存在，alias=" + key + ",type=" + value);
    }
    aliasMap.put(tuple, alias);
    return this;
}

/**
 * Created By zby on 9:20 2019/1/28
 * 數據轉換問題
 */
public SqlQuery setTransformer(ResultTransformer transformer) {
    if (CommonUtil.isNull(aliasMap)) {
        throw new IllegalArgumentException("請添加轉換的屬性數量");
    }
    transformer.transformTuple(aliasMap);
    this.transformer = transformer;
    return this;
}

以集合的方式返回數據：

/**
 * Created By zby on 17:02 2019/1/29
 * 設置查找參數
 */
public SqlQuery setParamter(int start, Object param) {
    if (CommonUtil.isNull(columnParamer))
        columnParamer = new HashMap<>();
    columnParamer.put(start, param);
    return this;
}

/**
 * Created By zby on 16:41 2019/1/24
 * 查找值
 */
public List list() {
    PreparedStatement statement = null;
    ResultSet resultSet = null;
    try {
        statement = connection.prepareStatement(sql);
        if (CommonUtil.isNotNull(columnParamer)) {
            for (Map.Entry entry : columnParamer.entrySet()) {
                int key = entry.getKey();
                Object value = entry.getValue();
                statement.setObject(key + 1, value);
            }
        }
        resultSet = statement.executeQuery();
        PersistentObject persistentObject = new PersistentObject(persistenceClass, resultSet);
        if (CommonUtil.isNotNull(aliasMap))
            return persistentObject.getPersist(transformer);
        return persistentObject.getPersist();
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        SessionClose.closeConnStateResSet(connection, statement, resultSet);
    }
    return null;
}

返回唯一值

/**
 * Created By zby on 16:41 2019/1/24
 * 查找值
 */
public T uniqueResult() {
    List list = list();
    if (CommonUtil.isNull(list))
        return null;
    if (list.size() > 1)
        throw new GenericException("本來需要返回一個對象，卻返回 " + list.size() + "個對象");
    return list.get(0);
}

測試

 @Test
    public void testList() {
        Session session = new SqlSessionFactory().openSession();
        String sql = "SELECT " +
                " customer_name AS customerName, " +
                "  `name` AS projectName " +
                "FROM " +
                " project where id >= ? and id <= ?";
        SqlQuery query = session.createSqlQuery(sql);
        query.setParamter(0, 1);
        query.setParamter(1, 2);
        query.addScalar("customerName", StandardBasicTypes.STRING)
                .addScalar("projectName", StandardBasicTypes.STRING);
        query.setTransformer(Transforms.aliasToBean(ProjectData.class));
        List projects = query.list();
        for (ProjectData project : projects) {
            System.out.println(project.getCustomerName() + " " + project.getProjectName());
        }
    }

    @Ignore
    public void testListNoData() {
        Session session = new SqlSessionFactory().openSession();
        String sql = "SELECT " +
                " customer_name AS customerName, " +
                "  `name` AS projectName " +
                "FROM " +
                " project where id >= ? and id <= ?";
        SqlQuery query = session.createSqlQuery(sql).
                setParamter(0, 1).
                setParamter(1, 2).
                addEntity(Project.class);
        List projects = query.list();
        for (Project project : projects) {
            System.out.println(project.getCustomerName() + " " + project.getGuestCost());
        }
    }

保存數據

我們這里以 merge 方法來保存數據，因為這個方法非常的特殊，我們在這里做特殊說明。如果該瞬時態的對象有主鍵，而且，其在數據表中已經存在該主鍵的字段對象，我們此時就更新該數據表。如果數據表中沒有當前主鍵的字段對象，我們向數據庫中添加該對象的值。如果該瞬時態的對象沒有主鍵，我們直接在數據表中添加該對象。

如代碼所示：

 /**
 * Created By zby on 15:41 2019/1/29
 * 合并，首先判斷id是否存在，若id存在則更新，若id不存在，則保存數據
 */
public T merge(T t) {
    if (CommonUtil.isNull(t))
        throw new IllegalArgumentException("參數為空");
    Class clazz = (Class) t.getClass();
    Field[] fields = clazz.getDeclaredFields();
    boolean isContainsId = CommonUtil.isNotNull(PropertyUtil.containId(fields)) ? true : false;
    long id = PropertyUtil.getIdValue(fields, t, propertyAccessor);
    if (isContainsId) {
        return id > 0L ? update(t) : save(t);
    }
    return save(t);
}

  /**
 * Created By zby on 17:37 2019/1/29
 * 保存數據
 */
public T save(T t) {
    if (CommonUtil.isNull(t))
        throw new RuntimeException("不能保存空對象");
    PreparedStatement statement = null;
    ResultSet resultSet = null;
    StringBuilder columnJoint = new StringBuilder();
    StringBuilder columnValue = new StringBuilder();
    try {
        Field[] fields = t.getClass().getDeclaredFields();
        String sql = " insert into " + ClassUtil.getClassNameByGenericity(t) + "(";
        for (int i = 0; i < fields.length; i++) {
            String propertyName = fields[i].getName();
            Object propertyValue = propertyAccessor.getPropertyValue(t, propertyName);
            if (CommonUtil.isNotNull(propertyValue)) {
                String columnName = PropertyUtil.propertyNameTransformColumnName(propertyName, true);
                if (StandardBasicTypes.BOOLEAN.equalsIgnoreCase(fields[i].getGenericType().toString())) {
                    columnJoint.append("is_" + columnName + ",");
                    columnValue.append(propertyValue + ",");
                } else if (StandardBasicTypes.LONG.equalsIgnoreCase(fields[i].getGenericType().toString())
                        || StandardBasicTypes.FLOAT.equalsIgnoreCase(fields[i].getGenericType().toString())
                        || StandardBasicTypes.DOUBLE.equalsIgnoreCase(fields[i].getGenericType().toString())
                        || StandardBasicTypes.INTEGER.equalsIgnoreCase(fields[i].getGenericType().toString())) {
                    columnJoint.append(columnName + ",");
                    columnValue.append(propertyValue + ",");
                } else if (StandardBasicTypes.DATE.equalsIgnoreCase(fields[i].getGenericType().toString())) {
                    columnJoint.append(columnName + ",");
                    columnValue.append(""" + DateUtil.SIMPLE_DATE_FORMAT.format((Date) propertyValue) + "",");
                } else {
                    columnJoint.append(columnName + ",");
                    columnValue.append(""" + propertyValue + "",");
                }
            }
        }
        columnJoint = StringUtil.replace(columnJoint, ",");
        columnValue = StringUtil.replace(columnValue, ",");
        sql += columnJoint + ") VALUES(" + columnValue + ")";
        System.out.println(sql);
        statement = connection.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
        statement.executeUpdate();
        resultSet = statement.getGeneratedKeys();
        while (resultSet.next()) {
            return load((Class) t.getClass(), resultSet.getLong(1));
        }
        return t;
    } catch (SQLException e) {
        System.out.println("保存數據出錯,實體對象為=" + t);
        e.printStackTrace();
    } finally {
        SessionClose.closeConnStateResSet(connection, statement, resultSet);
    }
    return null;
}

測試代碼：

 @Test
public void testSave() {
    Session session = new SqlSessionFactory().getCurrentSession();
    Project project = new Project();
    project.setCustomerName("hhhh");
    project.setCreateDatetime(new Date());
    project.setDeleted(true);
    project = (Project) session.save(project);
    System.out.println(project.getId());
}

通過id加載對象

有時，我們只要根據當前對象的id，獲取當前對象的全部信息，因而，我們可以這樣寫：

 /**
 * Created By zby on 16:36 2019/1/29
 * 通過id獲取對象
 */
public T load(Class clazz, Long id) {
    if (CommonUtil.isNull(clazz))
        throw new IllegalArgumentException("參數為空");
    String className = ClassUtil.getClassNameByClass(clazz);
    String sql = " select * from " + className + " where id= ? ";
    SqlQuery query = createSqlQuery(sql)
            .setParamter(0, id)
            .addEntity(clazz);
    return (T) query.uniqueResult();
}

測試代碼：

@Test
public void testload() {
    Session session = new SqlSessionFactory().openSession();
    Project project = (Project) session.load(Project.class, 4L);
    System.out.println(project);
}

回收連接對象

當我們使用完該連接對象后，需要將對象放回到容器中，并不是直接調用connection.close()方法,而是調用這個方法：

 /**
 * Created By zby on 16:10 2019/3/17
 * 獲取容器的對象，如果是關閉session，則將連接對象放回到容器中
 * 如果是開啟session，則從容器中刪除該連接對象
 */
protected static List getConnections() {
    return connections;
}

 /**
 * Created By zby on 22:45 2019/1/23
 * 
 * 當關閉當前會話時，這并非真正的關閉會話
 * 只是將連接對象放回到連接池中
 */
public static void closeConn(Connection connection) {
    SqlSessionFactory.getConnections().add(connection);
}

總結

寫框架其實是不難的，難就難在如何設計框架?；蛘哒f，難就難在基礎不牢。如果基礎打不牢的話，很難往上攀升。

GPU云服務器云服務器 Hibernate對象詳解設計表 hibernate hibernate hibernate 查詢

文章版權歸作者所有，未經允許請勿轉載,若此文章存在違規行為，您可以聯系管理員刪除。

轉載請注明本文地址：http://m.specialneedsforspecialkids.com/yun/73729.html

Java 程序員必備的 15 個框架，前 3 個地位無可動搖！

摘要：官網源碼推薦從開始手寫一個框架更多請在技術棧微信公眾號后臺回復關鍵字。是一個開放源代碼的對象關系映射框架，它對進行了非常輕量級的對象封裝，它將與數據庫表建立映射關系，是一個全自動的框架。 Java 程序員方向太多，且不說移動開發、大數據、區塊鏈、人工智能這些，大部分 Java 程序員都是 Java Web/后端開發。那作為一名 Java Web 開發程序員必須需要熟悉哪些框架呢？今天...

galaxy_robot 2019-08-16 13:37 評論0 收藏0
納稅服務系統【總結】

摘要：要是使用到日歷的話，我們想到使用這個日歷類上面僅僅是我個人總結的要點，如果有錯誤的地方還請大家給我指正。納稅服務系統總結納稅服務系統是我第一個做得比較大的項目(不同于javaWeb小項目),該項目系統來源于傳智Java32期，十天的視頻課程(想要視頻的同學關注我的公眾號就可以直接獲取了) 我跟著練習一步一步完成需求，才發覺原來Java是這樣用來做網站的，Java有那么多的類庫，頁面...

ispring 2019-08-15 14:48 評論0 收藏0
Java 事務詳解

摘要：一個事務的執行不能被其他事務干擾。持久性也稱永久性，指一個事務一旦提交，它對數據庫中數據的改變就應該是永久性的。事務與相關聯，并通過調用實例化。強制此事務回滾。為此事務注冊用戶同步回調。檢查事務是否成功提交。一、事務 (1)事務（Transaction），一般是指要做的或所做的事情。在計算機術語中是指訪問并可能更新數據庫中各種數據項的一個程序執行單元(unit)。 (2)事務應該具有...

cyrils 2019-08-16 12:41 評論0 收藏0
Hibernate配置及自定義注冊映射文件

摘要：一配置屬性詳解可以在各式各樣不同環境下工作而設計的因此存在著大量的配置參數。以簡便操作，多數配置參數都有默認的配置值也是我們日常使用的必須品。 Hibernate （開放源代碼的對象關系映射框架） Hibernate是一個開放源代碼的對象關系映射框架，它對JDBC進行了非常輕量級的對象封裝，它將POJO與數據庫表建立映射關系，是一個全自動的orm框架，hibernat...

coordinate35 2019-08-15 13:54 評論0 收藏0
Java3y文章目錄導航

摘要：前言由于寫的文章已經是有點多了，為了自己和大家的檢索方便，于是我就做了這么一個博客導航。前言由于寫的文章已經是有點多了，為了自己和大家的檢索方便，于是我就做了這么一個博客導航。由于更新比較頻繁，因此隔一段時間才會更新目錄導航哦～想要獲取最新原創的技術文章歡迎關注我的公眾號：Java3y Java3y文章目錄導航 Java基礎泛型就這么簡單注解就這么簡單 Druid數據庫連接池...

KevinYan 2019-08-15 15:13 評論0 收藏0