init commit

yangys

2024-03-27 e48aa2ac8dea1be5db11c63edf0b912c4ad5ce65

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
87
package com.qianwen.smartman.modules.visual.service.impl;
 
import com.baomidou.dynamic.datasource.toolkit.DynamicDataSourceContextHolder;
import java.sql.Connection;
import java.sql.DriverManager;
import java.util.LinkedHashMap;
import java.util.List;
import com.qianwen.core.log.exception.ServiceException;
import com.qianwen.core.mp.base.BaseServiceImpl;
import com.qianwen.core.secure.BladeUser;
import com.qianwen.core.secure.utils.AuthUtil;
import com.qianwen.core.tool.utils.BeanUtil;
import com.qianwen.core.tool.utils.PlaceholderUtil;
import com.qianwen.smartman.modules.visual.dto.VisualDbDTO;
import com.qianwen.smartman.modules.visual.dynamic.DynamicDataSourceHolder;
import com.qianwen.smartman.modules.visual.entity.VisualDb;
import com.qianwen.smartman.modules.visual.mapper.VisualDbMapper;
import com.qianwen.smartman.modules.visual.service.IVisualDbService;
import org.springframework.stereotype.Service;
 
@Service
/* loaded from: blade-api.jar:BOOT-INF/classes/org/springblade/modules/visual/service/impl/VisualDbServiceImpl.class */
public class VisualDbServiceImpl extends BaseServiceImpl<VisualDbMapper, VisualDb> implements IVisualDbService {
    private final VisualDbMapper mapper;
    private final DynamicDataSourceHolder holder;
 
    public VisualDbServiceImpl(final VisualDbMapper mapper, final DynamicDataSourceHolder holder) {
        this.mapper = mapper;
        this.holder = holder;
    }
 
    @Override // org.springblade.modules.visual.service.IVisualDbService
    public Boolean dbTest(VisualDb db) {
        Connection conn = null;
        try {
            try {
                Class.forName(db.getDriverClass());
                conn = DriverManager.getConnection(db.getUrl(), db.getUsername(), db.getPassword());
                conn.setAutoCommit(Boolean.FALSE.booleanValue());
                dbClose(conn);
                return true;
            } catch (Exception ex) {
                ex.printStackTrace();
                dbClose(conn);
                return false;
            }
        } catch (Throwable th) {
            dbClose(conn);
            throw th;
        }
    }
 
    private void dbClose(Connection conn) {
        if (conn != null) {
            try {
                conn.close();
            } catch (Exception ex) {
                ex.printStackTrace();
            }
        }
    }
 
    @Override // org.springblade.modules.visual.service.IVisualDbService
    public List<VisualDbDTO> dbList() {
        return this.mapper.dbList();
    }
 
    @Override // org.springblade.modules.visual.service.IVisualDbService
    public List<LinkedHashMap<String, Object>> dynamicQuery(String id, String sql) {
        try {
            try {
                this.holder.handleDataSource(id);
                DynamicDataSourceContextHolder.push(id);
                BladeUser user = AuthUtil.getUser();
                String dynamicSql = PlaceholderUtil.getDefaultResolver().resolveByMap(sql, BeanUtil.toMap(user));
                List<LinkedHashMap<String, Object>> dynamicQuery = this.mapper.dynamicQuery(dynamicSql);
                DynamicDataSourceContextHolder.poll();
                return dynamicQuery;
            } catch (Exception exception) {
                throw new ServiceException(exception.getMessage());
            }
        } catch (Throwable th) {
            DynamicDataSourceContextHolder.poll();
            throw th;
        }
    }
}