diff --git a/src/main/java/com/yutou/qqbot/QQNumberManager.java b/src/main/java/com/yutou/qqbot/QQNumberManager.java index 612925e..d885c09 100644 --- a/src/main/java/com/yutou/qqbot/QQNumberManager.java +++ b/src/main/java/com/yutou/qqbot/QQNumberManager.java @@ -13,56 +13,61 @@ import java.util.Set; public class QQNumberManager { private static QQNumberManager manager; - private QQNumberManager(){ + + private QQNumberManager() { } public static QQNumberManager getManager() { - if(manager==null) { - manager=new QQNumberManager(); + if (manager == null) { + manager = new QQNumberManager(); } return manager; } - public void addNumber(Long qq,boolean isGroup){ - if(RedisTools.exists(qq,null)){ + + public void addNumber(Long qq, boolean isGroup) { + if (RedisTools.exists(qq, null)) { return; } - JSONObject json=new JSONObject(); - json.put("group",isGroup); - json.put("power",new JSONArray()); - json.put("model",new JSONArray()); - RedisTools.set(qq,json.toJSONString()); + JSONObject json = new JSONObject(); + json.put("group", isGroup); + json.put("power", new JSONArray()); + json.put("model", new JSONArray()); + RedisTools.set(qq, json.toJSONString()); } - public List getNumber(){ - List list =new ArrayList<>(); - Jedis jedis=RedisTools.getRedis(); + + public List getNumber() { + List list = new ArrayList<>(); + Jedis jedis = RedisTools.getRedis(); jedis.select(RedisTools.QQBOT_USER); - Set set=jedis.keys("*"); + Set set = jedis.keys("*"); for (String s : set) { try { - list.add(Long.parseLong(s)); - }catch (Exception ignored){ + list.add(Long.parseLong(s)); + } catch (Exception ignored) { } } jedis.close(); return list; } - public boolean addPower(Long qq, String power){ - if(RedisTools.exists(qq,null)){ - JSONObject json=JSON.parseObject(RedisTools.get(qq)); - JSONArray array=json.getJSONArray("power"); + + public boolean addPower(Long qq, String power) { + if (RedisTools.exists(qq, null)) { + JSONObject json = JSON.parseObject(RedisTools.get(qq)); + JSONArray array = json.getJSONArray("power"); array.add(power); - json.put("power",array); - return RedisTools.set(qq,json.toJSONString()); + json.put("power", array); + return RedisTools.set(qq, json.toJSONString()); } return false; } - public List getPower(Long qq){ - List list=new ArrayList<>(); - if(RedisTools.exists(qq,null)){ - JSONObject json=JSON.parseObject(RedisTools.get(qq)); - JSONArray array=json.getJSONArray("power"); + + public List getPower(Long qq) { + List list = new ArrayList<>(); + if (RedisTools.exists(qq, null)) { + JSONObject json = JSON.parseObject(RedisTools.get(qq)); + JSONArray array = json.getJSONArray("power"); for (Object power : array) { list.add((String) power); } @@ -70,11 +75,12 @@ public class QQNumberManager { } return list; } + public List getUseModel(long qq) { - List list=new ArrayList<>(); - if(RedisTools.exists(qq,null)){ - JSONObject json=JSON.parseObject(RedisTools.get(qq)); - JSONArray array=json.getJSONArray("model"); + List list = new ArrayList<>(); + if (RedisTools.exists(qq, null)) { + JSONObject json = JSON.parseObject(RedisTools.get(qq)); + JSONArray array = json.getJSONArray("model"); for (Object power : array) { list.add((String) power); } @@ -82,43 +88,47 @@ public class QQNumberManager { } return list; } - public boolean delPower(Long qq, String power){ - if(RedisTools.exists(qq,null)){ - JSONObject json=JSON.parseObject(RedisTools.get(qq)); - JSONArray array=json.getJSONArray("power"); + + public boolean delPower(Long qq, String power) { + if (RedisTools.exists(qq, null)) { + JSONObject json = JSON.parseObject(RedisTools.get(qq)); + JSONArray array = json.getJSONArray("power"); array.remove(power); - json.put("power",array); - return RedisTools.set(qq,json.toJSONString()); + json.put("power", array); + return RedisTools.set(qq, json.toJSONString()); } return false; } - public boolean addUseModel(Long qq,Class modelClass){ - if(RedisTools.exists(qq,null)){ - JSONObject json=JSON.parseObject(RedisTools.get(qq)); - JSONArray array=json.getJSONArray("model"); + + public boolean addUseModel(Long qq, Class modelClass) { + if (RedisTools.exists(qq, null)) { + JSONObject json = JSON.parseObject(RedisTools.get(qq)); + JSONArray array = json.getJSONArray("model"); array.add(modelClass.getName()); - json.put("model",array); - return RedisTools.set(qq,json.toJSONString()); + json.put("model", array); + return RedisTools.set(qq, json.toJSONString()); } return false; } - public boolean delUseModel(Long qq,Class modelClass){ - if(RedisTools.exists(qq,null)){ - JSONObject json=JSON.parseObject(RedisTools.get(qq)); - JSONArray array=json.getJSONArray("model"); + + public boolean delUseModel(Long qq, Class modelClass) { + if (RedisTools.exists(qq, null)) { + JSONObject json = JSON.parseObject(RedisTools.get(qq)); + JSONArray array = json.getJSONArray("model"); array.remove(modelClass.getName()); - json.put("model",array); - return RedisTools.set(qq,json.toJSONString()); + json.put("model", array); + return RedisTools.set(qq, json.toJSONString()); } return false; } - public boolean isExistsPower(Long qq, String... power){ + + public boolean isExistsPower(Long qq, String... power) { //1 - if(RedisTools.exists(qq,null)){ - JSONObject json=JSON.parseObject(RedisTools.get(qq)); - JSONArray array=json.getJSONArray("power"); + if (RedisTools.exists(qq, null)) { + JSONObject json = JSON.parseObject(RedisTools.get(qq)); + JSONArray array = json.getJSONArray("power"); for (String key : power) { - if(!array.contains(key)){ + if (!array.contains(key)) { return false; } } @@ -126,18 +136,25 @@ public class QQNumberManager { } return false; } - public boolean isGroup(Long qq){ - if(RedisTools.exists(qq,null)){ - JSONObject json= JSON.parseObject(RedisTools.get(qq)); + + public boolean isGroup(Long qq) { + if (RedisTools.exists(qq, null)) { + JSONObject json = JSON.parseObject(RedisTools.get(qq)); return json.getBooleanValue("group"); } return false; } - public boolean isUseModel(Long qq,Class modelClass){ - if(RedisTools.exists(qq,null)){ - JSONObject json=JSON.parseObject(RedisTools.get(qq)); - JSONArray array=json.getJSONArray("model"); - return array.contains(modelClass.getName()); + + public boolean isUseModel(Long qq, Class modelClass) { + try { + if (RedisTools.exists(qq, null)) { + JSONObject json = JSON.parseObject(RedisTools.get(qq)); + JSONArray array = json.getJSONArray("model"); + return array.contains(modelClass.getName()); + } + } catch (Exception e) { + e.printStackTrace(); + Log.i("isUseModel", qq + " " + modelClass.getName()); } return false; }