增加部署武威市民勤系统

wuzeyu

2024-12-27 f5f8138f4f50eedd3598a9073956f0afa56e7139

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package com.dy.common.multiDataSource;
 
import org.apache.ibatis.transaction.Transaction;
import org.springframework.jdbc.CannotGetJdbcConnectionException;
import org.springframework.jdbc.datasource.DataSourceUtils;
 
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
 
/**
 * 多数据源的事务类，事务对象由事务工厂创建，完成数据库事务操作。
 * 事务对象包裹着：
 *   数据库连接、事务提交、事务回滚等对象及操作
 * 在事务对象中具体决定采用的具体数据源
 */
public class MultiDataSourceTransaction implements Transaction {
 
    private final DataSource multidataSource;
 
    private final ConcurrentMap<String, Connection> curConMap;
 
    private boolean autoCommit;
 
 
    public MultiDataSourceTransaction(DataSource dataSource, boolean autoCommit) {
        //dataSource是DynamicDataSource类的实例
        this.multidataSource = dataSource;
        this.autoCommit = autoCommit;
        curConMap = new ConcurrentHashMap<>();
    }
 
 
    @Override
    public Connection getConnection() throws SQLException {
        String nowDsName = DataSourceContext.get();
        if (nowDsName == null || "".equals(nowDsName)) {//StringUtils.isEmpty(databaseIdentification)
            //databaseIdentification = DataSourceName.USERDATASOURCE.getName();
            throw new CannotGetJdbcConnectionException("未得到数据源名称");
            //nowDsName = "ym" ;
        }
        //获取数据源连接
        if (!this.curConMap.containsKey(nowDsName)) {
            try {
                Connection conn = this.multidataSource.getConnection();
                this.autoCommit = false;
                conn.setAutoCommit(false);
                this.curConMap.put(nowDsName, conn);
            } catch (SQLException ex) {
                throw new CannotGetJdbcConnectionException("未得到数据源JDBC连接", ex);
            }
        }
        return this.curConMap.get(nowDsName);
    }
 
 
    @Override
    public void commit() throws SQLException {
        for (Connection connection : curConMap.values()) {
            if (!autoCommit) {
                connection.commit();
            }
        }
    }
 
    @Override
    public void rollback() throws SQLException {
        for (Connection connection : curConMap.values()) {
            connection.rollback();
        }
    }
 
    @Override
    public void close() throws SQLException {
        for (Connection connection : curConMap.values()) {
            DataSourceUtils.releaseConnection(connection, this.multidataSource);
        }
    }
 
    @Override
    public Integer getTimeout() throws SQLException {
        return null;
    }
}