“第一次提交”

付延余

2022-12-16 f0f8ee8c4a945adbc742d9bab69382b28ad311fb

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
88
89
90
91
package com.wgcloud.util;
 
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoClients;
import com.mongodb.client.MongoIterable;
import com.wgcloud.entity.DbInfo;
import com.wgcloud.service.DbInfoService;
import com.wgcloud.service.LogInfoService;
import com.wgcloud.util.msg.WarnMailUtil;
import com.wgcloud.util.msg.WarnPools;
import com.wgcloud.util.staticvar.StaticKeys;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
 
import javax.annotation.Resource;
 
/**
 * @version v3.4
 * @ClassName: MongoDbUtil
 * @author: http://www.wgstart.com
 * @date: 2022年11月10日
 * @Description: MongoDb工具类
 * @Copyright: 2019-2022 wgcloud. All rights reserved.
 */
@Component
public class MongoDbUtil {
 
    private static final Logger logger = LoggerFactory.getLogger(MongoDbUtil.class);
 
    @Resource
    private LogInfoService logInfoService;
    @Resource
    private DbInfoService dbInfoService;
 
 
    /**
     * 连接mongodb
     *
     * @param dbInfo
     * @return
     */
    public void connectMongoDb(DbInfo dbInfo) throws Exception {
        MongoClient mongoClient = null;
        try {
            dbInfo.setDbState(StaticKeys.ON_STATE);
            //连接指定的MongoDB，只需要dbUrl
            mongoClient = MongoClients.create(dbInfo.getDbUrl());
 
            //打印
            logger.info("mongoClient----------" + mongoClient);
 
            if (null != mongoClient) {
                MongoIterable<String> listDataBaseNames = mongoClient.listDatabaseNames();
                for (String dbName : listDataBaseNames) {
                    logger.info("连接mongodb dbName-------" + dbName);
                }
                //关闭mongoClient
                mongoClient.close();
                mongoClient = null;
                if (null != WarnPools.MEM_WARN_MAP && null != WarnPools.MEM_WARN_MAP.get(dbInfo.getId())) {
                    Runnable runnable = () -> {
                        WarnMailUtil.sendDbDown(dbInfo, false);
                    };
                    ThreadPoolUtil.executor.execute(runnable);
                }
                //只有数据源在线时候，才更新时间，连接失败就不更新时间
                dbInfo.setCreateTime(DateUtil.getNowTime());
            } else {
                dbInfo.setDbState(StaticKeys.DOWN_STATE);
            }
            dbInfoService.updateById(dbInfo);
        } catch (Exception e) {
            dbInfo.setDbState(StaticKeys.DOWN_STATE);
            logger.error("连接mongodb错误", e);
            logInfoService.save("连接mongodb错误：" + dbInfo.getAliasName(), "数据库别名：" + dbInfo.getAliasName() + "，" + e.toString(), StaticKeys.LOG_XTCZ);
            dbInfoService.updateById(dbInfo);
            Runnable runnable = () -> {
                WarnMailUtil.sendDbDown(dbInfo, true);
            };
            ThreadPoolUtil.executor.execute(runnable);
            //关闭mongoClient
            if (null != mongoClient) {
                mongoClient.close();
                mongoClient = null;
            }
        }
    }
 
 
}