working on json file store
This commit is contained in:
parent
99ce43b6b8
commit
b930f00b8e
2 changed files with 155 additions and 0 deletions
21
net/src/main/java/org/xbib/net/Store.java
Normal file
21
net/src/main/java/org/xbib/net/Store.java
Normal file
|
@ -0,0 +1,21 @@
|
|||
package org.xbib.net;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
public interface Store<S> {
|
||||
|
||||
String getName();
|
||||
|
||||
long size() throws IOException;
|
||||
|
||||
S read(String key) throws IOException;
|
||||
|
||||
void readAll(String key, Consumer<S> consumer) throws IOException;
|
||||
|
||||
void write(String key, S s) throws IOException;
|
||||
|
||||
void remove(String key) throws IOException;
|
||||
|
||||
void purge(long expiredAfterSeconds) throws IOException;
|
||||
}
|
134
net/src/main/java/org/xbib/net/util/JsonFileStore.java
Normal file
134
net/src/main/java/org/xbib/net/util/JsonFileStore.java
Normal file
|
@ -0,0 +1,134 @@
|
|||
package org.xbib.net.util;
|
||||
|
||||
import org.xbib.net.Store;
|
||||
import org.xbib.net.PercentEncoder;
|
||||
import org.xbib.net.PercentEncoders;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.Writer;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.nio.file.attribute.FileTime;
|
||||
import java.time.Duration;
|
||||
import java.time.Instant;
|
||||
import java.util.Map;
|
||||
import java.util.Objects;
|
||||
import java.util.concurrent.atomic.AtomicLong;
|
||||
import java.util.concurrent.locks.ReentrantReadWriteLock;
|
||||
import java.util.function.Consumer;
|
||||
import java.util.logging.Level;
|
||||
import java.util.logging.Logger;
|
||||
import java.util.stream.Stream;
|
||||
|
||||
public class JsonFileStore implements Store<Map<String, Object>> {
|
||||
|
||||
private static final Logger logger = Logger.getLogger(JsonFileStore.class.getName());
|
||||
|
||||
private final String name;
|
||||
|
||||
private final ReentrantReadWriteLock lock;
|
||||
|
||||
private final Path path;
|
||||
|
||||
public JsonFileStore(String name, Path path) throws IOException {
|
||||
this.name = name;
|
||||
this.path = path;
|
||||
this.lock = new ReentrantReadWriteLock();
|
||||
Files.createDirectories(path);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return name;
|
||||
}
|
||||
|
||||
@Override
|
||||
public long size() throws IOException {
|
||||
AtomicLong size = new AtomicLong(0L);
|
||||
try (Stream<Path> stream = Files.walk(path)) {
|
||||
stream.forEach(p -> {
|
||||
if (Files.isRegularFile(p)) {
|
||||
size.incrementAndGet();
|
||||
}
|
||||
});
|
||||
return size.get();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void readAll(String key, Consumer<Map<String, Object>> consumer) throws IOException {
|
||||
Objects.requireNonNull(consumer);
|
||||
try (Stream<Path> stream = Files.walk(path.resolve(key))) {
|
||||
stream.forEach(p -> {
|
||||
try {
|
||||
consumer.accept(read(p.getFileName().toString()));
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "i/o error while consuming: " + e.getMessage(), e);
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public Map<String, Object> read(String key) throws IOException {
|
||||
ReentrantReadWriteLock.ReadLock readLock = lock.readLock();
|
||||
try {
|
||||
readLock.lock();
|
||||
PercentEncoder percentEncoder = PercentEncoders.getUnreservedEncoder(StandardCharsets.UTF_8);
|
||||
return JsonUtil.toMap(Files.readString(path.resolve(percentEncoder.encode(key))));
|
||||
} finally {
|
||||
readLock.unlock();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void write(String key, Map<String, Object> map) throws IOException {
|
||||
ReentrantReadWriteLock.WriteLock writeLock = lock.writeLock();
|
||||
try {
|
||||
writeLock.lock();
|
||||
PercentEncoder percentEncoder = PercentEncoders.getUnreservedEncoder(StandardCharsets.UTF_8);
|
||||
Path p = path.resolve(percentEncoder.encode(key));
|
||||
if (!Files.exists(p.getParent())) {
|
||||
Files.createDirectories(p.getParent());
|
||||
}
|
||||
try (Writer writer = Files.newBufferedWriter(p)) {
|
||||
writer.write(JsonUtil.toString(map));
|
||||
}
|
||||
} finally {
|
||||
writeLock.unlock();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void remove(String key) throws IOException {
|
||||
ReentrantReadWriteLock.WriteLock writeLock = lock.writeLock();
|
||||
try {
|
||||
writeLock.lock();
|
||||
PercentEncoder percentEncoder = PercentEncoders.getUnreservedEncoder(StandardCharsets.UTF_8);
|
||||
Files.deleteIfExists(path.resolve(percentEncoder.encode(key)));
|
||||
} finally {
|
||||
writeLock.unlock();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void purge(long expiredAfterSeconds) throws IOException {
|
||||
if (path != null && expiredAfterSeconds > 0L) {
|
||||
Instant instant = Instant.now();
|
||||
try (Stream<Path> stream = Files.walk(path)) {
|
||||
stream.forEach(p -> {
|
||||
try {
|
||||
FileTime fileTime = Files.getLastModifiedTime(p);
|
||||
Duration duration = Duration.between(fileTime.toInstant(), instant);
|
||||
if (duration.toSeconds() > expiredAfterSeconds) {
|
||||
Files.delete(p);
|
||||
}
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "i/o error while purge: " + e.getMessage(), e);
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue