移植之前版本所有功能

This commit is contained in:
Yutousama 2021-12-07 21:59:41 +08:00
parent 06cb246af7
commit 697bf302ac
24 changed files with 1127 additions and 258 deletions

View File

@ -0,0 +1,11 @@
package com.yutou.qqbot.Annotations;
import java.lang.annotation.*;
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface UseModel {
String value() default "";
}

View File

@ -0,0 +1,62 @@
package com.yutou.qqbot.Listeners;
import com.yutou.qqbot.MessageEvent.AdminMessage;
import com.yutou.qqbot.QQNumberManager;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.Log;
import kotlin.coroutines.CoroutineContext;
import net.mamoe.mirai.event.EventHandler;
import net.mamoe.mirai.event.ListeningStatus;
import net.mamoe.mirai.event.SimpleListenerHost;
import net.mamoe.mirai.event.events.GroupMessageEvent;
import net.mamoe.mirai.event.events.MessageEvent;
import org.jetbrains.annotations.NotNull;
public class QQMessageListener extends SimpleListenerHost {
@Override
public void handleException(@NotNull CoroutineContext context, @NotNull Throwable exception) {
// super.handleException(context, exception);
exception.printStackTrace();
}
@EventHandler
public ListeningStatus onMessage(MessageEvent event) {
long qqNumber;
boolean isGroup;
if (event instanceof GroupMessageEvent) {
qqNumber = ((GroupMessageEvent) event).getGroup().getId();
isGroup = true;
} else {
isGroup = false;
qqNumber = event.getSource().getFromId();
}
QQNumberManager.getManager().addNumber(qqNumber, isGroup);
if (event.getSource().getFromId() == 583819556) {
new AdminMessage(qqNumber, event.getMessage().contentToString());
}
System.out.println("isGroup = " + isGroup);
for (Class<?> model : Model.classList) {
if (QQNumberManager.getManager().isUseModel(qqNumber, model)) {
try {
Model useModel = (Model) model.getDeclaredConstructor().newInstance();
if (!useModel.isUserPublic()) {
if (!QQNumberManager.getManager().isExistsPower(event.getSource().getFromId(), useModel.getUsePowers())) {
Log.i("用户没有权限",event.getSource().getFromId());
break;
}
}
useModel.onMessage(qqNumber, event, isGroup);
} catch (Exception e) {
e.printStackTrace();
}
}
}
return ListeningStatus.LISTENING; // 表示继续监听事件
}
}

View File

@ -0,0 +1,180 @@
package com.yutou.qqbot.MessageEvent;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.QQNumberManager;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.StringUtils;
import java.util.List;
public class AdminMessage extends Message {
public static class QQCommands {
private static final String POWER_ADD = "!新增权限";
private static final String POWER_SHOW = "!查看权限";
private static final String POWER_DEL = "!删除权限";
private static final String MODEL_ADD = "!新增模块";
private static final String MODEL_SHOW = "!查看模块";
private static final String MODEL_DEL = "!删除模块";
}
public AdminMessage(Long qq, String msg) {
sendQQ = qq;
String msgSrc = msg;
String[] msgGroup = msg.replace("", "!").split(" ");
System.err.println(Model.classList.size());
for (Class<?> aClass : Model.classList) {
System.err.println(aClass.getName());
}
String[] commands;
switch (msgGroup[0]) {
case QQCommands.POWER_ADD -> {
if (msgGroup.length == 1) {
StringBuilder builder = Model.getCommands(Model.QQGroupCommands.class);
builder.append("\n");
builder.append(Model.getCommands(Model.QQFromCommands.class));
QQBotManager.getInstance().sendMessage(qq, "使用方法:" + QQCommands.POWER_ADD + " [qqnumber] 权限" + "\n" + "可设置以下权限,用&分割");
QQBotManager.getInstance().sendMessage(qq, builder.toString());
} else {
if (msgGroup.length <= 2) {
addPower(msgGroup[1], null);
} else {
addPower(msgGroup[1], msgGroup[2]);
}
}
}
case QQCommands.POWER_SHOW -> {
if (msgGroup.length == 1) {
QQBotManager.getInstance().sendMessage(qq, "使用方法:" + QQCommands.POWER_SHOW + " [qqnumber]");
} else {
List<String> list = QQNumberManager.getManager().getPower(Long.parseLong(msgGroup[1]));
StringBuilder builder = new StringBuilder();
for (String power : list) {
builder.append(power).append("\n");
}
QQBotManager.getInstance().sendMessage(qq, Long.parseLong(msgGroup[1]) + " 拥有以下权限:" + builder);
}
}
case QQCommands.POWER_DEL -> {
if (msgGroup.length == 1) {
QQBotManager.getInstance().sendMessage(qq, "使用方法:" + QQCommands.POWER_DEL + " [qqnumber] 权限 用&分割");
} else {
if (msgGroup.length <= 2) {
delPower(msgGroup[1], null);
} else {
delPower(msgGroup[1], msgGroup[2]);
}
}
}
case QQCommands.MODEL_ADD -> {
if (msgGroup.length == 1) {
StringBuilder builder = new StringBuilder();
for (Class<?> aClass : Model.classList) {
builder.append(aClass.getName()).append("\n");
}
QQBotManager.getInstance().sendMessage(qq, "使用方法:" + QQCommands.MODEL_ADD + " [qqnumber] 模块" + "\n" + "可设置以下模块,用&分割");
QQBotManager.getInstance().sendMessage(qq, builder.toString());
} else {
if (msgGroup.length <= 2) {
addModel(msgGroup[1], null);
} else {
addModel(msgGroup[1], msgGroup[2]);
}
}
}
case QQCommands.MODEL_SHOW -> {
if (msgGroup.length == 1) {
QQBotManager.getInstance().sendMessage(qq, "使用方法:" + QQCommands.MODEL_SHOW + " [qqnumber]");
} else {
List<String> list = QQNumberManager.getManager().getUseModel(Long.parseLong(msgGroup[1]));
StringBuilder builder = new StringBuilder();
for (String power : list) {
builder.append(power).append("\n");
}
QQBotManager.getInstance().sendMessage(qq, Long.parseLong(msgGroup[1]) + " 拥有以下模块:" + builder);
}
}
case QQCommands.MODEL_DEL -> {
if (msgGroup.length == 1) {
QQBotManager.getInstance().sendMessage(qq, "使用方法:" + QQCommands.MODEL_DEL + " [qqnumber] 模块 用&分割");
} else {
if (msgGroup.length <= 2) {
delModel(msgGroup[1], null);
} else {
delModel(msgGroup[1], msgGroup[2]);
}
}
}
}
}
private void addPower(String qq, String msg) {
boolean flag = false;
if (msg == null) {
StringBuilder builder = Model.getCommands(Model.QQGroupCommands.class);
builder.append("\n");
builder.append(Model.getCommands(Model.QQFromCommands.class));
for (String power : builder.toString().split("\n")) {
if (!StringUtils.isEmpty(power)) {
flag = QQNumberManager.getManager().addPower(Long.parseLong(qq), power);
}
}
} else {
for (String power : msg.split("&")) {
flag = QQNumberManager.getManager().addPower(Long.parseLong(qq), power);
}
}
QQBotManager.getInstance().sendMessage(sendQQ, flag ? "新增成功" : "新增失败");
}
private void delPower(String qq, String msg) {
if (msg == null) {
List<String> list = QQNumberManager.getManager().getPower(Long.parseLong(qq));
for (String power : list) {
QQNumberManager.getManager().delPower(Long.parseLong(qq), power);
}
} else {
for (String power : msg.split("&")) {
QQNumberManager.getManager().delPower(Long.parseLong(qq), power);
}
}
}
private void addModel(String qq, String model) {
boolean flag = false;
try {
if (model == null) {
for (Class<?> power : Model.classList) {
flag = QQNumberManager.getManager().addUseModel(Long.parseLong(qq), power);
}
} else {
for (String power : model.split("&")) {
flag = QQNumberManager.getManager().addUseModel(Long.parseLong(qq), Class.forName(power));
}
}
} catch (Exception e) {
e.printStackTrace();
}
QQBotManager.getInstance().sendMessage(sendQQ, flag ? "新增成功" : "新增失败");
}
private void delModel(String qq, String model) {
try {
if (model == null) {
List<String> list = QQNumberManager.getManager().getUseModel(Long.parseLong(qq));
for (String power : list) {
QQNumberManager.getManager().delUseModel(Long.parseLong(qq), Class.forName(power));
}
} else {
for (String power : model.split("&")) {
QQNumberManager.getManager().delUseModel(Long.parseLong(qq), Class.forName(power));
}
}
}catch (Exception e){
e.printStackTrace();
}
}
}

View File

@ -0,0 +1,7 @@
package com.yutou.qqbot.MessageEvent;
public class Message {
public long sendQQ=891655174L;
}

View File

@ -1,54 +1,39 @@
package com.yutou.qqbot;
import com.yutou.qqbot.interfaces.DownloadInterface;
import com.yutou.qqbot.models.audio.QQAudio;
import com.yutou.qqbot.Listeners.QQMessageListener;
import com.yutou.qqbot.models.Commands.Bangumi;
import com.yutou.qqbot.models.Commands.System.*;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.models.setu.QQSetu;
import com.yutou.qqbot.utlis.*;
import net.mamoe.mirai.Bot;
import net.mamoe.mirai.BotFactory;
import net.mamoe.mirai.event.GlobalEventChannel;
import net.mamoe.mirai.event.events.GroupMessageEvent;
import net.mamoe.mirai.message.data.Image;
import net.mamoe.mirai.message.data.MessageChainBuilder;
import net.mamoe.mirai.utils.BotConfiguration;
import net.mamoe.mirai.utils.ExternalResource;
import java.io.File;
import java.lang.reflect.Field;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Objects;
import java.util.function.Consumer;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class QQBotManager {
public static class QQCommands {
private final static String QQ_HELP = "!help";
private final static String QQ_SYSTEM_RESTART = "!restart";
private final static String QQ_UPDATE_IP = "!更新ip";
private final static String QQ_GET_IP = "!ip";
private final static String QQ_OPEN_PC = "!开机";
private final static String QQ_GET_VERSION = "!version";
private final static String QQ_CMD = "!cmd";
private final static String QQ_BANGUMI_TODAY = "!今日动画";
private final static String QQ_BANGUMI_LIST = "!新番";
private final static String QQ_BANGUMI_SUB = "!查动画";
private final static String QQ_AUDIO = "!语音";
private final static String QQ_AUDIO_OPEN_LAMP = "!开灯";
private final static String QQ_AUDIO_OPEN_AIR = "!开空调";
private final static String QQ_BT_RELOAD = "!刷bt";
private final static String QQ_TOOLS_IDEA = "!idea";
private final static String QQ_TOOLS_IDEA_FILE = "!idea>";
private final static String QQ_TOOLS_IDEA_URL = "!idea_url";
public final static String QQ_BANGUMI_INFO="!保存动画信息>";
static {
Model.classList.add(QQSetu.class);
Model.classList.add(Audio.class);
Model.classList.add(BtFlash.class);
Model.classList.add(Cmd.class);
Model.classList.add(Help.class);
Model.classList.add(IP.class);
Model.classList.add(OpenPC.class);
Model.classList.add(Restart.class);
Model.classList.add(ToolsIdea.class);
Model.classList.add(UpdateIP.class);
Model.classList.add(Version.class);
Model.classList.add(Bangumi.class);
}
private static QQBotManager botManager = null;
private Bot bot;
private static final long qqGroup = 891655174L;
@ -82,7 +67,8 @@ public class QQBotManager {
}
});
//Events.registerEvents(bot, new MessageListener());
GlobalEventChannel.INSTANCE.subscribeAlways(GroupMessageEvent.class, new MessageListener());
GlobalEventChannel.INSTANCE.registerListenerHost(new QQMessageListener());
// GlobalEventChannel.INSTANCE.subscribeAlways(GroupMessageEvent.class, new MessageListener());
bot.login();
new Thread(new Runnable() {
@Override
@ -115,9 +101,13 @@ public class QQBotManager {
return isLogin;
}
private Image getImage(File file) {
private Image getImage(File file,Long qq) {
if (bot != null) {
return Objects.requireNonNull(bot.getGroup(qqGroup)).uploadImage(ExternalResource.create(file));
if(QQNumberManager.getManager().isGroup(qq)) {
return Objects.requireNonNull(bot.getGroup(qq)).uploadImage(ExternalResource.create(file));
}else{
return Objects.requireNonNull(bot.getFriend(qq)).uploadImage(ExternalResource.create(file));
}
}
return null;
}
@ -126,9 +116,6 @@ public class QQBotManager {
return "没有登录QQ";
}
public void reportToDayBangumi() {
getInstance().sendMessage(BangumiTools.reportToDayBangumi());
}
public String sendMessage(String text) {
if (bot != null&&!StringUtils.isEmpty(text)) {
@ -144,7 +131,11 @@ public class QQBotManager {
public String sendMessage(Long group, String text) {
if (bot != null) {
try {
if(QQNumberManager.getManager().isGroup(group)) {
return Objects.requireNonNull(bot.getGroup(group)).sendMessage(text).toString();
}else{
return Objects.requireNonNull(bot.getFriend(group)).sendMessage(text).toString();
}
} catch (Exception e) {
e.printStackTrace();
}
@ -154,21 +145,28 @@ public class QQBotManager {
public void sendMessage(Long group, MessageChainBuilder builder) {
if (bot != null) {
if(QQNumberManager.getManager().isGroup(group)) {
Objects.requireNonNull(bot.getGroup(group)).sendMessage(builder.asMessageChain());
}else{
Objects.requireNonNull(bot.getFriend(group)).sendMessage(builder.asMessageChain());
}
}
}
public String sendMessage(File imageFile, String text) {
public String sendMessage(File imageFile,Long qq,String text){
try {
if (bot != null) {
Image image = getImage(imageFile);
Image image = getImage(imageFile,qq);
MessageChainBuilder builder = new MessageChainBuilder();
if (image != null) {
builder.append(image);
}
builder.append(text);
return Objects.requireNonNull(bot.getGroup(qqGroup)).sendMessage(builder.asMessageChain()).toString();
if(QQNumberManager.getManager().isGroup(qq)) {
return Objects.requireNonNull(bot.getGroup(qq)).sendMessage(builder.asMessageChain()).toString();
}else{
return Objects.requireNonNull(bot.getFriend(qq)).sendMessage(builder.asMessageChain()).toString();
}
}
} catch (Exception e) {
e.printStackTrace();
@ -176,192 +174,33 @@ public class QQBotManager {
return getNotLoginQQ();
}
public String sendMessage(File imageFile, String text) {
return sendMessage(imageFile,qqGroup,text);
}
public String sendMessage(List<File> imgs, String text) {
public String sendMessage(List<File> imgs,Long qq, String text) {
if (bot != null) {
MessageChainBuilder builder = new MessageChainBuilder();
for (File img : imgs) {
builder.append(Objects.requireNonNull(getImage(img)));
builder.append(Objects.requireNonNull(getImage(img,qq)));
}
builder.append(text);
return Objects.requireNonNull(bot.getGroup(qqGroup)).sendMessage(builder.asMessageChain()).toString();
if(QQNumberManager.getManager().isGroup(qq)) {
return Objects.requireNonNull(bot.getGroup(qq)).sendMessage(builder.asMessageChain()).toString();
}else{
return Objects.requireNonNull(bot.getFriend(qq)).sendMessage(builder.asMessageChain()).toString();
}
}
return getNotLoginQQ();
}
public static List<String> getImages(String str) {
List<String> list = new ArrayList<>();
String regex = "<img(.*?)/img>";
Pattern pattern = Pattern.compile(regex);
Matcher matcher = pattern.matcher(str);
while (matcher.find()) {
list.add(matcher.group().replace("<img", "")
.replace("/img>", "")
.trim());
}
return list;
}
public static void main(String[] args) {
getInstance();
}
private static class MessageListener implements Consumer<GroupMessageEvent> {
@Override
public void accept(GroupMessageEvent event) {
String msg = event.getMessage().contentToString();
switch (event.getGroup().getId() + "") {
case qqGroup + "":
myGroup(msg);
case "570197155":
new QQSetu().setu(msg, event);
}
}
private void myGroup(String msg) {
StringBuilder builder;
String msgSrc=msg;
msg = msg.replace("", "!").toLowerCase();
Log.i("QQBot","command = "+msg);
switch (msg) {
case QQCommands.QQ_OPEN_PC:
RedisTools.Consumer.system("openPC", null);
String time = new SimpleDateFormat("HH").format(new Date());
if (Integer.parseInt(time) < 18) {
break;
}
case QQCommands.QQ_AUDIO_OPEN_LAMP:
QQAudio.playText("小爱同学,开灯");
break;
case QQCommands.QQ_AUDIO_OPEN_AIR:
QQAudio.playText("小爱同学,开空调");
break;
case QQCommands.QQ_UPDATE_IP:
RedisTools.Consumer.system("updateIP", null);
break;
case QQCommands.QQ_GET_IP:
RedisTools.Consumer.bot("getip");
break;
case QQCommands.QQ_GET_VERSION:
sendVersion();
break;
case QQCommands.QQ_BANGUMI_TODAY:
RedisTools.remove("reportToDayBangumi");
QQBotManager.getInstance().sendMessage(BangumiTools.reportToDayBangumi());
break;
case QQCommands.QQ_BANGUMI_LIST:
getInstance().sendMessage("获取中...");
getInstance().sendMessage(BangumiTools.reportBangumiList());
break;
case QQCommands.QQ_BT_RELOAD:
HttpTools.get("http://192.168.31.88:8000/bt/down.do");
break;
case QQCommands.QQ_SYSTEM_RESTART:
getInstance().sendMessage("正在重启服务");
System.out.println("结束进程");
AppTools.exec("cd /home/yutou/public/servier/tools && ./start.sh",null,true,false);
break;
case QQCommands.QQ_HELP:
builder = new StringBuilder();
for (Field field : QQCommands.class.getDeclaredFields()) {
try {
field.setAccessible(true);
builder.append(field.get(null)).append("\n");
} catch (IllegalAccessException e) {
e.printStackTrace();
}
}
getInstance().sendMessage(builder.toString());
break;
case QQCommands.QQ_TOOLS_IDEA:
builder = new StringBuilder();
if(StringUtils.isEmpty(RedisTools.get("ideaUrl"))){
builder.append("暂未设置IDEA激活码下载地址");
}else {
for (String name : IdeaTools.getIdeaList(RedisTools.get("ideaUrl"))) {
builder.append(QQCommands.QQ_TOOLS_IDEA_FILE).append(name).append("\n");
}
if(builder.toString().trim().length()==0){
builder.append("激活码文件中未包含txt文件");
}
}
getInstance().sendMessage(builder.toString());
break;
default:
if (msg.startsWith(QQCommands.QQ_CMD)) {
RedisTools.Consumer.system("cmd", msg.replace(QQCommands.QQ_CMD, ""));
} else if (msg.startsWith(QQCommands.QQ_BANGUMI_SUB)) {
List<String> infos = null;
try {
int id = Integer.parseInt(msg.replace(QQCommands.QQ_BANGUMI_SUB, "").trim());
infos = BangumiTools.reportBangumiInfo(id);
} catch (Exception e) {
String key = msg.replace(QQCommands.QQ_BANGUMI_SUB, "").trim();
infos = BangumiTools.reportSearchBangumi(key);
}
for (String info : infos) {
List<String> imgs = new ArrayList<>();
if (info.contains("<img") && info.contains(" /img>")) {
imgs = getImages(info);
for (String img : imgs) {
info = info.replace("<img " + img + " /img>", "");
}
}
sendImagesMsg(imgs, info);
}
} else if (msg.startsWith(QQCommands.QQ_AUDIO)) {
QQAudio.playText(msg.replace(QQCommands.QQ_AUDIO, ""));
}else if(msg.startsWith(QQCommands.QQ_TOOLS_IDEA_FILE)){
getInstance().sendMessage(IdeaTools.getIdea(msgSrc.replace(QQCommands.QQ_TOOLS_IDEA_FILE,"")));
}else if(msg.startsWith(QQCommands.QQ_TOOLS_IDEA_URL)){
RedisTools.set("ideaUrl",msg.replace(QQCommands.QQ_TOOLS_IDEA_URL,"").trim());
getInstance().sendMessage("已设定下载地址:"+RedisTools.get("ideaUrl"));
}
}
}
private List<File> files;
private int index = 0;
private void sendImagesMsg(List<String> imgs, String text) {
files = new ArrayList<>();
index = 0;
if (imgs.size() == 0) {
getInstance().sendMessage(text);
return;
}
for (String img : imgs) {
HttpTools.download(img,null, new DownloadInterface() {
@Override
public void onDownload(File file) {
super.onDownload(file);
files.add(file);
send(imgs.size(), text);
}
@Override
public void onError(Exception e) {
super.onError(e);
index++;
send(imgs.size(), text);
}
});
}
}
private void send(int size, String text) {
if ((files.size() + index) == size) {
String str = getInstance().sendMessage(files, text);
Log.i("str = " + str);
}
}
private void sendVersion() {
public void sendVersion() {
String localVersion = QQBotApplication.version;
String serverVersion = HttpTools.get("http://tools.yutou233.cn:8000/public/version.do?token=zIrsh9TUZP2lfRW753PannG49E7VJvor");
String msg = "本地版本:" + localVersion + "\n" + "服务器版本:" + serverVersion;
@ -369,5 +208,6 @@ public class QQBotManager {
AppTools.sendServer("服务版本查询", msg);
}
}
}

View File

@ -0,0 +1,140 @@
package com.yutou.qqbot;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.yutou.qqbot.utlis.Log;
import com.yutou.qqbot.utlis.RedisTools;
import redis.clients.jedis.Jedis;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
public class QQNumberManager {
private static QQNumberManager manager;
private QQNumberManager(){
}
public static QQNumberManager getManager() {
if(manager==null) {
manager=new QQNumberManager();
}
return manager;
}
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());
}
public List<Long> getNumber(){
List<Long> list =new ArrayList<>();
Jedis jedis=RedisTools.getRedis();
jedis.select(3);
Set<String> set=jedis.keys("*");
for (String s : set) {
list.add(Long.parseLong(s));
}
jedis.close();
return list;
}
public boolean addPower(Long qq, String power){
if(RedisTools.exists(qq,null)){
JSONObject json=JSONObject.parseObject(RedisTools.get(qq));
JSONArray array=json.getJSONArray("power");
array.add(power);
json.put("power",array);
return RedisTools.set(qq,json.toJSONString());
}
return false;
}
public List<String> getPower(Long qq){
List<String> list=new ArrayList<>();
if(RedisTools.exists(qq,null)){
JSONObject json=JSONObject.parseObject(RedisTools.get(qq));
JSONArray array=json.getJSONArray("power");
for (Object power : array) {
list.add((String) power);
}
return list;
}
return list;
}
public List<String> getUseModel(long qq) {
List<String> list=new ArrayList<>();
if(RedisTools.exists(qq,null)){
JSONObject json=JSONObject.parseObject(RedisTools.get(qq));
JSONArray array=json.getJSONArray("model");
for (Object power : array) {
list.add((String) power);
}
return list;
}
return list;
}
public boolean delPower(Long qq, String power){
if(RedisTools.exists(qq,null)){
JSONObject json=JSONObject.parseObject(RedisTools.get(qq));
JSONArray array=json.getJSONArray("power");
array.remove(power);
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=JSONObject.parseObject(RedisTools.get(qq));
JSONArray array=json.getJSONArray("model");
array.add(modelClass.getName());
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=JSONObject.parseObject(RedisTools.get(qq));
JSONArray array=json.getJSONArray("model");
array.remove(modelClass.getName());
json.put("model",array);
return RedisTools.set(qq,json.toJSONString());
}
return false;
}
public boolean isExistsPower(Long qq, String... power){
if(RedisTools.exists(qq,null)){
JSONObject json=JSONObject.parseObject(RedisTools.get(qq));
JSONArray array=json.getJSONArray("power");
for (String key : power) {
if(!array.contains(key)){
return false;
}
}
return true;
}
return false;
}
public boolean isGroup(Long qq){
if(RedisTools.exists(qq,null)){
JSONObject json=JSONObject.parseObject(RedisTools.get(qq));
return json.getBoolean("group");
}
return false;
}
public boolean isUseModel(Long qq,Class<?> modelClass){
if(RedisTools.exists(qq,null)){
JSONObject json=JSONObject.parseObject(RedisTools.get(qq));
JSONArray array=json.getJSONArray("model");
return array.contains(modelClass.getName());
}
return false;
}
}

View File

@ -0,0 +1,6 @@
package com.yutou.qqbot.interfaces;
public interface ModelInterface {
boolean isUserPublic();
String[] getUsePowers();
}

View File

@ -0,0 +1,15 @@
package com.yutou.qqbot.models.Animal;
import com.yutou.qqbot.models.Model;
public class TurnipProphet extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[0];
}
}

View File

@ -0,0 +1,126 @@
package com.yutou.qqbot.models.Commands;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.interfaces.DownloadInterface;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.BangumiTools;
import com.yutou.qqbot.utlis.HttpTools;
import com.yutou.qqbot.utlis.Log;
import com.yutou.qqbot.utlis.RedisTools;
import net.mamoe.mirai.event.events.MessageEvent;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class Bangumi extends Model {
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if(!isGroupPower){
System.err.println(qq+" 没有权限");
return;
}
switch (msg){
case QQGroupCommands.QQ_BANGUMI_TODAY -> {
RedisTools.remove("reportToDayBangumi");
QQBotManager.getInstance().sendMessage(qq, BangumiTools.reportToDayBangumi());
}
case QQGroupCommands.QQ_BANGUMI_LIST -> {
QQBotManager.getInstance().sendMessage(qq,"获取中...");
QQBotManager.getInstance().sendMessage(qq,BangumiTools.reportBangumiList());
}default -> {
if(msg.startsWith(QQGroupCommands.QQ_BANGUMI_SUB)){
subBanGumi(qq,msg);
}
}
}
}
private void subBanGumi(long qq,String msg){
List<String> infos = null;
try {
int id = Integer.parseInt(msg.replace(QQGroupCommands.QQ_BANGUMI_SUB, "").trim());
infos = BangumiTools.reportBangumiInfo(id);
} catch (Exception e) {
String key = msg.replace(QQGroupCommands.QQ_BANGUMI_SUB, "").trim();
infos = BangumiTools.reportSearchBangumi(key);
}
for (String info : infos) {
List<String> imgs = new ArrayList<>();
if (info.contains("<img") && info.contains(" /img>")) {
imgs = getImages(info);
for (String img : imgs) {
info = info.replace("<img " + img + " /img>", "");
}
}
sendImagesMsg(imgs,qq, info);
}
}
public static List<String> getImages(String str) {
List<String> list = new ArrayList<>();
String regex = "<img(.*?)/img>";
Pattern pattern = Pattern.compile(regex);
Matcher matcher = pattern.matcher(str);
while (matcher.find()) {
list.add(matcher.group().replace("<img", "")
.replace("/img>", "")
.trim());
}
return list;
}
private List<File> files;
private int index = 0;
private void sendImagesMsg(List<String> imgs,Long qq, String text) {
files = new ArrayList<>();
index = 0;
if (imgs.size() == 0) {
QQBotManager.getInstance().sendMessage(qq,text);
return;
}
for (String img : imgs) {
HttpTools.download(img,null, new DownloadInterface() {
@Override
public void onDownload(File file) {
super.onDownload(file);
files.add(file);
send(imgs.size(),qq, text);
}
@Override
public void onError(Exception e) {
super.onError(e);
index++;
send(imgs.size(),qq, text);
}
});
}
}
private void send(int size,Long qq, String text) {
if ((files.size() + index) == size) {
String str = QQBotManager.getInstance().sendMessage(files,qq, text);
Log.i("str = " + str);
}
}
@Override
public boolean isUserPublic() {
return true;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_BANGUMI_SUB,
QQGroupCommands.QQ_BANGUMI_LIST,
QQGroupCommands.QQ_BANGUMI_TODAY
};
}
}

View File

@ -0,0 +1,36 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.models.audio.QQAudio;
import net.mamoe.mirai.event.events.MessageEvent;
public class Audio extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_AUDIO,
QQGroupCommands.QQ_AUDIO_OPEN_LAMP,
QQGroupCommands.QQ_AUDIO_OPEN_AIR
};
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
switch (msg){
case QQGroupCommands.QQ_AUDIO_OPEN_LAMP -> QQAudio.playText("小爱同学,开灯");
case QQGroupCommands.QQ_AUDIO_OPEN_AIR -> QQAudio.playText("小爱同学,开空调");
default -> {
if(msg.startsWith(QQGroupCommands.QQ_AUDIO)){
QQAudio.playText(msg.replace(QQGroupCommands.QQ_AUDIO, ""));
}
}
}
}
}

View File

@ -0,0 +1,29 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.HttpTools;
import net.mamoe.mirai.event.events.MessageEvent;
public class BtFlash extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_BT_RELOAD
};
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if(msg.equals(QQGroupCommands.QQ_BT_RELOAD)){
HttpTools.get("http://192.168.31.88:8000/bt/down.do");
QQBotManager.getInstance().sendMessage(qq,"已提交");
}
}
}

View File

@ -0,0 +1,28 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.RedisTools;
import net.mamoe.mirai.event.events.MessageEvent;
public class Cmd extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_CMD
};
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if (msg.startsWith(QQGroupCommands.QQ_CMD)) {
RedisTools.Consumer.system("cmd", msg.replace(QQGroupCommands.QQ_CMD, ""));
}
}
}

View File

@ -0,0 +1,26 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.models.Model;
import net.mamoe.mirai.event.events.MessageEvent;
public class Help extends Model {
@Override
public boolean isUserPublic() {
return true;
}
@Override
public String[] getUsePowers() {
return new String[0];
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if(msg.equals(QQGroupCommands.QQ_HELP)){
QQBotManager.getInstance().sendMessage(qq,getCommands(QQGroupCommands.class).toString());
}
}
}

View File

@ -0,0 +1,27 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.RedisTools;
import net.mamoe.mirai.event.events.MessageEvent;
public class IP extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_GET_IP
};
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if(msg.equals(QQGroupCommands.QQ_GET_IP)){
RedisTools.Consumer.bot("getip");
}
}
}

View File

@ -0,0 +1,30 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.RedisTools;
import net.mamoe.mirai.event.events.MessageEvent;
import java.text.SimpleDateFormat;
import java.util.Date;
public class OpenPC extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_OPEN_PC
};
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if(msg.equals(QQGroupCommands.QQ_OPEN_PC)){
RedisTools.Consumer.system("openPC", null);
}
}
}

View File

@ -0,0 +1,30 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.AppTools;
import net.mamoe.mirai.event.events.MessageEvent;
public class Restart extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_SYSTEM_RESTART
};
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if(msg.equals(QQGroupCommands.QQ_SYSTEM_RESTART)){
QQBotManager.getInstance().sendMessage(qq,"正在重启服务");
System.out.println("结束进程");
AppTools.exec("cd /home/yutou/public/servier/tools && ./start.sh",null,true,false);
}
}
}

View File

@ -0,0 +1,50 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.IdeaTools;
import com.yutou.qqbot.utlis.RedisTools;
import com.yutou.qqbot.utlis.StringUtils;
import net.mamoe.mirai.event.events.MessageEvent;
public class ToolsIdea extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_TOOLS_IDEA,
QQGroupCommands.QQ_TOOLS_IDEA_FILE,
QQGroupCommands.QQ_TOOLS_IDEA
};
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if (QQGroupCommands.QQ_TOOLS_IDEA.equals(msg)) {
StringBuilder builder = new StringBuilder();
if (StringUtils.isEmpty(RedisTools.get("ideaUrl"))) {
builder.append("暂未设置IDEA激活码下载地址");
} else {
for (String name : IdeaTools.getIdeaList(RedisTools.get("ideaUrl"))) {
builder.append(QQGroupCommands.QQ_TOOLS_IDEA_FILE).append(name).append("\n");
}
if (builder.toString().trim().length() == 0) {
builder.append("激活码文件中未包含txt文件");
}
}
QQBotManager.getInstance().sendMessage(qq, builder.toString());
} else {
if (msg.startsWith(QQGroupCommands.QQ_TOOLS_IDEA_URL)) {
RedisTools.set("ideaUrl", msg.replace(QQGroupCommands.QQ_TOOLS_IDEA_URL, "").trim());
QQBotManager.getInstance().sendMessage(qq, "已设定下载地址:" + RedisTools.get("ideaUrl"));
} else if (msg.startsWith(QQGroupCommands.QQ_TOOLS_IDEA_FILE)) {
QQBotManager.getInstance().sendMessage(qq, IdeaTools.getIdea(event.getMessage().contentToString().replace(QQGroupCommands.QQ_TOOLS_IDEA_FILE, "")));
}
}
}
}

View File

@ -0,0 +1,27 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.RedisTools;
import net.mamoe.mirai.event.events.MessageEvent;
public class UpdateIP extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_UPDATE_IP
};
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if(msg.equals(QQGroupCommands.QQ_UPDATE_IP)){
RedisTools.Consumer.system("updateIP", null);
}
}
}

View File

@ -0,0 +1,27 @@
package com.yutou.qqbot.models.Commands.System;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.models.Model;
import net.mamoe.mirai.event.events.MessageEvent;
public class Version extends Model {
@Override
public boolean isUserPublic() {
return false;
}
@Override
public String[] getUsePowers() {
return new String[]{
QQGroupCommands.QQ_GET_VERSION
};
}
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
super.onMessage(qq, event, isGroup);
if(msg.equals(QQGroupCommands.QQ_GET_VERSION)){
QQBotManager.getInstance().sendVersion();
}
}
}

View File

@ -0,0 +1,81 @@
package com.yutou.qqbot.models;
import com.yutou.qqbot.QQNumberManager;
import com.yutou.qqbot.interfaces.ModelInterface;
import net.mamoe.mirai.event.events.GroupMessageEvent;
import net.mamoe.mirai.event.events.MessageEvent;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;
public abstract class Model implements ModelInterface {
public static class QQGroupCommands {
public final static String QQ_HELP = "!help";
public final static String QQ_SYSTEM_RESTART = "!restart";
public final static String QQ_UPDATE_IP = "!更新ip";
public final static String QQ_GET_IP = "!ip";
public final static String QQ_OPEN_PC = "!开机";
public final static String QQ_GET_VERSION = "!version";
public final static String QQ_CMD = "!cmd";
public final static String QQ_BANGUMI_TODAY = "!今日动画";
public final static String QQ_BANGUMI_LIST = "!新番";
public final static String QQ_BANGUMI_SUB = "!查动画";
public final static String QQ_AUDIO = "!语音";
public final static String QQ_AUDIO_OPEN_LAMP = "!开灯";
public final static String QQ_AUDIO_OPEN_AIR = "!开空调";
public final static String QQ_BT_RELOAD = "!刷bt";
public final static String QQ_TOOLS_IDEA = "!idea";
public final static String QQ_TOOLS_IDEA_FILE = "!idea>";
public final static String QQ_TOOLS_IDEA_URL = "!idea_url";
public final static String QQ_BANGUMI_INFO = "!保存动画信息>";
}
public static class QQFromCommands {
public static final String ADD_POWER = "大头菜";
}
public static List<Class<?>> classList = new ArrayList<>();
long group;
public Model() {
if (!classList.contains(getClass())) {
classList.add(getClass());
}
}
public String msg;
protected boolean isGroupPower = false;
public void onMessage(Long qq, MessageEvent event, boolean isGroup) {
msg = event.getMessage().contentToString();
msg = msg.replace("", "!");
GroupMessageEvent groupEvent = (GroupMessageEvent) event;
group = groupEvent.getGroup().getId();
if (QQNumberManager.getManager().isExistsPower(group, msg.split(" ")[0])) {
isGroupPower = true;
}
}
public void onTime(String time) {
}
public static StringBuilder getCommands(Class<?> commands) {
StringBuilder builder = new StringBuilder();
for (Field field : commands.getDeclaredFields()) {
try {
field.setAccessible(true);
builder.append(field.get(null)).append("\n");
} catch (IllegalAccessException e) {
e.printStackTrace();
}
}
return builder;
}
}

View File

@ -3,11 +3,14 @@ package com.yutou.qqbot.models.setu;
import com.alibaba.fastjson.JSONObject;
import com.yutou.qqbot.Annotations.UseModel;
import com.yutou.qqbot.QQBotManager;
import com.yutou.qqbot.models.Model;
import com.yutou.qqbot.utlis.AppTools;
import com.yutou.qqbot.utlis.Log;
import com.yutou.qqbot.utlis.RedisTools;
import net.mamoe.mirai.event.events.GroupMessageEvent;
import net.mamoe.mirai.event.events.MessageEvent;
import net.mamoe.mirai.message.data.At;
import net.mamoe.mirai.message.data.Image;
import net.mamoe.mirai.message.data.MessageChainBuilder;
@ -15,16 +18,22 @@ import net.mamoe.mirai.message.data.MessageChainBuilder;
import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;
import java.util.Timer;
import java.util.TimerTask;
import java.text.SimpleDateFormat;
import java.util.*;
public class QQSetu {
public static void printTodaySetu() {
@UseModel
public class QQSetu extends Model {
private int db_print =1;//统计结果
private int db_user =3;//当次数据
private long group;
private Timer timer;
private final Map<String, Float> setuScore = new HashMap<>();
public void printTodaySetu() {
String redisKey= AppTools.getToDayTime() + "_setu";
Log.i("今日涩图 redisKey = " + redisKey);
String js = RedisTools.get(redisKey, 1);
String js = RedisTools.get(redisKey, db_print);
if (js != null) {
JSONObject json = JSONObject.parseObject(js);
if(json.containsKey("isPrint")&&json.getBoolean("isPrint")){
@ -49,7 +58,7 @@ public class QQSetu {
setu=json.getJSONObject(groupImage.get(id));
if(setu!=null){
json.put("isPrint",true);
RedisTools.set(1,redisKey,json.toJSONString());
RedisTools.set(db_print,redisKey,json.toJSONString());
JSONObject info=setu.getJSONObject("info");
JSONObject score=setu.getJSONObject("score");
MessageChainBuilder builder = new MessageChainBuilder();
@ -87,7 +96,7 @@ public class QQSetu {
if (!isSetu(image)) {
return;
}
if (RedisTools.get(event.getGroup().getId()+"setu") != null) {
if (RedisTools.get(event.getGroup().getId()+"setu", db_user) != null) {
printSetu(event.getGroup().getId());
}
setuScore.clear();
@ -96,15 +105,15 @@ public class QQSetu {
json.put("sourName", event.getSenderName());
json.put("sourQQ", event.getSender().getId());
json.put("group", event.getGroup().getId());
RedisTools.set(event.getGroup().getId()+"setu", json.toJSONString(),6*60);
RedisTools.set(event.getGroup().getId()+"setu", json.toJSONString(),6*60, db_user);
if (timer != null) {
timer.cancel();
timer = null;
}
startTimer(event.getGroup().getId());
startTimer();
}
private void startTimer(long group) {
private void startTimer() {
timer = new Timer();
timer.schedule(new TimerTask() {
@Override
@ -114,11 +123,11 @@ public class QQSetu {
}
timer = null;
}
}, 5 * 60 * 1000);
}, (long) (0.9 * 60 * 1000));
}
private void printSetu(long group) {
JSONObject jt = JSONObject.parseObject(RedisTools.get(group+"setu"));
JSONObject jt = JSONObject.parseObject(RedisTools.get(group+"setu", db_user));
String id = jt.getString("id");
float average = 0;
float max = 0;
@ -154,7 +163,7 @@ public class QQSetu {
"其中最高分由:" + maxName + " 给与:" + max +"\n "+
"其中最低分由:" + minName + " 给与:" + min;
QQBotManager.getInstance().sendMessage(group, builder);
String st = RedisTools.get(AppTools.getToDayTime() + "_setu", 1);
String st = RedisTools.get(AppTools.getToDayTime() + "_setu", db_print);
JSONObject json;
if (st == null) {
json = new JSONObject();
@ -173,16 +182,23 @@ public class QQSetu {
item.put("info", jt);
item.put("average", average);
json.put(id, item);
RedisTools.set(1, AppTools.getToDayTime() + "_setu", json.toJSONString());
RedisTools.set(db_print, AppTools.getToDayTime() + "_setu", json.toJSONString());
}
RedisTools.remove(group+"setu",0);
RedisTools.remove(group+"setu",db_user);
setuScore.clear();
}
public void setu(String msg, GroupMessageEvent event) {
if (msg.trim().equals("[图片]")) {
@Override
public void onMessage(Long qq, MessageEvent event, boolean isGroup){
System.out.println("qq = " + qq + ", event = " + event.getMessage().contentToString() + ", isGroup = " + isGroup);
if(!isGroup){
return;
}
group=qq;
String msg=event.getMessage().contentToString();
if ("[图片]".equals(msg.trim())) {
Image image = (Image) event.getMessage().stream().filter(Image.class::isInstance).findFirst().orElse(null);
if (image != null) {
setuBuilder(image, event);
setuBuilder(image, (GroupMessageEvent) event);
return;
}
}
@ -193,15 +209,30 @@ public class QQSetu {
float i = Float.parseFloat(msg.trim());
if (i > 0 && i <= 10) {
String name = event.getSenderName();
String qq=event.getSender().getId()+"";
String sender=event.getSender().getId()+"";
if (!setuScore.containsKey(name)) {
setuScore.put(name+"|"+qq, i);
setuScore.put(name+"|"+sender, i);
}
}
} catch (Exception ignored) {
}
}
private Timer timer;
private static Map<String, Float> setuScore = new HashMap<>();
@Override
public void onTime(String time) {
super.onTime(time);
if("23:59".equals(time)){
printTodaySetu();
}
}
@Override
public boolean isUserPublic() {
return true;
}
@Override
public String[] getUsePowers() {
return new String[0];
}
}

View File

@ -13,10 +13,15 @@ import org.springframework.util.ObjectUtils;
import java.io.*;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.net.JarURLConnection;
import java.net.URL;
import java.net.URLDecoder;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
public class AppTools {

View File

@ -0,0 +1,40 @@
package com.yutou.qqbot.utlis;
import com.yutou.qqbot.models.Model;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.stereotype.Component;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Timer;
import java.util.TimerTask;
@Component
public class ApplicationInit implements ApplicationRunner {
@Override
public void run(ApplicationArguments args) throws Exception {
new Timer().schedule(new TimerTask() {
private String oldTime = "";
@Override
public void run() {
try {
String time = new SimpleDateFormat("HH:mm").format(new Date());
if (time.equals(oldTime)) {
return;
}
for (Class<?> model : Model.classList) {
try {
Model useModel= (Model) model.getDeclaredConstructor().newInstance();
useModel.onTime(time);
} catch (Exception e) {
e.printStackTrace();
}
}
}catch (Exception e){
e.printStackTrace();
}
}
},0, 35 * 1000);
}
}

View File

@ -44,7 +44,7 @@ public class RedisTools {
Jedis jedis = getRedis();
jedis.select(dbIndex);
String ret = jedis.set(key, value);
Log.i("Redis set =" + ret);
Log.i("Redis set =" + ret,"key = "+key+" value = "+value);
jedis.close();
} catch (Exception e) {
// TODO: handle exception
@ -54,16 +54,15 @@ public class RedisTools {
return true;
}
public static boolean set(String key, String value) {
return set(0, key, value);
}
public static boolean set(String key, String value, int timeout) {
public static boolean set (String key,String value, long timeout,int db){
try {
if (isNotInstallRedis) {
return false;
}
Jedis jedis = getRedis();
if(db>-1){
jedis.select(db);
}
if (timeout == -1) {
jedis.set(key, value);
} else {
@ -78,6 +77,14 @@ public class RedisTools {
return true;
}
public static boolean set(Object key, String value) {
return set(3, key+"", value);
}
public static boolean set(String key, String value, long timeout) {
return set(key, value, timeout,3);
}
public static String get(String key, int dbIndex) {
String value = "-999";
if (isNotInstallRedis) {
@ -94,12 +101,12 @@ public class RedisTools {
return value;
}
public static String get(String key) {
return get(key, 0);
public static String get(Object key) {
return get(key+"", 3);
}
public static boolean remove(String key) {
return remove(key,0);
return remove(key,3);
}
public static void removeLoginState(String uid) {
@ -119,13 +126,21 @@ public class RedisTools {
jedis.close();
return tmp;
}
public static boolean exists(String key, String value) {
public static boolean exists(Object key,String value){
return exists(3,key,value);
}
public static boolean exists(int db,Object key, String value) {
if (isNotInstallRedis) {
return false;
}
Jedis jedis = getRedis();
boolean flag = value.equals(jedis.get(key));
jedis.select(db);
boolean flag=false;
if(value==null){
flag= jedis.exists(key+"");
}else {
flag = value.equals(jedis.get(key+""));
}
jedis.close();
return flag;
}