lywsvip/ChopperBot
1
0
Fork 0
mirror of https://github.com/Geniusay/ChopperBot.git synced 2026-05-23 04:39:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

FileCache更改

Browse Source
This commit is contained in:
userA
2023-04-25 19:38:47 +08:00
parent 8b0cef608d
commit e6214ee709
2 changed files with 89 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

84
FileModule/src/main/java/org/example/cache/FileCache.java vendored
View File

@@ -1,6 +1,7 @@
package org.example.cache;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import org.example.common.ConfigFile;
import org.example.exception.FileCacheException;
@@ -11,6 +12,8 @@ import org.slf4j.LoggerFactory;
import java.io.File;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Map;
import java.util.Objects;
import java.util.concurrent.*;
@@ -95,12 +98,18 @@ public class FileCache <T extends ConfigFile>{
return load(Paths.get(this.configFile.getFilePath(),this.configFile.getFileName()).toString());
}
//写入
public int write(String key,Object data) throws InterruptedException {
/**
* 递进寻找JsonObject中的对象，并改写
* @param data
* @param keys
* @return
* @throws InterruptedException
*/
public int writeKeys(Object data,String...keys) throws InterruptedException, FileCacheException {
String jsonDataStr = JSON.toJSONString(data);
int writeBytes = key.getBytes().length + jsonDataStr.getBytes().length;
int writeBytes = jsonDataStr.getBytes().length;
JSONObject jsonData = writeInData(key, data);
JSONObject jsonData = writeInData(data,keys);
jsonFile.put("data",jsonData);
ConcurrentHashMap<String,Object> temp = new ConcurrentHashMap<>(jsonFile);
@@ -114,24 +123,63 @@ public class FileCache <T extends ConfigFile>{
return writeBytes;
}
//写入
public int write(Object data,String key) throws InterruptedException, FileCacheException {
return this.writeKeys(data,key);
}
private JSONObject writeInData(Object value,String...keys) throws FileCacheException {
String[] finds = Arrays.copyOf(keys, keys.length - 1);
JSONObject data = JSONObject.parseObject(this.get("data").toString());
Object data1 = this.jsonFile.get("data");
Object temp = this.get(finds);
String tempStr = temp.toString();
if(tempStr.startsWith("[")&&tempStr.endsWith("]")){
(JSONArray.parseArray(tempStr)).add(Integer.parseInt(keys[keys.length-1]),value);
}
else if(temp instanceof JSONObject){
((JSONObject) temp).put(keys[keys.length-1],value);
}
else{
throw new FileCacheException("the keys is error!");
}
return data;
}
/**
* 追加内容
* @param key key
* @param keys 要查找的key
* @param append 追加内容
* @return
* @throws InterruptedException
*/
public int append(String key,Object append) throws InterruptedException {
Object data = getData(key);
public int append(Object append,String...keys) throws InterruptedException, FileCacheException {
Object data = get(keys);
StringBuffer buffer = new StringBuffer(JSON.toJSONString(data));
String jsonStr = buffer.append(JSON.toJSONString(append)).toString();
return write(key, jsonStr);
return writeKeys(jsonStr,keys);
}
private JSONObject writeInData(String key,Object value){
JSONObject data = JSONObject.parseObject(jsonFile.get("data").toString()); //此处必须返回一个新的JsonObject，否则会导致旧版本同步更新
data.put(key,value);
return data;
/**
* 根据key数组，不断向下获取内容
* @param keys
* @return
*/
public Object get(String...keys){
Object jsonObject = JSONObject.parse(this.get("data").toString());
for (String key : keys) {
String jsonStr = jsonObject.toString();
if(jsonStr.startsWith("{")&&jsonStr.endsWith("}")){
jsonObject = ((JSONObject) jsonObject).get(key);
}
else if(jsonStr.startsWith("[")&&jsonStr.endsWith("]")){
jsonObject = (JSONArray.parseArray(jsonObject.toString())).get(Integer.parseInt(key));
}else{
return jsonObject;
}
}
return jsonObject;
}
/**
@@ -143,16 +191,6 @@ public class FileCache <T extends ConfigFile>{
return jsonFile.get(key);
}
/**
* 获取文件数据内容
* @param key
* @return
*/
public Object getData(String key){
Object data = this.get("data");
JSONObject jsonObject = JSONObject.parseObject(data.toString());
return jsonObject.get(key);
}
/**
* 清除已写入的字节数记录
@@ -184,7 +222,7 @@ public class FileCache <T extends ConfigFile>{
}
/**
* 超过缓冲区刷入
* 缓冲区刷入
* @return
*/
private boolean sync(ConcurrentHashMap<String,Object> take){

30
FileModule/src/test/java/org/example/cache/FileCacheTest.java vendored
View File

@@ -1,5 +1,6 @@
package org.example.cache;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.JSONPObject;
import org.example.bean.ModuleSrcConfigFile;
@@ -12,6 +13,8 @@ import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import static org.apache.logging.log4j.message.MapMessage.MapFormat.JSON;
/**
* @author Genius
* @date 2023/04/24 03:30
@@ -33,13 +36,36 @@ public class FileCacheTest {
public void TestFileCacheManager() throws FileCacheException, InterruptedException {
ModuleSrcConfigFile moduleSrcConfigFile = new ModuleSrcConfigFile();
FileCache fileCache = new FileCache(moduleSrcConfigFile);
FileCacheManager manager = new FileCacheManager(List.of(fileCache));
manager.start();
for(int i=0;i<1000;i++){
for(int i=0;i<1;i++){
fileCache.append("sectionwork",Integer.toString(i));
}
Thread.sleep(500000);
}
@Test
public void TestGet() throws FileCacheException, InterruptedException {
ModuleSrcConfigFile moduleSrcConfigFile = new ModuleSrcConfigFile();
FileCache fileCache = new FileCache(moduleSrcConfigFile);
FileCacheManager manager = new FileCacheManager(List.of(fileCache));
manager.start();
System.out.println(fileCache.get("sectionwork", "src"));
//fileCache.append(1,"sectionwork","src","0");
}
@Test
public void TestJson(){
String a = "{username:'[1,2,3,4]',age:{'hello':'world'}}";
Object jsonObject = JSONObject.parse(a);
System.out.println(jsonObject);
System.out.println(jsonObject.getClass());
jsonObject = JSONArray.parse( ((JSONObject) jsonObject).get("username").toString() );
System.out.println(jsonObject);
System.out.println(jsonObject.getClass());
}
}