rax-remote-2/vital-signs/src/main/java/com/rax/vital/timer/VitalSignTimer.java

package com.rax.vital.timer;

import com.rax.common.security.util.SecurityUtils;
import com.rax.vital.datasource.MongoDBSource;
import com.rax.vital.datasource.MySQLSource;
import com.rax.vital.medicine.service.*;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cloud.context.config.annotation.RefreshScope;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.messaging.simp.SimpMessagingTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketSession;

import java.sql.Connection;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 生命体征和用药信息推送
 *
 * @author zhaoyz
 * @date 2024/2/29
 */
@RefreshScope
@Component
@RequiredArgsConstructor
public class VitalSignTimer {

	private final SimpMessagingTemplate simpMessagingTemplate;

	private final VitalSignsService vitalSignsService;

	private final AIMedicineService aiMedicineService;

	private final DoctorMedicineService doctorMedicineService;

	private final FlagService flagService;

	private final RevulsionService revulsionService;

	// mongoDB定时任务容器
	private static final Map<String, TimerTask> timerMongoTaskMap = new ConcurrentHashMap<>();

	// mongoDB链接工具类容器
	private static final Map<String, MongoDBSource> mongoDBSourceMap = new ConcurrentHashMap<>();

	// mysql定时任务容器
	private static final Map<String, TimerTask> timerMysqlTaskMap = new ConcurrentHashMap<>();

	// mysql链接容器
	private static final Map<String, MySQLSource> mysqlConnectionMap = new ConcurrentHashMap();

	// MongoDB的地址
	@Value("${vital-sign.mongodb.host}")
	private String mongoDBHost;

	// MongoDB的用户名
	@Value("${vital-sign.mongodb.username}")
	private String mongoUsername;

	// MongoDB的用户的密码
	@Value("${vital-sign.mongodb.password}")
	private String mongoPassword;

	// mysql地址
	@Value("${vital-sign.mysql.host}")
	private String mysqlHost;

	// mysql用户名
	@Value("${vital-sign.mysql.username}")
	private String mysqlUsername;

	// mysql用户密码
	@Value("${vital-sign.mysql.password}")
	private String mysqlPassword;

	private static final Map<String, String> masterControlMap = new ConcurrentHashMap<>();

	/**
	 * 根据当前用户和患者数据库进行查询生命体征和用药信息并推送，数据库类型是MongoDB
	 *
	 * @author zhaoyz
	 */
	public void createAndSendMessageMongo(String database, String username, String simpSessionId) {

		synchronized (username) {
			if (!masterControlMap.containsKey(database)) {
				masterControlMap.put(database, username);
			}
		}

		TimerTask task = timerMongoTaskMap.get(simpSessionId);
		if (task != null) {
			return;
		}

		MongoDBSource mongoDBSource = mongoDBSourceMap.get(simpSessionId);
		if (mongoDBSource == null) {
			mongoDBSource = new MongoDBSource(mongoDBHost, mongoPassword, mongoUsername, database);
			mongoDBSourceMap.put(simpSessionId, mongoDBSource);
			mongoDBSource.open();
		}

		MongoDBSource finalMongoDBSource = mongoDBSource;
		TimerTask timerTask = new TimerTask() {
			@Override
			public void run() {

				MongoTemplate template = finalMongoDBSource.getConnection();
				HashMap<String, Object> result = new HashMap();
				List vitalSignsList = vitalSignsService.getVitalSignsList(template);
				result.put("vitalSignsList", vitalSignsList);
				List aiMedicineList = aiMedicineService.getAIMedicine(template);
				result.put("aiMedicineList", aiMedicineList);
				List docMedicineList = doctorMedicineService.getDocMedicine(template);
				result.put("docMedicineList", docMedicineList);
				List revulsionList = revulsionService.getRevulsionServiceList(template);
				result.put("revulsionList", revulsionList);
				Map flags = flagService.getFlags(template);
				result.put("flags", flags);

				simpMessagingTemplate.convertAndSendToUser(username + ":" + database, "/surgeryData", result);
			}
		};
		// 定时任务，设置1秒
		Timer timer = new Timer();
		timer.schedule(timerTask, 0, 1000);
		timerMongoTaskMap.put(simpSessionId, timerTask);
	}


	/**
	 * 根据当前用户和患者数据库进行查询生命体征和用药信息并推送，数据库类型是MySQL
	 *
	 * @param database
	 */
	public void createAndSendMessageMySQL(String database) {
		String account = SecurityUtils.getUser().getUsername();
		TimerTask task = timerMysqlTaskMap.get(account + "-" + database);
		if (task != null) {
			return;
		}

		MySQLSource mySQLSource = mysqlConnectionMap.get(database);
		if (mySQLSource == null) {
			mySQLSource = new MySQLSource(mysqlHost, mysqlPassword, mysqlUsername, database);
			mysqlConnectionMap.put(database, mySQLSource);
			mySQLSource.open();
			mySQLSource.increaseCount();
		}

		MySQLSource finalMySQLSource = mySQLSource;
		TimerTask timerTask = new TimerTask() {
			@Override
			public void run() {
				HashMap<String, Object> result = new HashMap();
				Connection connection = finalMySQLSource.getConnection();
				List<Map> vitalSignsList = vitalSignsService.getVitalSignsList(connection);
				result.put("vitalSignsList", vitalSignsList);
				List<Map> aiMedicineList = aiMedicineService.getAIMedicine(connection);
				result.put("aiMedicineList", aiMedicineList);
				List<Map> docMedicineList = doctorMedicineService.getDocMedicine(connection);
				result.put("docMedicineList", docMedicineList);
				Map flag = flagService.getFlag(connection);
				result.put("flag", flag);
				simpMessagingTemplate.convertAndSendToUser(account, "/doctorMedicine", result);
			}
		};
		// 定时任务，设置1秒
		Timer timer = new Timer();
		timer.schedule(timerTask, 0, 1000);
		timerMysqlTaskMap.put(account + "-" + database, timerTask);
	}

	/**
	 * 停止指定的某个用户查询的患者数据库定时器，数据库类型是MongoDB
	 *
	 * @author zhaoyz
	 */
	public synchronized void stopTimerTaskMongo(String simpSessionId) {
		TimerTask timerTask = timerMongoTaskMap.get(simpSessionId);
		if (timerTask != null) {
			timerTask.cancel();
			MongoDBSource mongoDBSource = mongoDBSourceMap.get(simpSessionId);
			mongoDBSource.close();
			timerMongoTaskMap.remove(simpSessionId);
			mongoDBSourceMap.remove(simpSessionId);
		}
	}

	/**
	 * 停止指定的某个用户查询的患者数据库定时器，数据库类型是MySQL
	 *
	 * @param database
	 * @param user
	 */
	public synchronized void stopTimerTaskMySQL(String database, String user) {
		TimerTask timerTask = timerMysqlTaskMap.get(user + "-" + database);
		if (timerTask != null) {
			timerTask.cancel();
			timerMysqlTaskMap.remove(user + "-" + database);

			MySQLSource mySQLSource = mysqlConnectionMap.get(database);
			mySQLSource.decreaseCount();
			int count = mySQLSource.getCount();
			if (count == 0) {
				mySQLSource.close();
				mysqlConnectionMap.remove(database);
			}
		}
	}

	public void changeAIFlag(String database, String username, String simpSessionId, String flag, String medicine, String value) {

		synchronized (username) {
			if (masterControlMap.containsKey(database) && masterControlMap.get(database).equals(username)) {
				MongoDBSource mongoDBSource = mongoDBSourceMap.get(simpSessionId);
				if (mongoDBSource == null) {
					mongoDBSource = new MongoDBSource(mongoDBHost, mongoPassword, mongoUsername, database);
					mongoDBSourceMap.put(simpSessionId, mongoDBSource);
					mongoDBSource.open();
				}

				MongoTemplate template = mongoDBSource.getConnection();
				aiMedicineService.changeAIFlagMedicine(template, flag, medicine, value);

				HashMap<String, Object> result = new HashMap();
				result.put("status", 0);
				result.put("flag", flag);
				result.put("msg", "");
				simpMessagingTemplate.convertAndSendToUser(username + ":" + database, "/medicineData", result);
			} else {
				HashMap<String, Object> result = new HashMap();
				result.put("status", 1);
				result.put("msg", "不是主控人员");
				simpMessagingTemplate.convertAndSendToUser(username + ":" + database, "/medicineData", result);
			}
		}
	}

}