320 lines
12 KiB
Java
320 lines
12 KiB
Java
package taurus.newRobot;
|
||
|
||
import com.taurus.core.plugin.redis.Redis;
|
||
import redis.clients.jedis.*;
|
||
|
||
import java.util.*;
|
||
import taurus.client.TaurusClient;
|
||
|
||
|
||
public class AddRoomRobot {
|
||
|
||
|
||
|
||
public void addRoom() {
|
||
Jedis jedis2 = Redis.use("group1_db2").getJedis();
|
||
Jedis jedis11 = Redis.use("group1_db11").getJedis();
|
||
System.out.println("进入1111111111111111111111111");
|
||
//给玩法分配机器人
|
||
if (jedis11.keys("group:*") != null) {
|
||
Set<String> groupIds = jedis11.keys("group:*");
|
||
for (String group : groupIds) {
|
||
String groupId = group.substring(group.indexOf(":") + 1);
|
||
String groupKey = "g{" + groupId + "}:play:*";
|
||
if (jedis11.keys(groupKey) != null) {
|
||
Set<String> playIds = jedis11.keys(groupKey);
|
||
// 遍历圈子下面的玩法
|
||
for (String pid : playIds) {
|
||
String playId = pid.substring(pid.indexOf("play:") + 5);
|
||
String playKey = "g{" + groupId + "}:play:" + playId;
|
||
|
||
String leftover_robot = "0";
|
||
if (jedis11.hget(playKey, "leftover_robot") != null) {
|
||
leftover_robot = jedis11.hget(playKey, "leftover_robot");
|
||
if (Integer.parseInt(leftover_robot) > 0) {
|
||
//给圈子玩法分配机器人，创建到redis
|
||
addPlayRobot(groupId, playId, leftover_robot);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
System.out.println("bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb");
|
||
//机器人主动找真人
|
||
joinzhenrenRoom();
|
||
System.out.println("进入22222222222222222222222");
|
||
|
||
//机器人主动进入退出， 这里进入的是中括号机器人退出的房间
|
||
zhongJoninRoom();
|
||
// System.out.println("进入3333333333333333333");
|
||
|
||
//机器人创建房间 进入房间
|
||
List<String> gKeys = scanKeysByPattern(jedis2, "g{*}:play:*");
|
||
try {
|
||
Map<String, List<String>> stringListMap = analyzePlayStructure(gKeys);
|
||
//取出 圈子玩法里的 机器人使用所有的状态
|
||
Map<String, Map<String, Map<String, String>>> playDetails = getPlayDetails(jedis2, stringListMap);
|
||
System.out.println("result" + playDetails);
|
||
|
||
//统计 玩法里 等待中的次数 为1的有几个 ，然后需要建立几个房间
|
||
Map<String, Map<String, Integer>> stringMapMap = analyzeAndMarkRoomsWithCircle(playDetails);
|
||
System.out.println("stringIntegerMap" + stringMapMap);
|
||
// 按圈子ID排序处理
|
||
List<String> circleIds = new ArrayList<>(stringMapMap.keySet());
|
||
Collections.sort(circleIds);
|
||
for (String circleId : circleIds) {
|
||
Map<String, Integer> playRoomMap = stringMapMap.get(circleId);
|
||
System.out.println("创建房间,进入房间");
|
||
jiqiren jiqiren = new jiqiren();
|
||
System.out.println("circleId +++" + circleId);
|
||
System.out.println("playRoomMap +++ " + playRoomMap);
|
||
jiqiren.createRoomsForCircle(circleId, playRoomMap);
|
||
}
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
} finally {
|
||
jedis2.close();
|
||
jedis11.close();
|
||
}
|
||
}
|
||
|
||
|
||
//机器人进入中括号房间
|
||
public void zhongJoninRoom() {
|
||
Jedis jedis0 = Redis.use().getJedis();
|
||
try {
|
||
//获取所有房间
|
||
Set<String> roomIds = jedis0.keys("room:*");
|
||
for (String id : roomIds) {
|
||
jiqiren jiqiren = new jiqiren();
|
||
jiqiren.tcp_jiqirezhongkouhao(id);
|
||
}
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
} finally {
|
||
jedis0.close();
|
||
}
|
||
}
|
||
|
||
|
||
//统计 玩法里 等待中的次数
|
||
// 方法1：返回 Map<圈子ID, Map<玩法ID, 房间需求>>
|
||
public Map<String, Map<String, Integer>> analyzeAndMarkRoomsWithCircle(
|
||
Map<String, Map<String, Map<String, String>>> result) {
|
||
|
||
Map<String, Map<String, Integer>> circlePlayRoomMap = new HashMap<>();
|
||
|
||
for (Map.Entry<String, Map<String, Map<String, String>>> circleEntry : result.entrySet()) {
|
||
String circleId = circleEntry.getKey();
|
||
Map<String, Map<String, String>> playData = circleEntry.getValue();
|
||
|
||
Map<String, Integer> playRoomMark = new HashMap<>();
|
||
|
||
for (Map.Entry<String, Map<String, String>> playEntry : playData.entrySet()) {
|
||
String playId = playEntry.getKey();
|
||
Map<String, String> robotData = playEntry.getValue();
|
||
|
||
// 统计值为"1"的机器人数量
|
||
int countOfOnes = 0;
|
||
for (String value : robotData.values()) {
|
||
if ("1".equals(value)) {
|
||
countOfOnes++;
|
||
}
|
||
}
|
||
|
||
// 根据逻辑进行标记
|
||
if (countOfOnes == 0) {
|
||
playRoomMark.put(playId, 2); // 需要建立2个房间
|
||
} else if (countOfOnes == 1) {
|
||
playRoomMark.put(playId, 1); // 标记为1
|
||
} else if (countOfOnes >= 2) {
|
||
playRoomMark.put(playId, 0); // 标记为0
|
||
}
|
||
}
|
||
|
||
circlePlayRoomMap.put(circleId, playRoomMark);
|
||
}
|
||
|
||
return circlePlayRoomMap;
|
||
}
|
||
|
||
|
||
// 统计每个玩法的机器人总数
|
||
public Map<String, Map<String, Map<String, String>>> getPlayDetails(Jedis jedis,
|
||
Map<String, List<String>> circleToPlays) {
|
||
|
||
Map<String, Map<String, Map<String, String>>> result = new HashMap<>();
|
||
|
||
for (Map.Entry<String, List<String>> entry : circleToPlays.entrySet()) {
|
||
String circleId = entry.getKey();
|
||
List<String> playIds = entry.getValue();
|
||
|
||
Map<String, Map<String, String>> playDetails = new HashMap<>();
|
||
|
||
for (String playId : playIds) {
|
||
// 构建Redis键
|
||
String redisKey = String.format("g{%s}:play:%s", circleId, playId);
|
||
|
||
// 获取Hash中的所有字段和值
|
||
Map<String, String> robotData = jedis.hgetAll(redisKey);
|
||
|
||
playDetails.put(playId, robotData);
|
||
}
|
||
|
||
result.put(circleId, playDetails);
|
||
}
|
||
return result;
|
||
}
|
||
|
||
|
||
//分割圈子和玩法
|
||
public static Map<String, List<String>> analyzePlayStructure(List<String> gKeys) {
|
||
Map<String, List<String>> circleToPlays = new HashMap<>();
|
||
|
||
for (String key : gKeys) {
|
||
// 解析键：g{圈子ID}:play:玩法ID
|
||
String[] parts = key.split(":");
|
||
|
||
if (parts.length >= 3) {
|
||
// 提取圈子ID，去掉 g{ 和 }
|
||
String circlePart = parts[0]; // g{330800}
|
||
String circleId = circlePart.substring(2, circlePart.length() - 1);
|
||
|
||
// 提取玩法ID
|
||
String playId = parts[2]; // 玩法ID
|
||
|
||
// 将玩法ID添加到对应圈子的列表中
|
||
circleToPlays.computeIfAbsent(circleId, k -> new ArrayList<>())
|
||
.add(playId);
|
||
}
|
||
}
|
||
|
||
return circleToPlays;
|
||
}
|
||
|
||
/**
|
||
* 使用SCAN命令安全地扫描匹配模式的key
|
||
*/
|
||
private static List<String> scanKeysByPattern(Jedis jedis, String pattern) {
|
||
List<String> keys = new ArrayList<>();
|
||
String cursor = ScanParams.SCAN_POINTER_START;
|
||
ScanParams scanParams = new ScanParams();
|
||
scanParams.match(pattern);
|
||
scanParams.count(1000); // 每次扫描数量
|
||
|
||
do {
|
||
ScanResult<String> scanResult = jedis.scan(cursor, scanParams);
|
||
cursor = String.valueOf(scanResult.getCursor());
|
||
keys.addAll(scanResult.getResult());
|
||
} while (!cursor.equals(ScanParams.SCAN_POINTER_START));
|
||
|
||
return keys;
|
||
}
|
||
|
||
|
||
//给圈子玩法分配机器人，创建到redis
|
||
public void addPlayRobot(String groupId, String playId, String leftover_robot) {
|
||
Jedis jedis2 = Redis.use("group1_db2").getJedis();
|
||
// 为这个玩法创建机器人配置
|
||
String robotPlayKey = "g{" + groupId + "}:play:" + playId;
|
||
boolean exists = jedis2.exists(robotPlayKey);
|
||
if (!exists) {
|
||
// 计算需要配置的机器人数量
|
||
int num = Integer.parseInt(leftover_robot) * 2;
|
||
|
||
// 获取所有可用的机器人
|
||
List<String> availableRobots1 = getAvailableRobots();
|
||
|
||
List<String> availableRobots = new ArrayList<>();
|
||
|
||
if (availableRobots1.size() > 0) {
|
||
for (String key : availableRobots1) {
|
||
String robotId = key.substring(key.indexOf(":") + 1);
|
||
availableRobots.add(robotId);
|
||
}
|
||
// 检查是否有足够的机器人
|
||
if (availableRobots.size() < num) {
|
||
System.out.println("警告：玩法 " + playId + " 需要 " + num + " 个机器人，但只有 " + availableRobots.size() + " 个可用");
|
||
return;
|
||
}
|
||
}
|
||
|
||
|
||
// 存储机器人信息，使用Hash结构
|
||
Map<String, String> robotConfigs = new HashMap<>();
|
||
|
||
for (int i = 0; i < num; i++) {
|
||
String robotId = availableRobots.get(i);
|
||
// key格式：robot:机器人ID，value：使用状态（0-未使用）
|
||
robotConfigs.put(robotId, "0");
|
||
String robotKey = "{robot}:" + robotId;
|
||
jedis2.hset(robotKey, "start", "1");
|
||
}
|
||
|
||
// 批量设置机器人配置
|
||
if (!robotConfigs.isEmpty()) {
|
||
jedis2.hmset(robotPlayKey, robotConfigs);
|
||
System.out.println("玩法 " + playId + " 已配置 " + num + " 个机器人");
|
||
}
|
||
jedis2.close();
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取所有可用的机器人（start=0）
|
||
*
|
||
* @return 可用机器人ID列表
|
||
*/
|
||
public List<String> getAvailableRobots() {
|
||
Jedis jedis2 = Redis.use("group1_db2").getJedis();
|
||
|
||
Set<String> allRobotKeys = jedis2.keys("{robot}:*");
|
||
List<String> availableRobots = new ArrayList<>();
|
||
|
||
// 使用Pipeline提高效率
|
||
Pipeline pipeline = jedis2.pipelined();
|
||
List<Response<String>> responses = new ArrayList<>();
|
||
|
||
for (String robotKey : allRobotKeys) {
|
||
responses.add(pipeline.hget(robotKey, "start"));
|
||
}
|
||
|
||
pipeline.sync();
|
||
|
||
// 提取机器人ID并筛选
|
||
int index = 0;
|
||
for (String robotKey : allRobotKeys) {
|
||
String startStatus = responses.get(index).get();
|
||
if ("0".equals(startStatus)) {
|
||
String robotId = robotKey.substring(robotKey.indexOf(":") + 1);
|
||
availableRobots.add(robotId);
|
||
}
|
||
index++;
|
||
}
|
||
jedis2.close();
|
||
return availableRobots;
|
||
}
|
||
|
||
public void joinzhenrenRoom() {
|
||
Jedis jedis0 = Redis.use().getJedis();
|
||
Jedis jedis11 = Redis.use("group1_db11").getJedis();
|
||
Jedis jedis2 = Redis.use("group1_db2").getJedis();
|
||
try {
|
||
//获取所有房间
|
||
Set<String> roomIds = jedis0.keys("room:*");
|
||
for (String id : roomIds) {
|
||
jiqiren jiqiren = new jiqiren();
|
||
jiqiren.tcp_jiqirenzhenren(id);
|
||
}
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
} finally {
|
||
jedis0.close();
|
||
jedis11.close();
|
||
jedis2.close();
|
||
}
|
||
|
||
}
|
||
}
|