ustuzhanin
4 years ago
committed by
GitHub
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
332 changed files with 18966 additions and 4306 deletions
-
5.travis.yml
-
26README.md
-
35docker/Dockerfile.go_build_large
-
12go.mod
-
30go.sum
-
2k8s/seaweedfs/Chart.yaml
-
22k8s/seaweedfs/templates/filer-statefulset.yaml
-
30k8s/seaweedfs/templates/master-statefulset.yaml
-
2k8s/seaweedfs/templates/s3-deployment.yaml
-
4k8s/seaweedfs/templates/seaweefs-grafana-dashboard.yaml
-
25k8s/seaweedfs/templates/volume-statefulset.yaml
-
58k8s/seaweedfs/values.yaml
-
2other/java/client/pom.xml
-
2other/java/client/pom.xml.deploy
-
2other/java/client/pom_debug.xml
-
55other/java/client/src/main/java/seaweedfs/client/SeaweedRead.java
-
32other/java/client/src/main/proto/filer.proto
-
176other/java/hdfs2/dependency-reduced-pom.xml
-
4other/java/hdfs2/pom.xml
-
29other/java/hdfs2/src/main/java/seaweed/hdfs/SeaweedFileSystem.java
-
11other/java/hdfs2/src/main/java/seaweed/hdfs/SeaweedFileSystemStore.java
-
4other/java/hdfs2/src/main/java/seaweed/hdfs/SeaweedInputStream.java
-
184other/java/hdfs3/dependency-reduced-pom.xml
-
4other/java/hdfs3/pom.xml
-
29other/java/hdfs3/src/main/java/seaweed/hdfs/SeaweedFileSystem.java
-
11other/java/hdfs3/src/main/java/seaweed/hdfs/SeaweedFileSystemStore.java
-
4other/java/hdfs3/src/main/java/seaweed/hdfs/SeaweedInputStream.java
-
1856other/metrics/grafana_seaweedfs.json
-
58test/random_access/pom.xml
-
753test/random_access/src/main/java/seaweedfs/client/btree/BTreePersistentIndexedCache.java
-
59test/random_access/src/main/java/seaweedfs/client/btree/Block.java
-
51test/random_access/src/main/java/seaweedfs/client/btree/BlockPayload.java
-
75test/random_access/src/main/java/seaweedfs/client/btree/BlockPointer.java
-
68test/random_access/src/main/java/seaweedfs/client/btree/BlockStore.java
-
30test/random_access/src/main/java/seaweedfs/client/btree/BufferCaster.java
-
74test/random_access/src/main/java/seaweedfs/client/btree/ByteInput.java
-
74test/random_access/src/main/java/seaweedfs/client/btree/ByteOutput.java
-
129test/random_access/src/main/java/seaweedfs/client/btree/CachingBlockStore.java
-
22test/random_access/src/main/java/seaweedfs/client/btree/CorruptedCacheException.java
-
274test/random_access/src/main/java/seaweedfs/client/btree/FileBackedBlockStore.java
-
283test/random_access/src/main/java/seaweedfs/client/btree/FreeListBlockStore.java
-
75test/random_access/src/main/java/seaweedfs/client/btree/KeyHasher.java
-
54test/random_access/src/main/java/seaweedfs/client/btree/RandomAccessFileInputStream.java
-
48test/random_access/src/main/java/seaweedfs/client/btree/RandomAccessFileOutputStream.java
-
87test/random_access/src/main/java/seaweedfs/client/btree/StateCheckBlockStore.java
-
526test/random_access/src/main/java/seaweedfs/client/btree/StreamByteBuffer.java
-
88test/random_access/src/main/java/seaweedfs/client/btree/UncheckedException.java
-
36test/random_access/src/main/java/seaweedfs/client/btree/UncheckedIOException.java
-
133test/random_access/src/main/java/seaweedfs/client/btree/serialize/AbstractDecoder.java
-
101test/random_access/src/main/java/seaweedfs/client/btree/serialize/AbstractEncoder.java
-
40test/random_access/src/main/java/seaweedfs/client/btree/serialize/AbstractSerializer.java
-
79test/random_access/src/main/java/seaweedfs/client/btree/serialize/Cast.java
-
43test/random_access/src/main/java/seaweedfs/client/btree/serialize/ClassLoaderObjectInputStream.java
-
140test/random_access/src/main/java/seaweedfs/client/btree/serialize/Decoder.java
-
73test/random_access/src/main/java/seaweedfs/client/btree/serialize/DefaultSerializer.java
-
110test/random_access/src/main/java/seaweedfs/client/btree/serialize/Encoder.java
-
31test/random_access/src/main/java/seaweedfs/client/btree/serialize/FlushableEncoder.java
-
28test/random_access/src/main/java/seaweedfs/client/btree/serialize/ObjectReader.java
-
21test/random_access/src/main/java/seaweedfs/client/btree/serialize/ObjectWriter.java
-
33test/random_access/src/main/java/seaweedfs/client/btree/serialize/Serializer.java
-
33test/random_access/src/main/java/seaweedfs/client/btree/serialize/StatefulSerializer.java
-
210test/random_access/src/main/java/seaweedfs/client/btree/serialize/kryo/KryoBackedDecoder.java
-
134test/random_access/src/main/java/seaweedfs/client/btree/serialize/kryo/KryoBackedEncoder.java
-
188test/random_access/src/main/java/seaweedfs/client/btree/serialize/kryo/StringDeduplicatingKryoBackedDecoder.java
-
128test/random_access/src/main/java/seaweedfs/client/btree/serialize/kryo/StringDeduplicatingKryoBackedEncoder.java
-
51test/random_access/src/main/java/seaweedfs/client/btree/serialize/kryo/TypeSafeSerializer.java
-
476test/random_access/src/test/java/seaweedfs/client/btree/BTreePersistentIndexedCacheTest.java
-
143test/random_access/src/test/java/seaweedfs/file/MmapFileTest.java
-
70test/random_access/src/test/java/seaweedfs/file/RandomeAccessFileTest.java
-
21test/s3/basic/basic_test.go
-
175test/s3/multipart/aws_upload.go
-
10unmaintained/diff_volume_servers/diff_volume_servers.go
-
4unmaintained/fix_dat/fix_dat.go
-
73unmaintained/s3/presigned_put/presigned_put.go
-
83unmaintained/see_dat/see_dat_gzip.go
-
2unmaintained/see_idx/see_idx.go
-
4unmaintained/see_log_entry/see_log_entry.go
-
27weed/Makefile
-
1weed/command/command.go
-
14weed/command/download.go
-
24weed/command/export.go
-
31weed/command/filer.go
-
2weed/command/filer_copy.go
-
337weed/command/filer_sync.go
-
2weed/command/fix.go
-
2weed/command/master.go
-
4weed/command/mount.go
-
34weed/command/mount_std.go
-
24weed/command/s3.go
-
18weed/command/scaffold.go
-
15weed/command/server.go
-
73weed/command/volume.go
-
60weed/command/watch.go
-
54weed/filer/abstract_sql/abstract_sql_store.go
-
87weed/filer/abstract_sql/abstract_sql_store_kv.go
-
0weed/filer/cassandra/README.txt
-
36weed/filer/cassandra/cassandra_store.go
-
61weed/filer/cassandra/cassandra_store_kv.go
-
2weed/filer/configuration.go
-
338weed/filer/elastic/v7/elastic_store.go
@ -0,0 +1,35 @@ |
|||
FROM frolvlad/alpine-glibc as builder |
|||
RUN apk add git go g++ |
|||
RUN mkdir -p /go/src/github.com/chrislusf/ |
|||
RUN git clone https://github.com/chrislusf/seaweedfs /go/src/github.com/chrislusf/seaweedfs |
|||
RUN cd /go/src/github.com/chrislusf/seaweedfs/weed && go install -tags 5BytesOffset |
|||
|
|||
FROM alpine AS final |
|||
LABEL author="Chris Lu" |
|||
COPY --from=builder /root/go/bin/weed /usr/bin/ |
|||
RUN mkdir -p /etc/seaweedfs |
|||
COPY --from=builder /go/src/github.com/chrislusf/seaweedfs/docker/filer.toml /etc/seaweedfs/filer.toml |
|||
COPY --from=builder /go/src/github.com/chrislusf/seaweedfs/docker/entrypoint.sh /entrypoint.sh |
|||
|
|||
# volume server gprc port |
|||
EXPOSE 18080 |
|||
# volume server http port |
|||
EXPOSE 8080 |
|||
# filer server gprc port |
|||
EXPOSE 18888 |
|||
# filer server http port |
|||
EXPOSE 8888 |
|||
# master server shared gprc port |
|||
EXPOSE 19333 |
|||
# master server shared http port |
|||
EXPOSE 9333 |
|||
# s3 server http port |
|||
EXPOSE 8333 |
|||
|
|||
RUN mkdir -p /data/filerldb2 |
|||
|
|||
VOLUME /data |
|||
|
|||
RUN chmod +x /entrypoint.sh |
|||
|
|||
ENTRYPOINT ["/entrypoint.sh"] |
|||
@ -1,4 +1,4 @@ |
|||
apiVersion: v1 |
|||
description: SeaweedFS |
|||
name: seaweedfs |
|||
version: 1.87 |
|||
version: 2.02 |
|||
1856
other/metrics/grafana_seaweedfs.json
File diff suppressed because it is too large
View File
File diff suppressed because it is too large
View File
@ -0,0 +1,58 @@ |
|||
<?xml version="1.0" encoding="UTF-8"?> |
|||
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
|||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> |
|||
<modelVersion>4.0.0</modelVersion> |
|||
<groupId>com.seaweedfs.test</groupId> |
|||
<artifactId>random_access</artifactId> |
|||
<packaging>jar</packaging> |
|||
<version>1.0-SNAPSHOT</version> |
|||
|
|||
<properties> |
|||
<guava.version>28.0-jre</guava.version> |
|||
</properties> |
|||
|
|||
<dependencies> |
|||
<dependency> |
|||
<groupId>com.google.guava</groupId> |
|||
<artifactId>guava</artifactId> |
|||
<version>${guava.version}</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>org.slf4j</groupId> |
|||
<artifactId>slf4j-api</artifactId> |
|||
<version>1.7.25</version> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>junit</groupId> |
|||
<artifactId>junit</artifactId> |
|||
<version>4.12</version> |
|||
<scope>test</scope> |
|||
</dependency> |
|||
<dependency> |
|||
<groupId>com.esotericsoftware.kryo</groupId> |
|||
<artifactId>kryo</artifactId> |
|||
<version>2.24.0</version> |
|||
</dependency> |
|||
</dependencies> |
|||
|
|||
<build> |
|||
<extensions> |
|||
<extension> |
|||
<groupId>kr.motd.maven</groupId> |
|||
<artifactId>os-maven-plugin</artifactId> |
|||
<version>1.6.2</version> |
|||
</extension> |
|||
</extensions> |
|||
<plugins> |
|||
<plugin> |
|||
<groupId>org.apache.maven.plugins</groupId> |
|||
<artifactId>maven-compiler-plugin</artifactId> |
|||
<configuration> |
|||
<source>8</source> |
|||
<target>8</target> |
|||
</configuration> |
|||
</plugin> |
|||
</plugins> |
|||
</build> |
|||
|
|||
</project> |
|||
@ -0,0 +1,753 @@ |
|||
/* |
|||
* Copyright 2010 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import com.google.common.collect.ImmutableSet; |
|||
import seaweedfs.client.btree.serialize.Serializer; |
|||
import seaweedfs.client.btree.serialize.kryo.KryoBackedDecoder; |
|||
import seaweedfs.client.btree.serialize.kryo.KryoBackedEncoder; |
|||
import org.slf4j.Logger; |
|||
import org.slf4j.LoggerFactory; |
|||
|
|||
import java.io.DataInputStream; |
|||
import java.io.DataOutputStream; |
|||
import java.io.File; |
|||
import java.io.IOException; |
|||
import java.util.ArrayList; |
|||
import java.util.Collection; |
|||
import java.util.Collections; |
|||
import java.util.Comparator; |
|||
import java.util.List; |
|||
|
|||
// todo - stream serialised value to file |
|||
// todo - handle hash collisions (properly, this time) |
|||
// todo - don't store null links to child blocks in leaf index blocks |
|||
// todo - align block boundaries |
|||
// todo - thread safety control |
|||
// todo - merge small values into a single data block |
|||
// todo - discard when file corrupt |
|||
// todo - include data directly in index entry when serializer can guarantee small fixed sized data |
|||
// todo - free list leaks disk space |
|||
// todo - merge adjacent free blocks |
|||
// todo - use more efficient lookup for free block with nearest size |
|||
@SuppressWarnings("unchecked") |
|||
public class BTreePersistentIndexedCache<K, V> { |
|||
private static final Logger LOGGER = LoggerFactory.getLogger(BTreePersistentIndexedCache.class); |
|||
private final File cacheFile; |
|||
private final KeyHasher<K> keyHasher; |
|||
private final Serializer<V> serializer; |
|||
private final short maxChildIndexEntries; |
|||
private final int minIndexChildNodes; |
|||
private final StateCheckBlockStore store; |
|||
private HeaderBlock header; |
|||
|
|||
public BTreePersistentIndexedCache(File cacheFile, Serializer<K> keySerializer, Serializer<V> valueSerializer) { |
|||
this(cacheFile, keySerializer, valueSerializer, (short) 512, 512); |
|||
} |
|||
|
|||
public BTreePersistentIndexedCache(File cacheFile, Serializer<K> keySerializer, Serializer<V> valueSerializer, |
|||
short maxChildIndexEntries, int maxFreeListEntries) { |
|||
this.cacheFile = cacheFile; |
|||
this.keyHasher = new KeyHasher<K>(keySerializer); |
|||
this.serializer = valueSerializer; |
|||
this.maxChildIndexEntries = maxChildIndexEntries; |
|||
this.minIndexChildNodes = maxChildIndexEntries / 2; |
|||
BlockStore cachingStore = new CachingBlockStore(new FileBackedBlockStore(cacheFile), ImmutableSet.of(IndexBlock.class, FreeListBlockStore.FreeListBlock.class)); |
|||
this.store = new StateCheckBlockStore(new FreeListBlockStore(cachingStore, maxFreeListEntries)); |
|||
try { |
|||
open(); |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(String.format("Could not open %s.", this), e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public String toString() { |
|||
return "cache " + cacheFile.getName() + " (" + cacheFile + ")"; |
|||
} |
|||
|
|||
private void open() throws Exception { |
|||
LOGGER.debug("Opening {}", this); |
|||
try { |
|||
doOpen(); |
|||
} catch (CorruptedCacheException e) { |
|||
rebuild(); |
|||
} |
|||
} |
|||
|
|||
private void doOpen() throws Exception { |
|||
BlockStore.Factory factory = new BlockStore.Factory() { |
|||
@Override |
|||
public Object create(Class<? extends BlockPayload> type) { |
|||
if (type == HeaderBlock.class) { |
|||
return new HeaderBlock(); |
|||
} |
|||
if (type == IndexBlock.class) { |
|||
return new IndexBlock(); |
|||
} |
|||
if (type == DataBlock.class) { |
|||
return new DataBlock(); |
|||
} |
|||
throw new UnsupportedOperationException(); |
|||
} |
|||
}; |
|||
Runnable initAction = new Runnable() { |
|||
@Override |
|||
public void run() { |
|||
header = new HeaderBlock(); |
|||
store.write(header); |
|||
header.index.newRoot(); |
|||
store.flush(); |
|||
} |
|||
}; |
|||
|
|||
store.open(initAction, factory); |
|||
header = store.readFirst(HeaderBlock.class); |
|||
} |
|||
|
|||
public V get(K key) { |
|||
try { |
|||
try { |
|||
DataBlock block = header.getRoot().get(key); |
|||
if (block != null) { |
|||
return block.getValue(); |
|||
} |
|||
return null; |
|||
} catch (CorruptedCacheException e) { |
|||
rebuild(); |
|||
return null; |
|||
} |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(String.format("Could not read entry '%s' from %s.", key, this), e); |
|||
} |
|||
} |
|||
|
|||
public void put(K key, V value) { |
|||
try { |
|||
long hashCode = keyHasher.getHashCode(key); |
|||
Lookup lookup = header.getRoot().find(hashCode); |
|||
DataBlock newBlock = null; |
|||
if (lookup.entry != null) { |
|||
DataBlock block = store.read(lookup.entry.dataBlock, DataBlock.class); |
|||
DataBlockUpdateResult updateResult = block.useNewValue(value); |
|||
if (updateResult.isFailed()) { |
|||
store.remove(block); |
|||
newBlock = new DataBlock(value, updateResult.getSerializedValue()); |
|||
} |
|||
} else { |
|||
newBlock = new DataBlock(value); |
|||
} |
|||
if (newBlock != null) { |
|||
store.write(newBlock); |
|||
lookup.indexBlock.put(hashCode, newBlock.getPos()); |
|||
} |
|||
store.flush(); |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(String.format("Could not add entry '%s' to %s.", key, this), e); |
|||
} |
|||
} |
|||
|
|||
public void remove(K key) { |
|||
try { |
|||
Lookup lookup = header.getRoot().find(key); |
|||
if (lookup.entry == null) { |
|||
return; |
|||
} |
|||
lookup.indexBlock.remove(lookup.entry); |
|||
DataBlock block = store.read(lookup.entry.dataBlock, DataBlock.class); |
|||
store.remove(block); |
|||
store.flush(); |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(String.format("Could not remove entry '%s' from %s.", key, this), e); |
|||
} |
|||
} |
|||
|
|||
private IndexBlock load(BlockPointer pos, IndexRoot root, IndexBlock parent, int index) { |
|||
IndexBlock block = store.read(pos, IndexBlock.class); |
|||
block.root = root; |
|||
block.parent = parent; |
|||
block.parentEntryIndex = index; |
|||
return block; |
|||
} |
|||
|
|||
public void reset() { |
|||
close(); |
|||
try { |
|||
open(); |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
} |
|||
|
|||
public void close() { |
|||
LOGGER.debug("Closing {}", this); |
|||
try { |
|||
store.close(); |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
} |
|||
|
|||
public boolean isOpen() { |
|||
return store.isOpen(); |
|||
} |
|||
|
|||
private void rebuild() { |
|||
LOGGER.warn("{} is corrupt. Discarding.", this); |
|||
try { |
|||
clear(); |
|||
} catch (Exception e) { |
|||
LOGGER.warn("{} couldn't be rebuilt. Closing.", this); |
|||
close(); |
|||
} |
|||
} |
|||
|
|||
public void verify() { |
|||
try { |
|||
doVerify(); |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(String.format("Some problems were found when checking the integrity of %s.", |
|||
this), e); |
|||
} |
|||
} |
|||
|
|||
private void doVerify() throws Exception { |
|||
List<BlockPayload> blocks = new ArrayList<BlockPayload>(); |
|||
|
|||
HeaderBlock header = store.readFirst(HeaderBlock.class); |
|||
blocks.add(header); |
|||
verifyTree(header.getRoot(), "", blocks, Long.MAX_VALUE, true); |
|||
|
|||
Collections.sort(blocks, new Comparator<BlockPayload>() { |
|||
@Override |
|||
public int compare(BlockPayload block, BlockPayload block1) { |
|||
return block.getPos().compareTo(block1.getPos()); |
|||
} |
|||
}); |
|||
|
|||
for (int i = 0; i < blocks.size() - 1; i++) { |
|||
Block b1 = blocks.get(i).getBlock(); |
|||
Block b2 = blocks.get(i + 1).getBlock(); |
|||
if (b1.getPos().getPos() + b1.getSize() > b2.getPos().getPos()) { |
|||
throw new IOException(String.format("%s overlaps with %s", b1, b2)); |
|||
} |
|||
} |
|||
} |
|||
|
|||
private void verifyTree(IndexBlock current, String prefix, Collection<BlockPayload> blocks, long maxValue, |
|||
boolean loadData) throws Exception { |
|||
blocks.add(current); |
|||
|
|||
if (!prefix.equals("") && current.entries.size() < maxChildIndexEntries / 2) { |
|||
throw new IOException(String.format("Too few entries found in %s", current)); |
|||
} |
|||
if (current.entries.size() > maxChildIndexEntries) { |
|||
throw new IOException(String.format("Too many entries found in %s", current)); |
|||
} |
|||
|
|||
boolean isLeaf = current.entries.size() == 0 || current.entries.get(0).childIndexBlock.isNull(); |
|||
if (isLeaf ^ current.tailPos.isNull()) { |
|||
throw new IOException(String.format("Mismatched leaf/tail-node in %s", current)); |
|||
} |
|||
|
|||
long min = Long.MIN_VALUE; |
|||
for (IndexEntry entry : current.entries) { |
|||
if (isLeaf ^ entry.childIndexBlock.isNull()) { |
|||
throw new IOException(String.format("Mismatched leaf/non-leaf entry in %s", current)); |
|||
} |
|||
if (entry.hashCode >= maxValue || entry.hashCode <= min) { |
|||
throw new IOException(String.format("Out-of-order key in %s", current)); |
|||
} |
|||
min = entry.hashCode; |
|||
if (!entry.childIndexBlock.isNull()) { |
|||
IndexBlock child = store.read(entry.childIndexBlock, IndexBlock.class); |
|||
verifyTree(child, " " + prefix, blocks, entry.hashCode, loadData); |
|||
} |
|||
if (loadData) { |
|||
DataBlock block = store.read(entry.dataBlock, DataBlock.class); |
|||
blocks.add(block); |
|||
} |
|||
} |
|||
if (!current.tailPos.isNull()) { |
|||
IndexBlock tail = store.read(current.tailPos, IndexBlock.class); |
|||
verifyTree(tail, " " + prefix, blocks, maxValue, loadData); |
|||
} |
|||
} |
|||
|
|||
public void clear() { |
|||
store.clear(); |
|||
close(); |
|||
try { |
|||
doOpen(); |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
} |
|||
|
|||
private class IndexRoot { |
|||
private BlockPointer rootPos = BlockPointer.start(); |
|||
private HeaderBlock owner; |
|||
|
|||
private IndexRoot(HeaderBlock owner) { |
|||
this.owner = owner; |
|||
} |
|||
|
|||
public void setRootPos(BlockPointer rootPos) { |
|||
this.rootPos = rootPos; |
|||
store.write(owner); |
|||
} |
|||
|
|||
public IndexBlock getRoot() { |
|||
return load(rootPos, this, null, 0); |
|||
} |
|||
|
|||
public IndexBlock newRoot() { |
|||
IndexBlock block = new IndexBlock(); |
|||
store.write(block); |
|||
setRootPos(block.getPos()); |
|||
return block; |
|||
} |
|||
} |
|||
|
|||
private class HeaderBlock extends BlockPayload { |
|||
private IndexRoot index; |
|||
|
|||
private HeaderBlock() { |
|||
index = new IndexRoot(this); |
|||
} |
|||
|
|||
@Override |
|||
protected byte getType() { |
|||
return 0x55; |
|||
} |
|||
|
|||
@Override |
|||
protected int getSize() { |
|||
return Block.LONG_SIZE + Block.SHORT_SIZE; |
|||
} |
|||
|
|||
@Override |
|||
protected void read(DataInputStream instr) throws Exception { |
|||
index.rootPos = BlockPointer.pos(instr.readLong()); |
|||
|
|||
short actualChildIndexEntries = instr.readShort(); |
|||
if (actualChildIndexEntries != maxChildIndexEntries) { |
|||
throw blockCorruptedException(); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
protected void write(DataOutputStream outstr) throws Exception { |
|||
outstr.writeLong(index.rootPos.getPos()); |
|||
outstr.writeShort(maxChildIndexEntries); |
|||
} |
|||
|
|||
public IndexBlock getRoot() throws Exception { |
|||
return index.getRoot(); |
|||
} |
|||
} |
|||
|
|||
private class IndexBlock extends BlockPayload { |
|||
private final List<IndexEntry> entries = new ArrayList<IndexEntry>(); |
|||
private BlockPointer tailPos = BlockPointer.start(); |
|||
// Transient fields |
|||
private IndexBlock parent; |
|||
private int parentEntryIndex; |
|||
private IndexRoot root; |
|||
|
|||
@Override |
|||
protected byte getType() { |
|||
return 0x77; |
|||
} |
|||
|
|||
@Override |
|||
protected int getSize() { |
|||
return Block.INT_SIZE + Block.LONG_SIZE + (3 * Block.LONG_SIZE) * maxChildIndexEntries; |
|||
} |
|||
|
|||
@Override |
|||
public void read(DataInputStream instr) throws IOException { |
|||
int count = instr.readInt(); |
|||
entries.clear(); |
|||
for (int i = 0; i < count; i++) { |
|||
IndexEntry entry = new IndexEntry(); |
|||
entry.hashCode = instr.readLong(); |
|||
entry.dataBlock = BlockPointer.pos(instr.readLong()); |
|||
entry.childIndexBlock = BlockPointer.pos(instr.readLong()); |
|||
entries.add(entry); |
|||
} |
|||
tailPos = BlockPointer.pos(instr.readLong()); |
|||
} |
|||
|
|||
@Override |
|||
public void write(DataOutputStream outstr) throws IOException { |
|||
outstr.writeInt(entries.size()); |
|||
for (IndexEntry entry : entries) { |
|||
outstr.writeLong(entry.hashCode); |
|||
outstr.writeLong(entry.dataBlock.getPos()); |
|||
outstr.writeLong(entry.childIndexBlock.getPos()); |
|||
} |
|||
outstr.writeLong(tailPos.getPos()); |
|||
} |
|||
|
|||
public void put(long hashCode, BlockPointer pos) throws Exception { |
|||
int index = Collections.binarySearch(entries, new IndexEntry(hashCode)); |
|||
IndexEntry entry; |
|||
if (index >= 0) { |
|||
entry = entries.get(index); |
|||
} else { |
|||
assert tailPos.isNull(); |
|||
entry = new IndexEntry(); |
|||
entry.hashCode = hashCode; |
|||
entry.childIndexBlock = BlockPointer.start(); |
|||
index = -index - 1; |
|||
entries.add(index, entry); |
|||
} |
|||
|
|||
entry.dataBlock = pos; |
|||
store.write(this); |
|||
|
|||
maybeSplit(); |
|||
} |
|||
|
|||
private void maybeSplit() throws Exception { |
|||
if (entries.size() > maxChildIndexEntries) { |
|||
int splitPos = entries.size() / 2; |
|||
IndexEntry splitEntry = entries.remove(splitPos); |
|||
if (parent == null) { |
|||
parent = root.newRoot(); |
|||
} |
|||
IndexBlock sibling = new IndexBlock(); |
|||
store.write(sibling); |
|||
List<IndexEntry> siblingEntries = entries.subList(splitPos, entries.size()); |
|||
sibling.entries.addAll(siblingEntries); |
|||
siblingEntries.clear(); |
|||
sibling.tailPos = tailPos; |
|||
tailPos = splitEntry.childIndexBlock; |
|||
splitEntry.childIndexBlock = BlockPointer.start(); |
|||
parent.add(this, splitEntry, sibling); |
|||
} |
|||
} |
|||
|
|||
private void add(IndexBlock left, IndexEntry entry, IndexBlock right) throws Exception { |
|||
int index = left.parentEntryIndex; |
|||
if (index < entries.size()) { |
|||
IndexEntry parentEntry = entries.get(index); |
|||
assert parentEntry.childIndexBlock.equals(left.getPos()); |
|||
parentEntry.childIndexBlock = right.getPos(); |
|||
} else { |
|||
assert index == entries.size() && (tailPos.isNull() || tailPos.equals(left.getPos())); |
|||
tailPos = right.getPos(); |
|||
} |
|||
entries.add(index, entry); |
|||
entry.childIndexBlock = left.getPos(); |
|||
store.write(this); |
|||
|
|||
maybeSplit(); |
|||
} |
|||
|
|||
public DataBlock get(K key) throws Exception { |
|||
Lookup lookup = find(key); |
|||
if (lookup.entry == null) { |
|||
return null; |
|||
} |
|||
|
|||
return store.read(lookup.entry.dataBlock, DataBlock.class); |
|||
} |
|||
|
|||
public Lookup find(K key) throws Exception { |
|||
long checksum = keyHasher.getHashCode(key); |
|||
return find(checksum); |
|||
} |
|||
|
|||
private Lookup find(long hashCode) throws Exception { |
|||
int index = Collections.binarySearch(entries, new IndexEntry(hashCode)); |
|||
if (index >= 0) { |
|||
return new Lookup(this, entries.get(index)); |
|||
} |
|||
|
|||
index = -index - 1; |
|||
BlockPointer childBlockPos; |
|||
if (index == entries.size()) { |
|||
childBlockPos = tailPos; |
|||
} else { |
|||
childBlockPos = entries.get(index).childIndexBlock; |
|||
} |
|||
if (childBlockPos.isNull()) { |
|||
return new Lookup(this, null); |
|||
} |
|||
|
|||
IndexBlock childBlock = load(childBlockPos, root, this, index); |
|||
return childBlock.find(hashCode); |
|||
} |
|||
|
|||
public void remove(IndexEntry entry) throws Exception { |
|||
int index = entries.indexOf(entry); |
|||
assert index >= 0; |
|||
entries.remove(index); |
|||
store.write(this); |
|||
|
|||
if (entry.childIndexBlock.isNull()) { |
|||
maybeMerge(); |
|||
} else { |
|||
// Not a leaf node. Move up an entry from a leaf node, then possibly merge the leaf node |
|||
IndexBlock leafBlock = load(entry.childIndexBlock, root, this, index); |
|||
leafBlock = leafBlock.findHighestLeaf(); |
|||
IndexEntry highestEntry = leafBlock.entries.remove(leafBlock.entries.size() - 1); |
|||
highestEntry.childIndexBlock = entry.childIndexBlock; |
|||
entries.add(index, highestEntry); |
|||
store.write(leafBlock); |
|||
leafBlock.maybeMerge(); |
|||
} |
|||
} |
|||
|
|||
private void maybeMerge() throws Exception { |
|||
if (parent == null) { |
|||
// This is the root block. Can have any number of children <= maxChildIndexEntries |
|||
if (entries.size() == 0 && !tailPos.isNull()) { |
|||
// This is an empty root block, discard it |
|||
header.index.setRootPos(tailPos); |
|||
store.remove(this); |
|||
} |
|||
return; |
|||
} |
|||
|
|||
// This is not the root block. Must have children >= minIndexChildNodes |
|||
if (entries.size() >= minIndexChildNodes) { |
|||
return; |
|||
} |
|||
|
|||
// Attempt to merge with the left sibling |
|||
IndexBlock left = parent.getPrevious(this); |
|||
if (left != null) { |
|||
assert entries.size() + left.entries.size() <= maxChildIndexEntries * 2; |
|||
if (left.entries.size() > minIndexChildNodes) { |
|||
// There are enough entries in this block and the left sibling to make up 2 blocks, so redistribute |
|||
// the entries evenly between them |
|||
left.mergeFrom(this); |
|||
left.maybeSplit(); |
|||
return; |
|||
} else { |
|||
// There are only enough entries to make up 1 block, so move the entries of the left sibling into |
|||
// this block and discard the left sibling. Might also need to merge the parent |
|||
left.mergeFrom(this); |
|||
parent.maybeMerge(); |
|||
return; |
|||
} |
|||
} |
|||
|
|||
// Attempt to merge with the right sibling |
|||
IndexBlock right = parent.getNext(this); |
|||
if (right != null) { |
|||
assert entries.size() + right.entries.size() <= maxChildIndexEntries * 2; |
|||
if (right.entries.size() > minIndexChildNodes) { |
|||
// There are enough entries in this block and the right sibling to make up 2 blocks, so redistribute |
|||
// the entries evenly between them |
|||
mergeFrom(right); |
|||
maybeSplit(); |
|||
return; |
|||
} else { |
|||
// There are only enough entries to make up 1 block, so move the entries of the right sibling into |
|||
// this block and discard this block. Might also need to merge the parent |
|||
mergeFrom(right); |
|||
parent.maybeMerge(); |
|||
return; |
|||
} |
|||
} |
|||
|
|||
// Should not happen |
|||
throw new IllegalStateException(String.format("%s does not have any siblings.", getBlock())); |
|||
} |
|||
|
|||
private void mergeFrom(IndexBlock right) throws Exception { |
|||
IndexEntry newChildEntry = parent.entries.remove(parentEntryIndex); |
|||
if (right.getPos().equals(parent.tailPos)) { |
|||
parent.tailPos = getPos(); |
|||
} else { |
|||
IndexEntry newParentEntry = parent.entries.get(parentEntryIndex); |
|||
assert newParentEntry.childIndexBlock.equals(right.getPos()); |
|||
newParentEntry.childIndexBlock = getPos(); |
|||
} |
|||
entries.add(newChildEntry); |
|||
entries.addAll(right.entries); |
|||
newChildEntry.childIndexBlock = tailPos; |
|||
tailPos = right.tailPos; |
|||
store.write(parent); |
|||
store.write(this); |
|||
store.remove(right); |
|||
} |
|||
|
|||
private IndexBlock getNext(IndexBlock indexBlock) throws Exception { |
|||
int index = indexBlock.parentEntryIndex + 1; |
|||
if (index > entries.size()) { |
|||
return null; |
|||
} |
|||
if (index == entries.size()) { |
|||
return load(tailPos, root, this, index); |
|||
} |
|||
return load(entries.get(index).childIndexBlock, root, this, index); |
|||
} |
|||
|
|||
private IndexBlock getPrevious(IndexBlock indexBlock) throws Exception { |
|||
int index = indexBlock.parentEntryIndex - 1; |
|||
if (index < 0) { |
|||
return null; |
|||
} |
|||
return load(entries.get(index).childIndexBlock, root, this, index); |
|||
} |
|||
|
|||
private IndexBlock findHighestLeaf() throws Exception { |
|||
if (tailPos.isNull()) { |
|||
return this; |
|||
} |
|||
return load(tailPos, root, this, entries.size()).findHighestLeaf(); |
|||
} |
|||
} |
|||
|
|||
private static class IndexEntry implements Comparable<IndexEntry> { |
|||
long hashCode; |
|||
BlockPointer dataBlock; |
|||
BlockPointer childIndexBlock; |
|||
|
|||
private IndexEntry() { |
|||
} |
|||
|
|||
private IndexEntry(long hashCode) { |
|||
this.hashCode = hashCode; |
|||
} |
|||
|
|||
@Override |
|||
public int compareTo(IndexEntry indexEntry) { |
|||
if (hashCode > indexEntry.hashCode) { |
|||
return 1; |
|||
} |
|||
if (hashCode < indexEntry.hashCode) { |
|||
return -1; |
|||
} |
|||
return 0; |
|||
} |
|||
} |
|||
|
|||
private class Lookup { |
|||
final IndexBlock indexBlock; |
|||
final IndexEntry entry; |
|||
|
|||
private Lookup(IndexBlock indexBlock, IndexEntry entry) { |
|||
this.indexBlock = indexBlock; |
|||
this.entry = entry; |
|||
} |
|||
} |
|||
|
|||
private class DataBlock extends BlockPayload { |
|||
private int size; |
|||
private StreamByteBuffer buffer; |
|||
private V value; |
|||
|
|||
private DataBlock() { |
|||
} |
|||
|
|||
public DataBlock(V value) throws Exception { |
|||
this.value = value; |
|||
setValue(value); |
|||
size = buffer.totalBytesUnread(); |
|||
} |
|||
|
|||
public DataBlock(V value, StreamByteBuffer buffer) throws Exception { |
|||
this.value = value; |
|||
this.buffer = buffer; |
|||
size = buffer.totalBytesUnread(); |
|||
} |
|||
|
|||
public void setValue(V value) throws Exception { |
|||
buffer = StreamByteBuffer.createWithChunkSizeInDefaultRange(size); |
|||
KryoBackedEncoder encoder = new KryoBackedEncoder(buffer.getOutputStream()); |
|||
serializer.write(encoder, value); |
|||
encoder.flush(); |
|||
} |
|||
|
|||
public V getValue() throws Exception { |
|||
if (value == null) { |
|||
value = serializer.read(new KryoBackedDecoder(buffer.getInputStream())); |
|||
buffer = null; |
|||
} |
|||
return value; |
|||
} |
|||
|
|||
@Override |
|||
protected byte getType() { |
|||
return 0x33; |
|||
} |
|||
|
|||
@Override |
|||
protected int getSize() { |
|||
return 2 * Block.INT_SIZE + size; |
|||
} |
|||
|
|||
@Override |
|||
public void read(DataInputStream instr) throws Exception { |
|||
size = instr.readInt(); |
|||
int bytes = instr.readInt(); |
|||
buffer = StreamByteBuffer.of(instr, bytes); |
|||
} |
|||
|
|||
@Override |
|||
public void write(DataOutputStream outstr) throws Exception { |
|||
outstr.writeInt(size); |
|||
outstr.writeInt(buffer.totalBytesUnread()); |
|||
buffer.writeTo(outstr); |
|||
buffer = null; |
|||
} |
|||
|
|||
public DataBlockUpdateResult useNewValue(V value) throws Exception { |
|||
setValue(value); |
|||
boolean ok = buffer.totalBytesUnread() <= size; |
|||
if (ok) { |
|||
this.value = value; |
|||
store.write(this); |
|||
return DataBlockUpdateResult.success(); |
|||
} else { |
|||
return DataBlockUpdateResult.failed(buffer); |
|||
} |
|||
} |
|||
} |
|||
|
|||
private static class DataBlockUpdateResult { |
|||
private static final DataBlockUpdateResult SUCCESS = new DataBlockUpdateResult(true, null); |
|||
private final boolean success; |
|||
private final StreamByteBuffer serializedValue; |
|||
|
|||
private DataBlockUpdateResult(boolean success, StreamByteBuffer serializedValue) { |
|||
this.success = success; |
|||
this.serializedValue = serializedValue; |
|||
} |
|||
|
|||
static DataBlockUpdateResult success() { |
|||
return SUCCESS; |
|||
} |
|||
|
|||
static DataBlockUpdateResult failed(StreamByteBuffer serializedValue) { |
|||
return new DataBlockUpdateResult(false, serializedValue); |
|||
} |
|||
|
|||
public boolean isFailed() { |
|||
return !success; |
|||
} |
|||
|
|||
public StreamByteBuffer getSerializedValue() { |
|||
return serializedValue; |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,59 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
public abstract class Block { |
|||
static final int LONG_SIZE = 8; |
|||
static final int INT_SIZE = 4; |
|||
static final int SHORT_SIZE = 2; |
|||
|
|||
private BlockPayload payload; |
|||
|
|||
protected Block(BlockPayload payload) { |
|||
this.payload = payload; |
|||
payload.setBlock(this); |
|||
} |
|||
|
|||
public BlockPayload getPayload() { |
|||
return payload; |
|||
} |
|||
|
|||
protected void detach() { |
|||
payload.setBlock(null); |
|||
payload = null; |
|||
} |
|||
|
|||
public abstract BlockPointer getPos(); |
|||
|
|||
public abstract int getSize(); |
|||
|
|||
public abstract RuntimeException blockCorruptedException(); |
|||
|
|||
@Override |
|||
public String toString() { |
|||
return payload.getClass().getSimpleName() + " " + getPos(); |
|||
} |
|||
|
|||
public BlockPointer getNextPos() { |
|||
return BlockPointer.pos(getPos().getPos() + getSize()); |
|||
} |
|||
|
|||
public abstract boolean hasPos(); |
|||
|
|||
public abstract void setPos(BlockPointer pos); |
|||
|
|||
public abstract void setSize(int size); |
|||
} |
|||
@ -0,0 +1,51 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import java.io.DataInputStream; |
|||
import java.io.DataOutputStream; |
|||
|
|||
public abstract class BlockPayload { |
|||
private Block block; |
|||
|
|||
public Block getBlock() { |
|||
return block; |
|||
} |
|||
|
|||
public void setBlock(Block block) { |
|||
this.block = block; |
|||
} |
|||
|
|||
public BlockPointer getPos() { |
|||
return getBlock().getPos(); |
|||
} |
|||
|
|||
public BlockPointer getNextPos() { |
|||
return getBlock().getNextPos(); |
|||
} |
|||
|
|||
protected abstract int getSize(); |
|||
|
|||
protected abstract byte getType(); |
|||
|
|||
protected abstract void read(DataInputStream inputStream) throws Exception; |
|||
|
|||
protected abstract void write(DataOutputStream outputStream) throws Exception; |
|||
|
|||
protected RuntimeException blockCorruptedException() { |
|||
return getBlock().blockCorruptedException(); |
|||
} |
|||
} |
|||
@ -0,0 +1,75 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import com.google.common.primitives.Longs; |
|||
|
|||
public class BlockPointer implements Comparable<BlockPointer> { |
|||
|
|||
private static final BlockPointer NULL = new BlockPointer(-1); |
|||
|
|||
public static BlockPointer start() { |
|||
return NULL; |
|||
} |
|||
|
|||
public static BlockPointer pos(long pos) { |
|||
if (pos < -1) { |
|||
throw new CorruptedCacheException("block pointer must be >= -1, but was" + pos); |
|||
} |
|||
if (pos == -1) { |
|||
return NULL; |
|||
} |
|||
return new BlockPointer(pos); |
|||
} |
|||
|
|||
private final long pos; |
|||
|
|||
private BlockPointer(long pos) { |
|||
this.pos = pos; |
|||
} |
|||
|
|||
public boolean isNull() { |
|||
return pos < 0; |
|||
} |
|||
|
|||
public long getPos() { |
|||
return pos; |
|||
} |
|||
|
|||
@Override |
|||
public String toString() { |
|||
return String.valueOf(pos); |
|||
} |
|||
|
|||
@Override |
|||
public boolean equals(Object obj) { |
|||
if (obj == null || obj.getClass() != getClass()) { |
|||
return false; |
|||
} |
|||
BlockPointer other = (BlockPointer) obj; |
|||
return pos == other.pos; |
|||
} |
|||
|
|||
@Override |
|||
public int hashCode() { |
|||
return Longs.hashCode(pos); |
|||
} |
|||
|
|||
@Override |
|||
public int compareTo(BlockPointer o) { |
|||
return Longs.compare(pos, o.pos); |
|||
} |
|||
} |
|||
@ -0,0 +1,68 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
public interface BlockStore { |
|||
/** |
|||
* Opens this store, calling the given action if the store is empty. |
|||
*/ |
|||
void open(Runnable initAction, Factory factory); |
|||
|
|||
/** |
|||
* Closes this store. |
|||
*/ |
|||
void close(); |
|||
|
|||
/** |
|||
* Discards all blocks from this store. |
|||
*/ |
|||
void clear(); |
|||
|
|||
/** |
|||
* Removes the given block from this store. |
|||
*/ |
|||
void remove(BlockPayload block); |
|||
|
|||
/** |
|||
* Reads the first block from this store. |
|||
*/ |
|||
<T extends BlockPayload> T readFirst(Class<T> payloadType); |
|||
|
|||
/** |
|||
* Reads a block from this store. |
|||
*/ |
|||
<T extends BlockPayload> T read(BlockPointer pos, Class<T> payloadType); |
|||
|
|||
/** |
|||
* Writes a block to this store, adding the block if required. |
|||
*/ |
|||
void write(BlockPayload block); |
|||
|
|||
/** |
|||
* Adds a new block to this store. Allocates space for the block, but does not write the contents of the block |
|||
* until {@link #write(BlockPayload)} is called. |
|||
*/ |
|||
void attach(BlockPayload block); |
|||
|
|||
/** |
|||
* Flushes any pending updates for this store. |
|||
*/ |
|||
void flush(); |
|||
|
|||
interface Factory { |
|||
Object create(Class<? extends BlockPayload> type); |
|||
} |
|||
} |
|||
@ -0,0 +1,30 @@ |
|||
/* |
|||
* Copyright 2018 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree; |
|||
|
|||
import java.nio.Buffer; |
|||
|
|||
public class BufferCaster { |
|||
/** |
|||
* Without this cast, when the code compiled by Java 9+ is executed on Java 8, it will throw |
|||
* java.lang.NoSuchMethodError: Method flip()Ljava/nio/ByteBuffer; does not exist in class java.nio.ByteBuffer |
|||
*/ |
|||
@SuppressWarnings("RedundantCast") |
|||
public static <T extends Buffer> Buffer cast(T byteBuffer) { |
|||
return (Buffer) byteBuffer; |
|||
} |
|||
} |
|||
@ -0,0 +1,74 @@ |
|||
/* |
|||
* Copyright 2014 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree; |
|||
|
|||
import com.google.common.io.CountingInputStream; |
|||
|
|||
import java.io.BufferedInputStream; |
|||
import java.io.DataInputStream; |
|||
import java.io.IOException; |
|||
import java.io.InputStream; |
|||
import java.io.RandomAccessFile; |
|||
|
|||
/** |
|||
* Allows a stream of bytes to be read from a particular location of some backing byte stream. |
|||
*/ |
|||
class ByteInput { |
|||
private final RandomAccessFile file; |
|||
private final ResettableBufferedInputStream bufferedInputStream; |
|||
private CountingInputStream countingInputStream; |
|||
|
|||
public ByteInput(RandomAccessFile file) { |
|||
this.file = file; |
|||
bufferedInputStream = new ResettableBufferedInputStream(new RandomAccessFileInputStream(file)); |
|||
} |
|||
|
|||
/** |
|||
* Starts reading from the given offset. |
|||
*/ |
|||
public DataInputStream start(long offset) throws IOException { |
|||
file.seek(offset); |
|||
bufferedInputStream.clear(); |
|||
countingInputStream = new CountingInputStream(bufferedInputStream); |
|||
return new DataInputStream(countingInputStream); |
|||
} |
|||
|
|||
/** |
|||
* Returns the number of bytes read since {@link #start(long)} was called. |
|||
*/ |
|||
public long getBytesRead() { |
|||
return countingInputStream.getCount(); |
|||
} |
|||
|
|||
/** |
|||
* Finishes reading, resetting any buffered state. |
|||
*/ |
|||
public void done() { |
|||
countingInputStream = null; |
|||
} |
|||
|
|||
private static class ResettableBufferedInputStream extends BufferedInputStream { |
|||
ResettableBufferedInputStream(InputStream input) { |
|||
super(input); |
|||
} |
|||
|
|||
void clear() { |
|||
count = 0; |
|||
pos = 0; |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,74 @@ |
|||
/* |
|||
* Copyright 2014 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree; |
|||
|
|||
import com.google.common.io.CountingOutputStream; |
|||
|
|||
import java.io.BufferedOutputStream; |
|||
import java.io.DataOutputStream; |
|||
import java.io.IOException; |
|||
import java.io.OutputStream; |
|||
import java.io.RandomAccessFile; |
|||
|
|||
/** |
|||
* Allows a stream of bytes to be written to a particular location of some backing byte stream. |
|||
*/ |
|||
class ByteOutput { |
|||
private final RandomAccessFile file; |
|||
private final ResettableBufferedOutputStream bufferedOutputStream; |
|||
private CountingOutputStream countingOutputStream; |
|||
|
|||
public ByteOutput(RandomAccessFile file) { |
|||
this.file = file; |
|||
bufferedOutputStream = new ResettableBufferedOutputStream(new RandomAccessFileOutputStream(file)); |
|||
} |
|||
|
|||
/** |
|||
* Starts writing to the given offset. Can be beyond the current length of the file. |
|||
*/ |
|||
public DataOutputStream start(long offset) throws IOException { |
|||
file.seek(offset); |
|||
bufferedOutputStream.clear(); |
|||
countingOutputStream = new CountingOutputStream(bufferedOutputStream); |
|||
return new DataOutputStream(countingOutputStream); |
|||
} |
|||
|
|||
/** |
|||
* Returns the number of byte written since {@link #start(long)} was called. |
|||
*/ |
|||
public long getBytesWritten() { |
|||
return countingOutputStream.getCount(); |
|||
} |
|||
|
|||
/** |
|||
* Finishes writing, flushing and resetting any buffered state |
|||
*/ |
|||
public void done() throws IOException { |
|||
countingOutputStream.flush(); |
|||
countingOutputStream = null; |
|||
} |
|||
|
|||
private static class ResettableBufferedOutputStream extends BufferedOutputStream { |
|||
ResettableBufferedOutputStream(OutputStream output) { |
|||
super(output); |
|||
} |
|||
|
|||
void clear() { |
|||
count = 0; |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,129 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import com.google.common.cache.Cache; |
|||
import com.google.common.cache.CacheBuilder; |
|||
import com.google.common.collect.ImmutableSet; |
|||
|
|||
import javax.annotation.Nullable; |
|||
import java.util.Collection; |
|||
import java.util.Iterator; |
|||
import java.util.LinkedHashMap; |
|||
import java.util.Map; |
|||
|
|||
public class CachingBlockStore implements BlockStore { |
|||
private final BlockStore store; |
|||
private final Map<BlockPointer, BlockPayload> dirty = new LinkedHashMap<BlockPointer, BlockPayload>(); |
|||
private final Cache<BlockPointer, BlockPayload> indexBlockCache = CacheBuilder.newBuilder().maximumSize(100).concurrencyLevel(1).build(); |
|||
private final ImmutableSet<Class<? extends BlockPayload>> cacheableBlockTypes; |
|||
|
|||
public CachingBlockStore(BlockStore store, Collection<Class<? extends BlockPayload>> cacheableBlockTypes) { |
|||
this.store = store; |
|||
this.cacheableBlockTypes = ImmutableSet.copyOf(cacheableBlockTypes); |
|||
} |
|||
|
|||
@Override |
|||
public void open(Runnable initAction, Factory factory) { |
|||
store.open(initAction, factory); |
|||
} |
|||
|
|||
@Override |
|||
public void close() { |
|||
flush(); |
|||
indexBlockCache.invalidateAll(); |
|||
store.close(); |
|||
} |
|||
|
|||
@Override |
|||
public void clear() { |
|||
dirty.clear(); |
|||
indexBlockCache.invalidateAll(); |
|||
store.clear(); |
|||
} |
|||
|
|||
@Override |
|||
public void flush() { |
|||
Iterator<BlockPayload> iterator = dirty.values().iterator(); |
|||
while (iterator.hasNext()) { |
|||
BlockPayload block = iterator.next(); |
|||
iterator.remove(); |
|||
store.write(block); |
|||
} |
|||
store.flush(); |
|||
} |
|||
|
|||
@Override |
|||
public void attach(BlockPayload block) { |
|||
store.attach(block); |
|||
} |
|||
|
|||
@Override |
|||
public void remove(BlockPayload block) { |
|||
dirty.remove(block.getPos()); |
|||
if (isCacheable(block)) { |
|||
indexBlockCache.invalidate(block.getPos()); |
|||
} |
|||
store.remove(block); |
|||
} |
|||
|
|||
@Override |
|||
public <T extends BlockPayload> T readFirst(Class<T> payloadType) { |
|||
T block = store.readFirst(payloadType); |
|||
maybeCache(block); |
|||
return block; |
|||
} |
|||
|
|||
@Override |
|||
public <T extends BlockPayload> T read(BlockPointer pos, Class<T> payloadType) { |
|||
T block = payloadType.cast(dirty.get(pos)); |
|||
if (block != null) { |
|||
return block; |
|||
} |
|||
block = maybeGetFromCache(pos, payloadType); |
|||
if (block != null) { |
|||
return block; |
|||
} |
|||
block = store.read(pos, payloadType); |
|||
maybeCache(block); |
|||
return block; |
|||
} |
|||
|
|||
@Nullable |
|||
private <T extends BlockPayload> T maybeGetFromCache(BlockPointer pos, Class<T> payloadType) { |
|||
if (cacheableBlockTypes.contains(payloadType)) { |
|||
return payloadType.cast(indexBlockCache.getIfPresent(pos)); |
|||
} |
|||
return null; |
|||
} |
|||
|
|||
@Override |
|||
public void write(BlockPayload block) { |
|||
store.attach(block); |
|||
maybeCache(block); |
|||
dirty.put(block.getPos(), block); |
|||
} |
|||
|
|||
private <T extends BlockPayload> void maybeCache(T block) { |
|||
if (isCacheable(block)) { |
|||
indexBlockCache.put(block.getPos(), block); |
|||
} |
|||
} |
|||
|
|||
private <T extends BlockPayload> boolean isCacheable(T block) { |
|||
return cacheableBlockTypes.contains(block.getClass()); |
|||
} |
|||
} |
|||
@ -0,0 +1,22 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
class CorruptedCacheException extends RuntimeException { |
|||
CorruptedCacheException(String message) { |
|||
super(message); |
|||
} |
|||
} |
|||
@ -0,0 +1,274 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import java.io.DataInputStream; |
|||
import java.io.DataOutputStream; |
|||
import java.io.File; |
|||
import java.io.FileNotFoundException; |
|||
import java.io.IOException; |
|||
import java.io.RandomAccessFile; |
|||
|
|||
public class FileBackedBlockStore implements BlockStore { |
|||
private final File cacheFile; |
|||
private RandomAccessFile file; |
|||
private ByteOutput output; |
|||
private ByteInput input; |
|||
private long nextBlock; |
|||
private Factory factory; |
|||
private long currentFileSize; |
|||
|
|||
public FileBackedBlockStore(File cacheFile) { |
|||
this.cacheFile = cacheFile; |
|||
} |
|||
|
|||
@Override |
|||
public String toString() { |
|||
return "cache '" + cacheFile + "'"; |
|||
} |
|||
|
|||
@Override |
|||
public void open(Runnable runnable, Factory factory) { |
|||
this.factory = factory; |
|||
try { |
|||
cacheFile.getParentFile().mkdirs(); |
|||
file = openRandomAccessFile(); |
|||
output = new ByteOutput(file); |
|||
input = new ByteInput(file); |
|||
currentFileSize = file.length(); |
|||
nextBlock = currentFileSize; |
|||
if (currentFileSize == 0) { |
|||
runnable.run(); |
|||
} |
|||
} catch (IOException e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
} |
|||
|
|||
private RandomAccessFile openRandomAccessFile() throws FileNotFoundException { |
|||
try { |
|||
return randomAccessFile("rw"); |
|||
} catch (FileNotFoundException e) { |
|||
return randomAccessFile("r"); |
|||
} |
|||
} |
|||
|
|||
private RandomAccessFile randomAccessFile(String mode) throws FileNotFoundException { |
|||
return new RandomAccessFile(cacheFile, mode); |
|||
} |
|||
|
|||
@Override |
|||
public void close() { |
|||
try { |
|||
file.close(); |
|||
} catch (IOException e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void clear() { |
|||
try { |
|||
file.setLength(0); |
|||
currentFileSize = 0; |
|||
} catch (IOException e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
nextBlock = 0; |
|||
} |
|||
|
|||
@Override |
|||
public void attach(BlockPayload block) { |
|||
if (block.getBlock() == null) { |
|||
block.setBlock(new BlockImpl(block)); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void remove(BlockPayload block) { |
|||
BlockImpl blockImpl = (BlockImpl) block.getBlock(); |
|||
blockImpl.detach(); |
|||
} |
|||
|
|||
@Override |
|||
public void flush() { |
|||
} |
|||
|
|||
@Override |
|||
public <T extends BlockPayload> T readFirst(Class<T> payloadType) { |
|||
return read(BlockPointer.pos(0), payloadType); |
|||
} |
|||
|
|||
@Override |
|||
public <T extends BlockPayload> T read(BlockPointer pos, Class<T> payloadType) { |
|||
assert !pos.isNull(); |
|||
try { |
|||
T payload = payloadType.cast(factory.create(payloadType)); |
|||
BlockImpl block = new BlockImpl(payload, pos); |
|||
block.read(); |
|||
return payload; |
|||
} catch (CorruptedCacheException e) { |
|||
throw e; |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void write(BlockPayload block) { |
|||
BlockImpl blockImpl = (BlockImpl) block.getBlock(); |
|||
try { |
|||
blockImpl.write(); |
|||
} catch (CorruptedCacheException e) { |
|||
throw e; |
|||
} catch (Exception e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
} |
|||
|
|||
private long alloc(long length) { |
|||
long pos = nextBlock; |
|||
nextBlock += length; |
|||
return pos; |
|||
} |
|||
|
|||
private final class BlockImpl extends Block { |
|||
private static final int HEADER_SIZE = 1 + INT_SIZE; // type, payload size |
|||
private static final int TAIL_SIZE = INT_SIZE; |
|||
|
|||
private BlockPointer pos; |
|||
private int payloadSize; |
|||
|
|||
private BlockImpl(BlockPayload payload, BlockPointer pos) { |
|||
this(payload); |
|||
setPos(pos); |
|||
} |
|||
|
|||
public BlockImpl(BlockPayload payload) { |
|||
super(payload); |
|||
pos = null; |
|||
payloadSize = -1; |
|||
} |
|||
|
|||
@Override |
|||
public boolean hasPos() { |
|||
return pos != null; |
|||
} |
|||
|
|||
@Override |
|||
public BlockPointer getPos() { |
|||
if (pos == null) { |
|||
pos = BlockPointer.pos(alloc(getSize())); |
|||
} |
|||
return pos; |
|||
} |
|||
|
|||
@Override |
|||
public void setPos(BlockPointer pos) { |
|||
assert this.pos == null && !pos.isNull(); |
|||
this.pos = pos; |
|||
} |
|||
|
|||
@Override |
|||
public int getSize() { |
|||
if (payloadSize < 0) { |
|||
payloadSize = getPayload().getSize(); |
|||
} |
|||
return payloadSize + HEADER_SIZE + TAIL_SIZE; |
|||
} |
|||
|
|||
@Override |
|||
public void setSize(int size) { |
|||
int newPayloadSize = size - HEADER_SIZE - TAIL_SIZE; |
|||
assert newPayloadSize >= payloadSize; |
|||
payloadSize = newPayloadSize; |
|||
} |
|||
|
|||
public void write() throws Exception { |
|||
long pos = getPos().getPos(); |
|||
|
|||
DataOutputStream outputStream = output.start(pos); |
|||
|
|||
BlockPayload payload = getPayload(); |
|||
|
|||
// Write header |
|||
outputStream.writeByte(payload.getType()); |
|||
outputStream.writeInt(payloadSize); |
|||
long finalSize = pos + HEADER_SIZE + TAIL_SIZE + payloadSize; |
|||
|
|||
// Write body |
|||
payload.write(outputStream); |
|||
|
|||
// Write count |
|||
long bytesWritten = output.getBytesWritten(); |
|||
if (bytesWritten > Integer.MAX_VALUE) { |
|||
throw new IllegalArgumentException("Block payload exceeds maximum size"); |
|||
} |
|||
outputStream.writeInt((int) bytesWritten); |
|||
output.done(); |
|||
|
|||
// System.out.println(String.format("wrote [%d,%d)", pos, pos + bytesWritten + 4)); |
|||
|
|||
// Pad |
|||
if (currentFileSize < finalSize) { |
|||
// System.out.println(String.format("pad length %d => %d", currentFileSize, finalSize)); |
|||
file.setLength(finalSize); |
|||
currentFileSize = finalSize; |
|||
} |
|||
} |
|||
|
|||
public void read() throws Exception { |
|||
long pos = getPos().getPos(); |
|||
assert pos >= 0; |
|||
if (pos + HEADER_SIZE >= currentFileSize) { |
|||
throw blockCorruptedException(); |
|||
} |
|||
|
|||
DataInputStream inputStream = input.start(pos); |
|||
|
|||
BlockPayload payload = getPayload(); |
|||
|
|||
// Read header |
|||
byte type = inputStream.readByte(); |
|||
if (type != payload.getType()) { |
|||
throw blockCorruptedException(); |
|||
} |
|||
|
|||
// Read body |
|||
payloadSize = inputStream.readInt(); |
|||
if (pos + HEADER_SIZE + TAIL_SIZE + payloadSize > currentFileSize) { |
|||
throw blockCorruptedException(); |
|||
} |
|||
payload.read(inputStream); |
|||
|
|||
// Read and verify count |
|||
long actualCount = input.getBytesRead(); |
|||
long count = inputStream.readInt(); |
|||
if (actualCount != count) { |
|||
System.out.println(String.format("read expected %d actual %d, pos %d payloadSize %d currentFileSize %d", count, actualCount, pos, payloadSize, currentFileSize)); |
|||
throw blockCorruptedException(); |
|||
} |
|||
input.done(); |
|||
} |
|||
|
|||
@Override |
|||
public RuntimeException blockCorruptedException() { |
|||
return new CorruptedCacheException(String.format("Corrupted %s found in %s.", this, |
|||
FileBackedBlockStore.this)); |
|||
} |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,283 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import java.io.DataInputStream; |
|||
import java.io.DataOutputStream; |
|||
import java.util.ArrayList; |
|||
import java.util.Collections; |
|||
import java.util.List; |
|||
|
|||
public class FreeListBlockStore implements BlockStore { |
|||
private final BlockStore store; |
|||
private final BlockStore freeListStore; |
|||
private final int maxBlockEntries; |
|||
private FreeListBlock freeListBlock; |
|||
|
|||
public FreeListBlockStore(BlockStore store, int maxBlockEntries) { |
|||
this.store = store; |
|||
freeListStore = this; |
|||
this.maxBlockEntries = maxBlockEntries; |
|||
} |
|||
|
|||
@Override |
|||
public void open(final Runnable initAction, final Factory factory) { |
|||
Runnable freeListInitAction = new Runnable() { |
|||
@Override |
|||
public void run() { |
|||
freeListBlock = new FreeListBlock(); |
|||
store.write(freeListBlock); |
|||
store.flush(); |
|||
initAction.run(); |
|||
} |
|||
}; |
|||
Factory freeListFactory = new Factory() { |
|||
@Override |
|||
public Object create(Class<? extends BlockPayload> type) { |
|||
if (type == FreeListBlock.class) { |
|||
return new FreeListBlock(); |
|||
} |
|||
return factory.create(type); |
|||
} |
|||
}; |
|||
|
|||
store.open(freeListInitAction, freeListFactory); |
|||
freeListBlock = store.readFirst(FreeListBlock.class); |
|||
} |
|||
|
|||
@Override |
|||
public void close() { |
|||
freeListBlock = null; |
|||
store.close(); |
|||
} |
|||
|
|||
@Override |
|||
public void clear() { |
|||
store.clear(); |
|||
} |
|||
|
|||
@Override |
|||
public void remove(BlockPayload block) { |
|||
Block container = block.getBlock(); |
|||
store.remove(block); |
|||
freeListBlock.add(container.getPos(), container.getSize()); |
|||
} |
|||
|
|||
@Override |
|||
public <T extends BlockPayload> T readFirst(Class<T> payloadType) { |
|||
return store.read(freeListBlock.getNextPos(), payloadType); |
|||
} |
|||
|
|||
@Override |
|||
public <T extends BlockPayload> T read(BlockPointer pos, Class<T> payloadType) { |
|||
return store.read(pos, payloadType); |
|||
} |
|||
|
|||
@Override |
|||
public void write(BlockPayload block) { |
|||
attach(block); |
|||
store.write(block); |
|||
} |
|||
|
|||
@Override |
|||
public void attach(BlockPayload block) { |
|||
store.attach(block); |
|||
freeListBlock.alloc(block.getBlock()); |
|||
} |
|||
|
|||
@Override |
|||
public void flush() { |
|||
store.flush(); |
|||
} |
|||
|
|||
private void verify() { |
|||
FreeListBlock block = store.readFirst(FreeListBlock.class); |
|||
verify(block, Integer.MAX_VALUE); |
|||
} |
|||
|
|||
private void verify(FreeListBlock block, int maxValue) { |
|||
if (block.largestInNextBlock > maxValue) { |
|||
throw new RuntimeException("corrupt free list"); |
|||
} |
|||
int current = 0; |
|||
for (FreeListEntry entry : block.entries) { |
|||
if (entry.size > maxValue) { |
|||
throw new RuntimeException("corrupt free list"); |
|||
} |
|||
if (entry.size < block.largestInNextBlock) { |
|||
throw new RuntimeException("corrupt free list"); |
|||
} |
|||
if (entry.size < current) { |
|||
throw new RuntimeException("corrupt free list"); |
|||
} |
|||
current = entry.size; |
|||
} |
|||
if (!block.nextBlock.isNull()) { |
|||
verify(store.read(block.nextBlock, FreeListBlock.class), block.largestInNextBlock); |
|||
} |
|||
} |
|||
|
|||
public class FreeListBlock extends BlockPayload { |
|||
private List<FreeListEntry> entries = new ArrayList<FreeListEntry>(); |
|||
private int largestInNextBlock; |
|||
private BlockPointer nextBlock = BlockPointer.start(); |
|||
// Transient fields |
|||
private FreeListBlock prev; |
|||
private FreeListBlock next; |
|||
|
|||
@Override |
|||
protected int getSize() { |
|||
return Block.LONG_SIZE + Block.INT_SIZE + Block.INT_SIZE + maxBlockEntries * (Block.LONG_SIZE |
|||
+ Block.INT_SIZE); |
|||
} |
|||
|
|||
@Override |
|||
protected byte getType() { |
|||
return 0x44; |
|||
} |
|||
|
|||
@Override |
|||
protected void read(DataInputStream inputStream) throws Exception { |
|||
nextBlock = BlockPointer.pos(inputStream.readLong()); |
|||
largestInNextBlock = inputStream.readInt(); |
|||
int count = inputStream.readInt(); |
|||
for (int i = 0; i < count; i++) { |
|||
BlockPointer pos = BlockPointer.pos(inputStream.readLong()); |
|||
int size = inputStream.readInt(); |
|||
entries.add(new FreeListEntry(pos, size)); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
protected void write(DataOutputStream outputStream) throws Exception { |
|||
outputStream.writeLong(nextBlock.getPos()); |
|||
outputStream.writeInt(largestInNextBlock); |
|||
outputStream.writeInt(entries.size()); |
|||
for (FreeListEntry entry : entries) { |
|||
outputStream.writeLong(entry.pos.getPos()); |
|||
outputStream.writeInt(entry.size); |
|||
} |
|||
} |
|||
|
|||
public void add(BlockPointer pos, int size) { |
|||
assert !pos.isNull() && size >= 0; |
|||
if (size == 0) { |
|||
return; |
|||
} |
|||
|
|||
if (size < largestInNextBlock) { |
|||
FreeListBlock next = getNextBlock(); |
|||
next.add(pos, size); |
|||
return; |
|||
} |
|||
|
|||
FreeListEntry entry = new FreeListEntry(pos, size); |
|||
int index = Collections.binarySearch(entries, entry); |
|||
if (index < 0) { |
|||
index = -index - 1; |
|||
} |
|||
entries.add(index, entry); |
|||
|
|||
if (entries.size() > maxBlockEntries) { |
|||
FreeListBlock newBlock = new FreeListBlock(); |
|||
newBlock.largestInNextBlock = largestInNextBlock; |
|||
newBlock.nextBlock = nextBlock; |
|||
newBlock.prev = this; |
|||
newBlock.next = next; |
|||
next = newBlock; |
|||
|
|||
List<FreeListEntry> newBlockEntries = entries.subList(0, entries.size() / 2); |
|||
newBlock.entries.addAll(newBlockEntries); |
|||
newBlockEntries.clear(); |
|||
largestInNextBlock = newBlock.entries.get(newBlock.entries.size() - 1).size; |
|||
freeListStore.write(newBlock); |
|||
nextBlock = newBlock.getPos(); |
|||
} |
|||
|
|||
freeListStore.write(this); |
|||
} |
|||
|
|||
private FreeListBlock getNextBlock() { |
|||
if (next == null) { |
|||
next = freeListStore.read(nextBlock, FreeListBlock.class); |
|||
next.prev = this; |
|||
} |
|||
return next; |
|||
} |
|||
|
|||
public void alloc(Block block) { |
|||
if (block.hasPos()) { |
|||
return; |
|||
} |
|||
|
|||
int requiredSize = block.getSize(); |
|||
|
|||
if (entries.isEmpty() || requiredSize <= largestInNextBlock) { |
|||
if (nextBlock.isNull()) { |
|||
return; |
|||
} |
|||
getNextBlock().alloc(block); |
|||
return; |
|||
} |
|||
|
|||
int index = Collections.binarySearch(entries, new FreeListEntry(null, requiredSize)); |
|||
if (index < 0) { |
|||
index = -index - 1; |
|||
} |
|||
if (index == entries.size()) { |
|||
// Largest free block is too small |
|||
return; |
|||
} |
|||
|
|||
FreeListEntry entry = entries.remove(index); |
|||
block.setPos(entry.pos); |
|||
block.setSize(entry.size); |
|||
freeListStore.write(this); |
|||
|
|||
if (entries.size() == 0 && prev != null) { |
|||
prev.nextBlock = nextBlock; |
|||
prev.largestInNextBlock = largestInNextBlock; |
|||
prev.next = next; |
|||
if (next != null) { |
|||
next.prev = prev; |
|||
} |
|||
freeListStore.write(prev); |
|||
freeListStore.remove(this); |
|||
} |
|||
} |
|||
} |
|||
|
|||
private static class FreeListEntry implements Comparable<FreeListEntry> { |
|||
final BlockPointer pos; |
|||
final int size; |
|||
|
|||
private FreeListEntry(BlockPointer pos, int size) { |
|||
this.pos = pos; |
|||
this.size = size; |
|||
} |
|||
|
|||
@Override |
|||
public int compareTo(FreeListEntry o) { |
|||
if (size > o.size) { |
|||
return 1; |
|||
} |
|||
if (size < o.size) { |
|||
return -1; |
|||
} |
|||
return 0; |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,75 @@ |
|||
/* |
|||
* Copyright 2014 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree; |
|||
|
|||
import seaweedfs.client.btree.serialize.Serializer; |
|||
import seaweedfs.client.btree.serialize.kryo.KryoBackedEncoder; |
|||
|
|||
import java.io.IOException; |
|||
import java.io.OutputStream; |
|||
import java.math.BigInteger; |
|||
import java.security.MessageDigest; |
|||
import java.security.NoSuchAlgorithmException; |
|||
|
|||
class KeyHasher<K> { |
|||
private final Serializer<K> serializer; |
|||
private final MessageDigestStream digestStream = new MessageDigestStream(); |
|||
private final KryoBackedEncoder encoder = new KryoBackedEncoder(digestStream); |
|||
|
|||
public KeyHasher(Serializer<K> serializer) { |
|||
this.serializer = serializer; |
|||
} |
|||
|
|||
long getHashCode(K key) throws Exception { |
|||
serializer.write(encoder, key); |
|||
encoder.flush(); |
|||
return digestStream.getChecksum(); |
|||
} |
|||
|
|||
private static class MessageDigestStream extends OutputStream { |
|||
MessageDigest messageDigest; |
|||
|
|||
private MessageDigestStream() { |
|||
try { |
|||
messageDigest = MessageDigest.getInstance("MD5"); |
|||
} catch (NoSuchAlgorithmException e) { |
|||
throw UncheckedException.throwAsUncheckedException(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void write(int b) throws IOException { |
|||
messageDigest.update((byte) b); |
|||
} |
|||
|
|||
@Override |
|||
public void write(byte[] b) throws IOException { |
|||
messageDigest.update(b); |
|||
} |
|||
|
|||
@Override |
|||
public void write(byte[] b, int off, int len) throws IOException { |
|||
messageDigest.update(b, off, len); |
|||
} |
|||
|
|||
long getChecksum() { |
|||
byte[] digest = messageDigest.digest(); |
|||
assert digest.length == 16; |
|||
return new BigInteger(digest).longValue(); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,54 @@ |
|||
/* |
|||
* Copyright 2013 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import java.io.IOException; |
|||
import java.io.InputStream; |
|||
import java.io.RandomAccessFile; |
|||
|
|||
/** |
|||
* Reads from a {@link RandomAccessFile}. Each operation reads from and advances the current position of the file. |
|||
* |
|||
* <p>Closing this stream does not close the underlying file. |
|||
*/ |
|||
public class RandomAccessFileInputStream extends InputStream { |
|||
private final RandomAccessFile file; |
|||
|
|||
public RandomAccessFileInputStream(RandomAccessFile file) { |
|||
this.file = file; |
|||
} |
|||
|
|||
@Override |
|||
public long skip(long n) throws IOException { |
|||
file.seek(file.getFilePointer() + n); |
|||
return n; |
|||
} |
|||
|
|||
@Override |
|||
public int read(byte[] bytes) throws IOException { |
|||
return file.read(bytes); |
|||
} |
|||
|
|||
@Override |
|||
public int read() throws IOException { |
|||
return file.read(); |
|||
} |
|||
|
|||
@Override |
|||
public int read(byte[] bytes, int offset, int length) throws IOException { |
|||
return file.read(bytes, offset, length); |
|||
} |
|||
} |
|||
@ -0,0 +1,48 @@ |
|||
/* |
|||
* Copyright 2013 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import java.io.IOException; |
|||
import java.io.OutputStream; |
|||
import java.io.RandomAccessFile; |
|||
|
|||
/** |
|||
* Writes to a {@link RandomAccessFile}. Each operation writes to and advances the current position of the file. |
|||
* |
|||
* <p>Closing this stream does not close the underlying file. Flushing this stream does nothing. |
|||
*/ |
|||
public class RandomAccessFileOutputStream extends OutputStream { |
|||
private final RandomAccessFile file; |
|||
|
|||
public RandomAccessFileOutputStream(RandomAccessFile file) { |
|||
this.file = file; |
|||
} |
|||
|
|||
@Override |
|||
public void write(int i) throws IOException { |
|||
file.write(i); |
|||
} |
|||
|
|||
@Override |
|||
public void write(byte[] bytes) throws IOException { |
|||
file.write(bytes); |
|||
} |
|||
|
|||
@Override |
|||
public void write(byte[] bytes, int offset, int length) throws IOException { |
|||
file.write(bytes, offset, length); |
|||
} |
|||
} |
|||
@ -0,0 +1,87 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
public class StateCheckBlockStore implements BlockStore { |
|||
private final BlockStore blockStore; |
|||
private boolean open; |
|||
|
|||
public StateCheckBlockStore(BlockStore blockStore) { |
|||
this.blockStore = blockStore; |
|||
} |
|||
|
|||
@Override |
|||
public void open(Runnable initAction, Factory factory) { |
|||
assert !open; |
|||
open = true; |
|||
blockStore.open(initAction, factory); |
|||
} |
|||
|
|||
public boolean isOpen() { |
|||
return open; |
|||
} |
|||
|
|||
@Override |
|||
public void close() { |
|||
if (!open) { |
|||
return; |
|||
} |
|||
open = false; |
|||
blockStore.close(); |
|||
} |
|||
|
|||
@Override |
|||
public void clear() { |
|||
assert open; |
|||
blockStore.clear(); |
|||
} |
|||
|
|||
@Override |
|||
public void remove(BlockPayload block) { |
|||
assert open; |
|||
blockStore.remove(block); |
|||
} |
|||
|
|||
@Override |
|||
public <T extends BlockPayload> T readFirst(Class<T> payloadType) { |
|||
assert open; |
|||
return blockStore.readFirst(payloadType); |
|||
} |
|||
|
|||
@Override |
|||
public <T extends BlockPayload> T read(BlockPointer pos, Class<T> payloadType) { |
|||
assert open; |
|||
return blockStore.read(pos, payloadType); |
|||
} |
|||
|
|||
@Override |
|||
public void write(BlockPayload block) { |
|||
assert open; |
|||
blockStore.write(block); |
|||
} |
|||
|
|||
@Override |
|||
public void attach(BlockPayload block) { |
|||
assert open; |
|||
blockStore.attach(block); |
|||
} |
|||
|
|||
@Override |
|||
public void flush() { |
|||
assert open; |
|||
blockStore.flush(); |
|||
} |
|||
} |
|||
@ -0,0 +1,526 @@ |
|||
/* |
|||
* Copyright 2016 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import java.io.EOFException; |
|||
import java.io.IOException; |
|||
import java.io.InputStream; |
|||
import java.io.OutputStream; |
|||
import java.nio.ByteBuffer; |
|||
import java.nio.CharBuffer; |
|||
import java.nio.charset.CharacterCodingException; |
|||
import java.nio.charset.Charset; |
|||
import java.nio.charset.CharsetDecoder; |
|||
import java.nio.charset.CoderResult; |
|||
import java.nio.charset.CodingErrorAction; |
|||
import java.util.ArrayList; |
|||
import java.util.LinkedList; |
|||
import java.util.List; |
|||
|
|||
|
|||
/** |
|||
* An in-memory buffer that provides OutputStream and InputStream interfaces. |
|||
* |
|||
* This is more efficient than using ByteArrayOutputStream/ByteArrayInputStream |
|||
* |
|||
* Reading the buffer will clear the buffer. |
|||
* This is not thread-safe, it is intended to be used by a single Thread. |
|||
*/ |
|||
public class StreamByteBuffer { |
|||
private static final int DEFAULT_CHUNK_SIZE = 4096; |
|||
private static final int MAX_CHUNK_SIZE = 1024 * 1024; |
|||
private LinkedList<StreamByteBufferChunk> chunks = new LinkedList<StreamByteBufferChunk>(); |
|||
private StreamByteBufferChunk currentWriteChunk; |
|||
private StreamByteBufferChunk currentReadChunk; |
|||
private int chunkSize; |
|||
private int nextChunkSize; |
|||
private int maxChunkSize; |
|||
private StreamByteBufferOutputStream output; |
|||
private StreamByteBufferInputStream input; |
|||
private int totalBytesUnreadInList; |
|||
|
|||
public StreamByteBuffer() { |
|||
this(DEFAULT_CHUNK_SIZE); |
|||
} |
|||
|
|||
public StreamByteBuffer(int chunkSize) { |
|||
this.chunkSize = chunkSize; |
|||
this.nextChunkSize = chunkSize; |
|||
this.maxChunkSize = Math.max(chunkSize, MAX_CHUNK_SIZE); |
|||
currentWriteChunk = new StreamByteBufferChunk(nextChunkSize); |
|||
output = new StreamByteBufferOutputStream(); |
|||
input = new StreamByteBufferInputStream(); |
|||
} |
|||
|
|||
public static StreamByteBuffer of(InputStream inputStream) throws IOException { |
|||
StreamByteBuffer buffer = new StreamByteBuffer(chunkSizeInDefaultRange(inputStream.available())); |
|||
buffer.readFully(inputStream); |
|||
return buffer; |
|||
} |
|||
|
|||
public static StreamByteBuffer of(InputStream inputStream, int len) throws IOException { |
|||
StreamByteBuffer buffer = new StreamByteBuffer(chunkSizeInDefaultRange(len)); |
|||
buffer.readFrom(inputStream, len); |
|||
return buffer; |
|||
} |
|||
|
|||
public static StreamByteBuffer createWithChunkSizeInDefaultRange(int value) { |
|||
return new StreamByteBuffer(chunkSizeInDefaultRange(value)); |
|||
} |
|||
|
|||
static int chunkSizeInDefaultRange(int value) { |
|||
return valueInRange(value, DEFAULT_CHUNK_SIZE, MAX_CHUNK_SIZE); |
|||
} |
|||
|
|||
private static int valueInRange(int value, int min, int max) { |
|||
return Math.min(Math.max(value, min), max); |
|||
} |
|||
|
|||
public OutputStream getOutputStream() { |
|||
return output; |
|||
} |
|||
|
|||
public InputStream getInputStream() { |
|||
return input; |
|||
} |
|||
|
|||
public void writeTo(OutputStream target) throws IOException { |
|||
while (prepareRead() != -1) { |
|||
currentReadChunk.writeTo(target); |
|||
} |
|||
} |
|||
|
|||
public void readFrom(InputStream inputStream, int len) throws IOException { |
|||
int bytesLeft = len; |
|||
while (bytesLeft > 0) { |
|||
int spaceLeft = allocateSpace(); |
|||
int limit = Math.min(spaceLeft, bytesLeft); |
|||
int readBytes = currentWriteChunk.readFrom(inputStream, limit); |
|||
if (readBytes == -1) { |
|||
throw new EOFException("Unexpected EOF"); |
|||
} |
|||
bytesLeft -= readBytes; |
|||
} |
|||
} |
|||
|
|||
public void readFully(InputStream inputStream) throws IOException { |
|||
while (true) { |
|||
int len = allocateSpace(); |
|||
int readBytes = currentWriteChunk.readFrom(inputStream, len); |
|||
if (readBytes == -1) { |
|||
break; |
|||
} |
|||
} |
|||
} |
|||
|
|||
public byte[] readAsByteArray() { |
|||
byte[] buf = new byte[totalBytesUnread()]; |
|||
input.readImpl(buf, 0, buf.length); |
|||
return buf; |
|||
} |
|||
|
|||
public List<byte[]> readAsListOfByteArrays() { |
|||
List<byte[]> listOfByteArrays = new ArrayList<byte[]>(chunks.size() + 1); |
|||
byte[] buf; |
|||
while ((buf = input.readNextBuffer()) != null) { |
|||
if (buf.length > 0) { |
|||
listOfByteArrays.add(buf); |
|||
} |
|||
} |
|||
return listOfByteArrays; |
|||
} |
|||
|
|||
public String readAsString(String encoding) { |
|||
Charset charset = Charset.forName(encoding); |
|||
return readAsString(charset); |
|||
} |
|||
|
|||
public String readAsString() { |
|||
return readAsString(Charset.defaultCharset()); |
|||
} |
|||
|
|||
public String readAsString(Charset charset) { |
|||
try { |
|||
return doReadAsString(charset); |
|||
} catch (CharacterCodingException e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
} |
|||
|
|||
private String doReadAsString(Charset charset) throws CharacterCodingException { |
|||
int unreadSize = totalBytesUnread(); |
|||
if (unreadSize > 0) { |
|||
return readAsCharBuffer(charset).toString(); |
|||
} |
|||
return ""; |
|||
} |
|||
|
|||
private CharBuffer readAsCharBuffer(Charset charset) throws CharacterCodingException { |
|||
CharsetDecoder decoder = charset.newDecoder().onMalformedInput( |
|||
CodingErrorAction.REPLACE).onUnmappableCharacter( |
|||
CodingErrorAction.REPLACE); |
|||
CharBuffer charbuffer = CharBuffer.allocate(totalBytesUnread()); |
|||
ByteBuffer buf = null; |
|||
boolean wasUnderflow = false; |
|||
ByteBuffer nextBuf = null; |
|||
boolean needsFlush = false; |
|||
while (hasRemaining(nextBuf) || hasRemaining(buf) || prepareRead() != -1) { |
|||
if (hasRemaining(buf)) { |
|||
// handle decoding underflow, multi-byte unicode character at buffer chunk boundary |
|||
if (!wasUnderflow) { |
|||
throw new IllegalStateException("Unexpected state. Buffer has remaining bytes without underflow in decoding."); |
|||
} |
|||
if (!hasRemaining(nextBuf) && prepareRead() != -1) { |
|||
nextBuf = currentReadChunk.readToNioBuffer(); |
|||
} |
|||
// copy one by one until the underflow has been resolved |
|||
buf = ByteBuffer.allocate(buf.remaining() + 1).put(buf); |
|||
buf.put(nextBuf.get()); |
|||
BufferCaster.cast(buf).flip(); |
|||
} else { |
|||
if (hasRemaining(nextBuf)) { |
|||
buf = nextBuf; |
|||
} else if (prepareRead() != -1) { |
|||
buf = currentReadChunk.readToNioBuffer(); |
|||
if (!hasRemaining(buf)) { |
|||
throw new IllegalStateException("Unexpected state. Buffer is empty."); |
|||
} |
|||
} |
|||
nextBuf = null; |
|||
} |
|||
boolean endOfInput = !hasRemaining(nextBuf) && prepareRead() == -1; |
|||
int bufRemainingBefore = buf.remaining(); |
|||
CoderResult result = decoder.decode(buf, charbuffer, false); |
|||
if (bufRemainingBefore > buf.remaining()) { |
|||
needsFlush = true; |
|||
} |
|||
if (endOfInput) { |
|||
result = decoder.decode(ByteBuffer.allocate(0), charbuffer, true); |
|||
if (!result.isUnderflow()) { |
|||
result.throwException(); |
|||
} |
|||
break; |
|||
} |
|||
wasUnderflow = result.isUnderflow(); |
|||
} |
|||
if (needsFlush) { |
|||
CoderResult result = decoder.flush(charbuffer); |
|||
if (!result.isUnderflow()) { |
|||
result.throwException(); |
|||
} |
|||
} |
|||
clear(); |
|||
// push back remaining bytes of multi-byte unicode character |
|||
while (hasRemaining(buf)) { |
|||
byte b = buf.get(); |
|||
try { |
|||
getOutputStream().write(b); |
|||
} catch (IOException e) { |
|||
throw new UncheckedIOException(e); |
|||
} |
|||
} |
|||
BufferCaster.cast(charbuffer).flip(); |
|||
return charbuffer; |
|||
} |
|||
|
|||
private boolean hasRemaining(ByteBuffer nextBuf) { |
|||
return nextBuf != null && nextBuf.hasRemaining(); |
|||
} |
|||
|
|||
public int totalBytesUnread() { |
|||
int total = totalBytesUnreadInList; |
|||
if (currentReadChunk != null) { |
|||
total += currentReadChunk.bytesUnread(); |
|||
} |
|||
if (currentWriteChunk != currentReadChunk && currentWriteChunk != null) { |
|||
total += currentWriteChunk.bytesUnread(); |
|||
} |
|||
return total; |
|||
} |
|||
|
|||
protected int allocateSpace() { |
|||
int spaceLeft = currentWriteChunk.spaceLeft(); |
|||
if (spaceLeft == 0) { |
|||
addChunk(currentWriteChunk); |
|||
currentWriteChunk = new StreamByteBufferChunk(nextChunkSize); |
|||
if (nextChunkSize < maxChunkSize) { |
|||
nextChunkSize = Math.min(nextChunkSize * 2, maxChunkSize); |
|||
} |
|||
spaceLeft = currentWriteChunk.spaceLeft(); |
|||
} |
|||
return spaceLeft; |
|||
} |
|||
|
|||
protected int prepareRead() { |
|||
int bytesUnread = (currentReadChunk != null) ? currentReadChunk.bytesUnread() : 0; |
|||
if (bytesUnread == 0) { |
|||
if (!chunks.isEmpty()) { |
|||
currentReadChunk = chunks.removeFirst(); |
|||
bytesUnread = currentReadChunk.bytesUnread(); |
|||
totalBytesUnreadInList -= bytesUnread; |
|||
} else if (currentReadChunk != currentWriteChunk) { |
|||
currentReadChunk = currentWriteChunk; |
|||
bytesUnread = currentReadChunk.bytesUnread(); |
|||
} else { |
|||
bytesUnread = -1; |
|||
} |
|||
} |
|||
return bytesUnread; |
|||
} |
|||
|
|||
public static StreamByteBuffer of(List<byte[]> listOfByteArrays) { |
|||
StreamByteBuffer buffer = new StreamByteBuffer(); |
|||
buffer.addChunks(listOfByteArrays); |
|||
return buffer; |
|||
} |
|||
|
|||
private void addChunks(List<byte[]> listOfByteArrays) { |
|||
for (byte[] buf : listOfByteArrays) { |
|||
addChunk(new StreamByteBufferChunk(buf)); |
|||
} |
|||
} |
|||
|
|||
private void addChunk(StreamByteBufferChunk chunk) { |
|||
chunks.add(chunk); |
|||
totalBytesUnreadInList += chunk.bytesUnread(); |
|||
} |
|||
|
|||
static class StreamByteBufferChunk { |
|||
private int pointer; |
|||
private byte[] buffer; |
|||
private int size; |
|||
private int used; |
|||
|
|||
public StreamByteBufferChunk(int size) { |
|||
this.size = size; |
|||
buffer = new byte[size]; |
|||
} |
|||
|
|||
public StreamByteBufferChunk(byte[] buf) { |
|||
this.size = buf.length; |
|||
this.buffer = buf; |
|||
this.used = buf.length; |
|||
} |
|||
|
|||
public ByteBuffer readToNioBuffer() { |
|||
if (pointer < used) { |
|||
ByteBuffer result; |
|||
if (pointer > 0 || used < size) { |
|||
result = ByteBuffer.wrap(buffer, pointer, used - pointer); |
|||
} else { |
|||
result = ByteBuffer.wrap(buffer); |
|||
} |
|||
pointer = used; |
|||
return result; |
|||
} |
|||
|
|||
return null; |
|||
} |
|||
|
|||
public boolean write(byte b) { |
|||
if (used < size) { |
|||
buffer[used++] = b; |
|||
return true; |
|||
} |
|||
|
|||
return false; |
|||
} |
|||
|
|||
public void write(byte[] b, int off, int len) { |
|||
System.arraycopy(b, off, buffer, used, len); |
|||
used = used + len; |
|||
} |
|||
|
|||
public void read(byte[] b, int off, int len) { |
|||
System.arraycopy(buffer, pointer, b, off, len); |
|||
pointer = pointer + len; |
|||
} |
|||
|
|||
public void writeTo(OutputStream target) throws IOException { |
|||
if (pointer < used) { |
|||
target.write(buffer, pointer, used - pointer); |
|||
pointer = used; |
|||
} |
|||
} |
|||
|
|||
public void reset() { |
|||
pointer = 0; |
|||
} |
|||
|
|||
public int bytesUsed() { |
|||
return used; |
|||
} |
|||
|
|||
public int bytesUnread() { |
|||
return used - pointer; |
|||
} |
|||
|
|||
public int read() { |
|||
if (pointer < used) { |
|||
return buffer[pointer++] & 0xff; |
|||
} |
|||
|
|||
return -1; |
|||
} |
|||
|
|||
public int spaceLeft() { |
|||
return size - used; |
|||
} |
|||
|
|||
public int readFrom(InputStream inputStream, int len) throws IOException { |
|||
int readBytes = inputStream.read(buffer, used, len); |
|||
if(readBytes > 0) { |
|||
used += readBytes; |
|||
} |
|||
return readBytes; |
|||
} |
|||
|
|||
public void clear() { |
|||
used = pointer = 0; |
|||
} |
|||
|
|||
public byte[] readBuffer() { |
|||
if (used == buffer.length && pointer == 0) { |
|||
pointer = used; |
|||
return buffer; |
|||
} else if (pointer < used) { |
|||
byte[] buf = new byte[used - pointer]; |
|||
read(buf, 0, used - pointer); |
|||
return buf; |
|||
} else { |
|||
return new byte[0]; |
|||
} |
|||
} |
|||
} |
|||
|
|||
class StreamByteBufferOutputStream extends OutputStream { |
|||
private boolean closed; |
|||
|
|||
@Override |
|||
public void write(byte[] b, int off, int len) throws IOException { |
|||
if (b == null) { |
|||
throw new NullPointerException(); |
|||
} |
|||
|
|||
if ((off < 0) || (off > b.length) || (len < 0) |
|||
|| ((off + len) > b.length) || ((off + len) < 0)) { |
|||
throw new IndexOutOfBoundsException(); |
|||
} |
|||
|
|||
if (len == 0) { |
|||
return; |
|||
} |
|||
|
|||
int bytesLeft = len; |
|||
int currentOffset = off; |
|||
while (bytesLeft > 0) { |
|||
int spaceLeft = allocateSpace(); |
|||
int writeBytes = Math.min(spaceLeft, bytesLeft); |
|||
currentWriteChunk.write(b, currentOffset, writeBytes); |
|||
bytesLeft -= writeBytes; |
|||
currentOffset += writeBytes; |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void close() throws IOException { |
|||
closed = true; |
|||
} |
|||
|
|||
public boolean isClosed() { |
|||
return closed; |
|||
} |
|||
|
|||
@Override |
|||
public void write(int b) throws IOException { |
|||
allocateSpace(); |
|||
currentWriteChunk.write((byte) b); |
|||
} |
|||
|
|||
public StreamByteBuffer getBuffer() { |
|||
return StreamByteBuffer.this; |
|||
} |
|||
} |
|||
|
|||
class StreamByteBufferInputStream extends InputStream { |
|||
@Override |
|||
public int read() throws IOException { |
|||
prepareRead(); |
|||
return currentReadChunk.read(); |
|||
} |
|||
|
|||
@Override |
|||
public int read(byte[] b, int off, int len) throws IOException { |
|||
return readImpl(b, off, len); |
|||
} |
|||
|
|||
int readImpl(byte[] b, int off, int len) { |
|||
if (b == null) { |
|||
throw new NullPointerException(); |
|||
} |
|||
|
|||
if ((off < 0) || (off > b.length) || (len < 0) |
|||
|| ((off + len) > b.length) || ((off + len) < 0)) { |
|||
throw new IndexOutOfBoundsException(); |
|||
} |
|||
|
|||
if (len == 0) { |
|||
return 0; |
|||
} |
|||
|
|||
int bytesLeft = len; |
|||
int currentOffset = off; |
|||
int bytesUnread = prepareRead(); |
|||
int totalBytesRead = 0; |
|||
while (bytesLeft > 0 && bytesUnread != -1) { |
|||
int readBytes = Math.min(bytesUnread, bytesLeft); |
|||
currentReadChunk.read(b, currentOffset, readBytes); |
|||
bytesLeft -= readBytes; |
|||
currentOffset += readBytes; |
|||
totalBytesRead += readBytes; |
|||
bytesUnread = prepareRead(); |
|||
} |
|||
if (totalBytesRead > 0) { |
|||
return totalBytesRead; |
|||
} |
|||
|
|||
return -1; |
|||
} |
|||
|
|||
@Override |
|||
public int available() throws IOException { |
|||
return totalBytesUnread(); |
|||
} |
|||
|
|||
public StreamByteBuffer getBuffer() { |
|||
return StreamByteBuffer.this; |
|||
} |
|||
|
|||
public byte[] readNextBuffer() { |
|||
if (prepareRead() != -1) { |
|||
return currentReadChunk.readBuffer(); |
|||
} |
|||
return null; |
|||
} |
|||
} |
|||
|
|||
public void clear() { |
|||
chunks.clear(); |
|||
currentReadChunk = null; |
|||
totalBytesUnreadInList = 0; |
|||
currentWriteChunk.clear(); |
|||
} |
|||
} |
|||
@ -0,0 +1,88 @@ |
|||
/* |
|||
* Copyright 2010 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree; |
|||
|
|||
import java.io.IOException; |
|||
import java.lang.reflect.InvocationTargetException; |
|||
import java.util.concurrent.Callable; |
|||
|
|||
/** |
|||
* Wraps a checked exception. Carries no other context. |
|||
*/ |
|||
public final class UncheckedException extends RuntimeException { |
|||
private UncheckedException(Throwable cause) { |
|||
super(cause); |
|||
} |
|||
|
|||
private UncheckedException(String message, Throwable cause) { |
|||
super(message, cause); |
|||
} |
|||
|
|||
/** |
|||
* Note: always throws the failure in some form. The return value is to keep the compiler happy. |
|||
*/ |
|||
public static RuntimeException throwAsUncheckedException(Throwable t) { |
|||
return throwAsUncheckedException(t, false); |
|||
} |
|||
|
|||
/** |
|||
* Note: always throws the failure in some form. The return value is to keep the compiler happy. |
|||
*/ |
|||
public static RuntimeException throwAsUncheckedException(Throwable t, boolean preserveMessage) { |
|||
if (t instanceof InterruptedException) { |
|||
Thread.currentThread().interrupt(); |
|||
} |
|||
if (t instanceof RuntimeException) { |
|||
throw (RuntimeException) t; |
|||
} |
|||
if (t instanceof Error) { |
|||
throw (Error) t; |
|||
} |
|||
if (t instanceof IOException) { |
|||
if (preserveMessage) { |
|||
throw new UncheckedIOException(t.getMessage(), t); |
|||
} else { |
|||
throw new UncheckedIOException(t); |
|||
} |
|||
} |
|||
if (preserveMessage) { |
|||
throw new UncheckedException(t.getMessage(), t); |
|||
} else { |
|||
throw new UncheckedException(t); |
|||
} |
|||
} |
|||
|
|||
public static <T> T callUnchecked(Callable<T> callable) { |
|||
try { |
|||
return callable.call(); |
|||
} catch (Exception e) { |
|||
throw throwAsUncheckedException(e); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* Unwraps passed InvocationTargetException hence making the stack of exceptions cleaner without losing information. |
|||
* |
|||
* Note: always throws the failure in some form. The return value is to keep the compiler happy. |
|||
* |
|||
* @param e to be unwrapped |
|||
* @return an instance of RuntimeException based on the target exception of the parameter. |
|||
*/ |
|||
public static RuntimeException unwrapAndRethrow(InvocationTargetException e) { |
|||
return UncheckedException.throwAsUncheckedException(e.getTargetException()); |
|||
} |
|||
} |
|||
@ -0,0 +1,36 @@ |
|||
/* |
|||
* Copyright 2012 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
/** |
|||
* <code>UncheckedIOException</code> is used to wrap an {@link java.io.IOException} into an unchecked exception. |
|||
*/ |
|||
public class UncheckedIOException extends RuntimeException { |
|||
public UncheckedIOException() { |
|||
} |
|||
|
|||
public UncheckedIOException(String message) { |
|||
super(message); |
|||
} |
|||
|
|||
public UncheckedIOException(String message, Throwable cause) { |
|||
super(message, cause); |
|||
} |
|||
|
|||
public UncheckedIOException(Throwable cause) { |
|||
super(cause); |
|||
} |
|||
} |
|||
@ -0,0 +1,133 @@ |
|||
/* |
|||
* Copyright 2013 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import javax.annotation.Nullable; |
|||
import java.io.EOFException; |
|||
import java.io.IOException; |
|||
import java.io.InputStream; |
|||
|
|||
public abstract class AbstractDecoder implements Decoder { |
|||
private DecoderStream stream; |
|||
|
|||
@Override |
|||
public InputStream getInputStream() { |
|||
if (stream == null) { |
|||
stream = new DecoderStream(); |
|||
} |
|||
return stream; |
|||
} |
|||
|
|||
@Override |
|||
public void readBytes(byte[] buffer) throws IOException { |
|||
readBytes(buffer, 0, buffer.length); |
|||
} |
|||
|
|||
@Override |
|||
public byte[] readBinary() throws EOFException, IOException { |
|||
int size = readSmallInt(); |
|||
byte[] result = new byte[size]; |
|||
readBytes(result); |
|||
return result; |
|||
} |
|||
|
|||
@Override |
|||
public int readSmallInt() throws EOFException, IOException { |
|||
return readInt(); |
|||
} |
|||
|
|||
@Override |
|||
public long readSmallLong() throws EOFException, IOException { |
|||
return readLong(); |
|||
} |
|||
|
|||
@Nullable |
|||
@Override |
|||
public Integer readNullableSmallInt() throws IOException { |
|||
if (readBoolean()) { |
|||
return readSmallInt(); |
|||
} else { |
|||
return null; |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public String readNullableString() throws EOFException, IOException { |
|||
if (readBoolean()) { |
|||
return readString(); |
|||
} else { |
|||
return null; |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void skipBytes(long count) throws EOFException, IOException { |
|||
long remaining = count; |
|||
while (remaining > 0) { |
|||
long skipped = maybeSkip(remaining); |
|||
if (skipped <= 0) { |
|||
break; |
|||
} |
|||
remaining -= skipped; |
|||
} |
|||
if (remaining > 0) { |
|||
throw new EOFException(); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public <T> T decodeChunked(DecodeAction<Decoder, T> decodeAction) throws EOFException, Exception { |
|||
throw new UnsupportedOperationException(); |
|||
} |
|||
|
|||
@Override |
|||
public void skipChunked() throws EOFException, IOException { |
|||
throw new UnsupportedOperationException(); |
|||
} |
|||
|
|||
protected abstract int maybeReadBytes(byte[] buffer, int offset, int count) throws IOException; |
|||
|
|||
protected abstract long maybeSkip(long count) throws IOException; |
|||
|
|||
private class DecoderStream extends InputStream { |
|||
byte[] buffer = new byte[1]; |
|||
|
|||
@Override |
|||
public long skip(long n) throws IOException { |
|||
return maybeSkip(n); |
|||
} |
|||
|
|||
@Override |
|||
public int read() throws IOException { |
|||
int read = maybeReadBytes(buffer, 0, 1); |
|||
if (read <= 0) { |
|||
return read; |
|||
} |
|||
return buffer[0] & 0xff; |
|||
} |
|||
|
|||
@Override |
|||
public int read(byte[] buffer) throws IOException { |
|||
return maybeReadBytes(buffer, 0, buffer.length); |
|||
} |
|||
|
|||
@Override |
|||
public int read(byte[] buffer, int offset, int count) throws IOException { |
|||
return maybeReadBytes(buffer, offset, count); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,101 @@ |
|||
/* |
|||
* Copyright 2013 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import javax.annotation.Nullable; |
|||
import java.io.IOException; |
|||
import java.io.OutputStream; |
|||
|
|||
public abstract class AbstractEncoder implements Encoder { |
|||
private EncoderStream stream; |
|||
|
|||
@Override |
|||
public OutputStream getOutputStream() { |
|||
if (stream == null) { |
|||
stream = new EncoderStream(); |
|||
} |
|||
return stream; |
|||
} |
|||
|
|||
@Override |
|||
public void writeBytes(byte[] bytes) throws IOException { |
|||
writeBytes(bytes, 0, bytes.length); |
|||
} |
|||
|
|||
@Override |
|||
public void writeBinary(byte[] bytes) throws IOException { |
|||
writeBinary(bytes, 0, bytes.length); |
|||
} |
|||
|
|||
@Override |
|||
public void writeBinary(byte[] bytes, int offset, int count) throws IOException { |
|||
writeSmallInt(count); |
|||
writeBytes(bytes, offset, count); |
|||
} |
|||
|
|||
@Override |
|||
public void encodeChunked(EncodeAction<Encoder> writeAction) throws Exception { |
|||
throw new UnsupportedOperationException(); |
|||
} |
|||
|
|||
@Override |
|||
public void writeSmallInt(int value) throws IOException { |
|||
writeInt(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeSmallLong(long value) throws IOException { |
|||
writeLong(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeNullableSmallInt(@Nullable Integer value) throws IOException { |
|||
if (value == null) { |
|||
writeBoolean(false); |
|||
} else { |
|||
writeBoolean(true); |
|||
writeSmallInt(value); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void writeNullableString(@Nullable CharSequence value) throws IOException { |
|||
if (value == null) { |
|||
writeBoolean(false); |
|||
} else { |
|||
writeBoolean(true); |
|||
writeString(value.toString()); |
|||
} |
|||
} |
|||
|
|||
private class EncoderStream extends OutputStream { |
|||
@Override |
|||
public void write(byte[] buffer) throws IOException { |
|||
writeBytes(buffer); |
|||
} |
|||
|
|||
@Override |
|||
public void write(byte[] buffer, int offset, int length) throws IOException { |
|||
writeBytes(buffer, offset, length); |
|||
} |
|||
|
|||
@Override |
|||
public void write(int b) throws IOException { |
|||
writeByte((byte) b); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,40 @@ |
|||
/* |
|||
* Copyright 2016 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import com.google.common.base.Objects; |
|||
|
|||
/** |
|||
* This abstract class provide a sensible default implementation for {@code Serializer} equality. This equality |
|||
* implementation is required to enable cache instance reuse within the same Gradle runtime. Serializers are used |
|||
* as cache parameter which need to be compared to determine compatible cache. |
|||
*/ |
|||
public abstract class AbstractSerializer<T> implements Serializer<T> { |
|||
@Override |
|||
public boolean equals(Object obj) { |
|||
if (obj == null) { |
|||
return false; |
|||
} |
|||
|
|||
return Objects.equal(obj.getClass(), getClass()); |
|||
} |
|||
|
|||
@Override |
|||
public int hashCode() { |
|||
return Objects.hashCode(getClass()); |
|||
} |
|||
} |
|||
@ -0,0 +1,79 @@ |
|||
/* |
|||
* Copyright 2012 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import javax.annotation.Nullable; |
|||
|
|||
public abstract class Cast { |
|||
|
|||
/** |
|||
* Casts the given object to the given type, providing a better error message than the default. |
|||
* |
|||
* The standard {@link Class#cast(Object)} method produces unsatisfactory error messages on some platforms |
|||
* when it fails. All this method does is provide a better, consistent, error message. |
|||
* |
|||
* This should be used whenever there is a chance the cast could fail. If in doubt, use this. |
|||
* |
|||
* @param outputType The type to cast the input to |
|||
* @param object The object to be cast (must not be {@code null}) |
|||
* @param <O> The type to be cast to |
|||
* @param <I> The type of the object to be vast |
|||
* @return The input object, cast to the output type |
|||
*/ |
|||
public static <O, I> O cast(Class<O> outputType, I object) { |
|||
try { |
|||
return outputType.cast(object); |
|||
} catch (ClassCastException e) { |
|||
throw new ClassCastException(String.format( |
|||
"Failed to cast object %s of type %s to target type %s", object, object.getClass().getName(), outputType.getName() |
|||
)); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* Casts the given object to the given type, providing a better error message than the default. |
|||
* |
|||
* The standard {@link Class#cast(Object)} method produces unsatisfactory error messages on some platforms |
|||
* when it fails. All this method does is provide a better, consistent, error message. |
|||
* |
|||
* This should be used whenever there is a chance the cast could fail. If in doubt, use this. |
|||
* |
|||
* @param outputType The type to cast the input to |
|||
* @param object The object to be cast |
|||
* @param <O> The type to be cast to |
|||
* @param <I> The type of the object to be vast |
|||
* @return The input object, cast to the output type |
|||
*/ |
|||
@Nullable |
|||
public static <O, I> O castNullable(Class<O> outputType, @Nullable I object) { |
|||
if (object == null) { |
|||
return null; |
|||
} |
|||
return cast(outputType, object); |
|||
} |
|||
|
|||
@SuppressWarnings("unchecked") |
|||
@Nullable |
|||
public static <T> T uncheckedCast(@Nullable Object object) { |
|||
return (T) object; |
|||
} |
|||
|
|||
@SuppressWarnings("unchecked") |
|||
public static <T> T uncheckedNonnullCast(Object object) { |
|||
return (T) object; |
|||
} |
|||
} |
|||
@ -0,0 +1,43 @@ |
|||
/* |
|||
* Copyright 2010 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import java.io.IOException; |
|||
import java.io.InputStream; |
|||
import java.io.ObjectInputStream; |
|||
import java.io.ObjectStreamClass; |
|||
|
|||
public class ClassLoaderObjectInputStream extends ObjectInputStream { |
|||
private final ClassLoader loader; |
|||
|
|||
public ClassLoaderObjectInputStream(InputStream in, ClassLoader loader) throws IOException { |
|||
super(in); |
|||
this.loader = loader; |
|||
} |
|||
|
|||
public ClassLoader getClassLoader() { |
|||
return loader; |
|||
} |
|||
|
|||
@Override |
|||
protected Class<?> resolveClass(ObjectStreamClass desc) throws IOException, ClassNotFoundException { |
|||
try { |
|||
return Class.forName(desc.getName(), false, loader); |
|||
} catch (ClassNotFoundException e) { |
|||
return super.resolveClass(desc); |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,140 @@ |
|||
/* |
|||
* Copyright 2013 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import javax.annotation.Nullable; |
|||
import java.io.EOFException; |
|||
import java.io.IOException; |
|||
import java.io.InputStream; |
|||
|
|||
/** |
|||
* Provides a way to decode structured data from a backing byte stream. Implementations may buffer incoming bytes read |
|||
* from the backing stream prior to decoding. |
|||
*/ |
|||
public interface Decoder { |
|||
/** |
|||
* Returns an InputStream which can be used to read raw bytes. |
|||
*/ |
|||
InputStream getInputStream(); |
|||
|
|||
/** |
|||
* Reads a signed 64 bit long value. Can read any value that was written using {@link Encoder#writeLong(long)}. |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the long value can be fully read. |
|||
*/ |
|||
long readLong() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a signed 64 bit int value. Can read any value that was written using {@link Encoder#writeSmallLong(long)}. |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the int value can be fully read. |
|||
*/ |
|||
long readSmallLong() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a signed 32 bit int value. Can read any value that was written using {@link Encoder#writeInt(int)}. |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the int value can be fully read. |
|||
*/ |
|||
int readInt() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a signed 32 bit int value. Can read any value that was written using {@link Encoder#writeSmallInt(int)}. |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the int value can be fully read. |
|||
*/ |
|||
int readSmallInt() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a nullable signed 32 bit int value. |
|||
* |
|||
* @see #readSmallInt() |
|||
*/ |
|||
@Nullable |
|||
Integer readNullableSmallInt() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a boolean value. Can read any value that was written using {@link Encoder#writeBoolean(boolean)}. |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the boolean value can be fully read. |
|||
*/ |
|||
boolean readBoolean() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a non-null string value. Can read any value that was written using {@link Encoder#writeString(CharSequence)}. |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the string can be fully read. |
|||
*/ |
|||
String readString() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a nullable string value. Can reads any value that was written using {@link Encoder#writeNullableString(CharSequence)}. |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the string can be fully read. |
|||
*/ |
|||
@Nullable |
|||
String readNullableString() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a byte value. Can read any byte value that was written using one of the raw byte methods on {@link Encoder}, such as {@link Encoder#writeByte(byte)} or {@link Encoder#getOutputStream()} |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached. |
|||
*/ |
|||
byte readByte() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads bytes into the given buffer, filling the buffer. Can read any byte values that were written using one of the raw byte methods on {@link Encoder}, such as {@link |
|||
* Encoder#writeBytes(byte[])} or {@link Encoder#getOutputStream()} |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the buffer is full. |
|||
*/ |
|||
void readBytes(byte[] buffer) throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads the specified number of bytes into the given buffer. Can read any byte values that were written using one of the raw byte methods on {@link Encoder}, such as {@link |
|||
* Encoder#writeBytes(byte[])} or {@link Encoder#getOutputStream()} |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the specified number of bytes were read. |
|||
*/ |
|||
void readBytes(byte[] buffer, int offset, int count) throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a byte array. Can read any byte array written using {@link Encoder#writeBinary(byte[])} or {@link Encoder#writeBinary(byte[], int, int)}. |
|||
* |
|||
* @throws EOFException when the end of the byte stream is reached before the byte array was fully read. |
|||
*/ |
|||
byte[] readBinary() throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Skips the given number of bytes. Can skip over any byte values that were written using one of the raw byte methods on {@link Encoder}. |
|||
*/ |
|||
void skipBytes(long count) throws EOFException, IOException; |
|||
|
|||
/** |
|||
* Reads a byte stream written using {@link Encoder#encodeChunked(Encoder.EncodeAction)}. |
|||
*/ |
|||
<T> T decodeChunked(DecodeAction<Decoder, T> decodeAction) throws EOFException, Exception; |
|||
|
|||
/** |
|||
* Skips over a byte stream written using {@link Encoder#encodeChunked(Encoder.EncodeAction)}, discarding its content. |
|||
*/ |
|||
void skipChunked() throws EOFException, IOException; |
|||
|
|||
interface DecodeAction<IN, OUT> { |
|||
OUT read(IN source) throws Exception; |
|||
} |
|||
} |
|||
@ -0,0 +1,73 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import com.google.common.base.Objects; |
|||
|
|||
import java.io.IOException; |
|||
import java.io.ObjectOutputStream; |
|||
import java.io.StreamCorruptedException; |
|||
|
|||
public class DefaultSerializer<T> extends AbstractSerializer<T> { |
|||
private ClassLoader classLoader; |
|||
|
|||
public DefaultSerializer() { |
|||
classLoader = getClass().getClassLoader(); |
|||
} |
|||
|
|||
public DefaultSerializer(ClassLoader classLoader) { |
|||
this.classLoader = classLoader != null ? classLoader : getClass().getClassLoader(); |
|||
} |
|||
|
|||
public ClassLoader getClassLoader() { |
|||
return classLoader; |
|||
} |
|||
|
|||
public void setClassLoader(ClassLoader classLoader) { |
|||
this.classLoader = classLoader; |
|||
} |
|||
|
|||
@Override |
|||
public T read(Decoder decoder) throws Exception { |
|||
try { |
|||
return Cast.uncheckedNonnullCast(new ClassLoaderObjectInputStream(decoder.getInputStream(), classLoader).readObject()); |
|||
} catch (StreamCorruptedException e) { |
|||
return null; |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void write(Encoder encoder, T value) throws IOException { |
|||
ObjectOutputStream objectStr = new ObjectOutputStream(encoder.getOutputStream()); |
|||
objectStr.writeObject(value); |
|||
objectStr.flush(); |
|||
} |
|||
|
|||
@Override |
|||
public boolean equals(Object obj) { |
|||
if (!super.equals(obj)) { |
|||
return false; |
|||
} |
|||
|
|||
DefaultSerializer<?> rhs = (DefaultSerializer<?>) obj; |
|||
return Objects.equal(classLoader, rhs.classLoader); |
|||
} |
|||
|
|||
@Override |
|||
public int hashCode() { |
|||
return Objects.hashCode(super.hashCode(), classLoader); |
|||
} |
|||
} |
|||
@ -0,0 +1,110 @@ |
|||
/* |
|||
* Copyright 2013 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import javax.annotation.Nullable; |
|||
import java.io.IOException; |
|||
import java.io.OutputStream; |
|||
|
|||
/** |
|||
* Provides a way to encode structured data to a backing byte stream. Implementations may buffer outgoing encoded bytes prior |
|||
* to writing to the backing byte stream. |
|||
*/ |
|||
public interface Encoder { |
|||
/** |
|||
* Returns an {@link OutputStream) that can be used to write raw bytes to the stream. |
|||
*/ |
|||
OutputStream getOutputStream(); |
|||
|
|||
/** |
|||
* Writes a raw byte value to the stream. |
|||
*/ |
|||
void writeByte(byte value) throws IOException; |
|||
|
|||
/** |
|||
* Writes the given raw bytes to the stream. Does not encode any length information. |
|||
*/ |
|||
void writeBytes(byte[] bytes) throws IOException; |
|||
|
|||
/** |
|||
* Writes the given raw bytes to the stream. Does not encode any length information. |
|||
*/ |
|||
void writeBytes(byte[] bytes, int offset, int count) throws IOException; |
|||
|
|||
/** |
|||
* Writes the given byte array to the stream. Encodes the bytes and length information. |
|||
*/ |
|||
void writeBinary(byte[] bytes) throws IOException; |
|||
|
|||
/** |
|||
* Writes the given byte array to the stream. Encodes the bytes and length information. |
|||
*/ |
|||
void writeBinary(byte[] bytes, int offset, int count) throws IOException; |
|||
|
|||
/** |
|||
* Appends an encoded stream to this stream. Encodes the stream as a series of chunks with length information. |
|||
*/ |
|||
void encodeChunked(EncodeAction<Encoder> writeAction) throws Exception; |
|||
|
|||
/** |
|||
* Writes a signed 64 bit long value. The implementation may encode the value as a variable number of bytes, not necessarily as 8 bytes. |
|||
*/ |
|||
void writeLong(long value) throws IOException; |
|||
|
|||
/** |
|||
* Writes a signed 64 bit long value whose value is likely to be small and positive but may not be. The implementation may encode the value in a way that is more efficient for small positive |
|||
* values. |
|||
*/ |
|||
void writeSmallLong(long value) throws IOException; |
|||
|
|||
/** |
|||
* Writes a signed 32 bit int value. The implementation may encode the value as a variable number of bytes, not necessarily as 4 bytes. |
|||
*/ |
|||
void writeInt(int value) throws IOException; |
|||
|
|||
/** |
|||
* Writes a signed 32 bit int value whose value is likely to be small and positive but may not be. The implementation may encode the value in a way that |
|||
* is more efficient for small positive values. |
|||
*/ |
|||
void writeSmallInt(int value) throws IOException; |
|||
|
|||
/** |
|||
* Writes a nullable signed 32 bit int value whose value is likely to be small and positive but may not be. |
|||
* |
|||
* @see #writeSmallInt(int) |
|||
*/ |
|||
void writeNullableSmallInt(@Nullable Integer value) throws IOException; |
|||
|
|||
/** |
|||
* Writes a boolean value. |
|||
*/ |
|||
void writeBoolean(boolean value) throws IOException; |
|||
|
|||
/** |
|||
* Writes a non-null string value. |
|||
*/ |
|||
void writeString(CharSequence value) throws IOException; |
|||
|
|||
/** |
|||
* Writes a nullable string value. |
|||
*/ |
|||
void writeNullableString(@Nullable CharSequence value) throws IOException; |
|||
|
|||
interface EncodeAction<T> { |
|||
void write(T target) throws Exception; |
|||
} |
|||
} |
|||
@ -0,0 +1,31 @@ |
|||
/* |
|||
* Copyright 2013 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import java.io.Flushable; |
|||
import java.io.IOException; |
|||
|
|||
/** |
|||
* Represents an {@link Encoder} that buffers encoded data prior to writing to the backing stream. |
|||
*/ |
|||
public interface FlushableEncoder extends Encoder, Flushable { |
|||
/** |
|||
* Ensures that all buffered data has been written to the backing stream. Does not flush the backing stream. |
|||
*/ |
|||
@Override |
|||
void flush() throws IOException; |
|||
} |
|||
@ -0,0 +1,28 @@ |
|||
/* |
|||
* Copyright 2012 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import java.io.EOFException; |
|||
|
|||
public interface ObjectReader<T> { |
|||
/** |
|||
* Reads the next object from the stream. |
|||
* |
|||
* @throws EOFException When the next object cannot be fully read due to reaching the end of stream. |
|||
*/ |
|||
T read() throws EOFException, Exception; |
|||
} |
|||
@ -0,0 +1,21 @@ |
|||
/* |
|||
* Copyright 2012 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
public interface ObjectWriter<T> { |
|||
void write(T value) throws Exception; |
|||
} |
|||
@ -0,0 +1,33 @@ |
|||
/* |
|||
* Copyright 2009 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
import java.io.EOFException; |
|||
|
|||
public interface Serializer<T> { |
|||
/** |
|||
* Reads the next object from the given stream. The implementation must not perform any buffering, so that it reads only those bytes from the input stream that are |
|||
* required to deserialize the next object. |
|||
* |
|||
* @throws EOFException When the next object cannot be fully read due to reaching the end of stream. |
|||
*/ |
|||
T read(Decoder decoder) throws EOFException, Exception; |
|||
|
|||
/** |
|||
* Writes the given object to the given stream. The implementation must not perform any buffering. |
|||
*/ |
|||
void write(Encoder encoder, T value) throws Exception; |
|||
} |
|||
@ -0,0 +1,33 @@ |
|||
/* |
|||
* Copyright 2012 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize; |
|||
|
|||
/** |
|||
* Implementations must allow concurrent reading and writing, so that a thread can read and a thread can write at the same time. |
|||
* Implementations do not need to support multiple read threads or multiple write threads. |
|||
*/ |
|||
public interface StatefulSerializer<T> { |
|||
/** |
|||
* Should not perform any buffering |
|||
*/ |
|||
ObjectReader<T> newReader(Decoder decoder); |
|||
|
|||
/** |
|||
* Should not perform any buffering |
|||
*/ |
|||
ObjectWriter<T> newWriter(Encoder encoder); |
|||
} |
|||
@ -0,0 +1,210 @@ |
|||
/* |
|||
* Copyright 2013 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize.kryo; |
|||
|
|||
import com.esotericsoftware.kryo.KryoException; |
|||
import com.esotericsoftware.kryo.io.Input; |
|||
import seaweedfs.client.btree.serialize.AbstractDecoder; |
|||
import seaweedfs.client.btree.serialize.Decoder; |
|||
|
|||
import java.io.Closeable; |
|||
import java.io.EOFException; |
|||
import java.io.IOException; |
|||
import java.io.InputStream; |
|||
|
|||
/** |
|||
* Note that this decoder uses buffering, so will attempt to read beyond the end of the encoded data. This means you should use this type only when this decoder will be used to decode the entire |
|||
* stream. |
|||
*/ |
|||
public class KryoBackedDecoder extends AbstractDecoder implements Decoder, Closeable { |
|||
private final Input input; |
|||
private final InputStream inputStream; |
|||
private long extraSkipped; |
|||
private KryoBackedDecoder nested; |
|||
|
|||
public KryoBackedDecoder(InputStream inputStream) { |
|||
this(inputStream, 4096); |
|||
} |
|||
|
|||
public KryoBackedDecoder(InputStream inputStream, int bufferSize) { |
|||
this.inputStream = inputStream; |
|||
input = new Input(this.inputStream, bufferSize); |
|||
} |
|||
|
|||
@Override |
|||
protected int maybeReadBytes(byte[] buffer, int offset, int count) { |
|||
return input.read(buffer, offset, count); |
|||
} |
|||
|
|||
@Override |
|||
protected long maybeSkip(long count) throws IOException { |
|||
// Work around some bugs in Input.skip() |
|||
int remaining = input.limit() - input.position(); |
|||
if (remaining == 0) { |
|||
long skipped = inputStream.skip(count); |
|||
if (skipped > 0) { |
|||
extraSkipped += skipped; |
|||
} |
|||
return skipped; |
|||
} else if (count <= remaining) { |
|||
input.setPosition(input.position() + (int) count); |
|||
return count; |
|||
} else { |
|||
input.setPosition(input.limit()); |
|||
return remaining; |
|||
} |
|||
} |
|||
|
|||
private RuntimeException maybeEndOfStream(KryoException e) throws EOFException { |
|||
if (e.getMessage().equals("Buffer underflow.")) { |
|||
throw (EOFException) (new EOFException().initCause(e)); |
|||
} |
|||
throw e; |
|||
} |
|||
|
|||
@Override |
|||
public byte readByte() throws EOFException { |
|||
try { |
|||
return input.readByte(); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void readBytes(byte[] buffer, int offset, int count) throws EOFException { |
|||
try { |
|||
input.readBytes(buffer, offset, count); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public long readLong() throws EOFException { |
|||
try { |
|||
return input.readLong(); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public long readSmallLong() throws EOFException, IOException { |
|||
try { |
|||
return input.readLong(true); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public int readInt() throws EOFException { |
|||
try { |
|||
return input.readInt(); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public int readSmallInt() throws EOFException { |
|||
try { |
|||
return input.readInt(true); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public boolean readBoolean() throws EOFException { |
|||
try { |
|||
return input.readBoolean(); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public String readString() throws EOFException { |
|||
return readNullableString(); |
|||
} |
|||
|
|||
@Override |
|||
public String readNullableString() throws EOFException { |
|||
try { |
|||
return input.readString(); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void skipChunked() throws EOFException, IOException { |
|||
while (true) { |
|||
int count = readSmallInt(); |
|||
if (count == 0) { |
|||
break; |
|||
} |
|||
skipBytes(count); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public <T> T decodeChunked(DecodeAction<Decoder, T> decodeAction) throws EOFException, Exception { |
|||
if (nested == null) { |
|||
nested = new KryoBackedDecoder(new InputStream() { |
|||
@Override |
|||
public int read() throws IOException { |
|||
throw new UnsupportedOperationException(); |
|||
} |
|||
|
|||
@Override |
|||
public int read(byte[] buffer, int offset, int length) throws IOException { |
|||
int count = readSmallInt(); |
|||
if (count == 0) { |
|||
// End of stream has been reached |
|||
return -1; |
|||
} |
|||
if (count > length) { |
|||
// For now, assume same size buffers used to read and write |
|||
throw new UnsupportedOperationException(); |
|||
} |
|||
readBytes(buffer, offset, count); |
|||
return count; |
|||
} |
|||
}); |
|||
} |
|||
T value = decodeAction.read(nested); |
|||
if (readSmallInt() != 0) { |
|||
throw new IllegalStateException("Expecting the end of nested stream."); |
|||
} |
|||
return value; |
|||
} |
|||
|
|||
/** |
|||
* Returns the total number of bytes consumed by this decoder. Some additional bytes may also be buffered by this decoder but have not been consumed. |
|||
*/ |
|||
public long getReadPosition() { |
|||
return input.total() + extraSkipped; |
|||
} |
|||
|
|||
@Override |
|||
public void close() throws IOException { |
|||
input.close(); |
|||
} |
|||
} |
|||
@ -0,0 +1,134 @@ |
|||
/* |
|||
* Copyright 2013 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize.kryo; |
|||
|
|||
import com.esotericsoftware.kryo.io.Output; |
|||
import seaweedfs.client.btree.serialize.AbstractEncoder; |
|||
import seaweedfs.client.btree.serialize.Encoder; |
|||
import seaweedfs.client.btree.serialize.FlushableEncoder; |
|||
|
|||
import javax.annotation.Nullable; |
|||
import java.io.Closeable; |
|||
import java.io.IOException; |
|||
import java.io.OutputStream; |
|||
|
|||
public class KryoBackedEncoder extends AbstractEncoder implements FlushableEncoder, Closeable { |
|||
private final Output output; |
|||
private KryoBackedEncoder nested; |
|||
|
|||
public KryoBackedEncoder(OutputStream outputStream) { |
|||
this(outputStream, 4096); |
|||
} |
|||
|
|||
public KryoBackedEncoder(OutputStream outputStream, int bufferSize) { |
|||
output = new Output(outputStream, bufferSize); |
|||
} |
|||
|
|||
@Override |
|||
public void writeByte(byte value) { |
|||
output.writeByte(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeBytes(byte[] bytes, int offset, int count) { |
|||
output.writeBytes(bytes, offset, count); |
|||
} |
|||
|
|||
@Override |
|||
public void writeLong(long value) { |
|||
output.writeLong(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeSmallLong(long value) { |
|||
output.writeLong(value, true); |
|||
} |
|||
|
|||
@Override |
|||
public void writeInt(int value) { |
|||
output.writeInt(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeSmallInt(int value) { |
|||
output.writeInt(value, true); |
|||
} |
|||
|
|||
@Override |
|||
public void writeBoolean(boolean value) { |
|||
output.writeBoolean(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeString(CharSequence value) { |
|||
if (value == null) { |
|||
throw new IllegalArgumentException("Cannot encode a null string."); |
|||
} |
|||
output.writeString(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeNullableString(@Nullable CharSequence value) { |
|||
output.writeString(value); |
|||
} |
|||
|
|||
@Override |
|||
public void encodeChunked(EncodeAction<Encoder> writeAction) throws Exception { |
|||
if (nested == null) { |
|||
nested = new KryoBackedEncoder(new OutputStream() { |
|||
@Override |
|||
public void write(byte[] buffer, int offset, int length) { |
|||
if (length == 0) { |
|||
return; |
|||
} |
|||
writeSmallInt(length); |
|||
writeBytes(buffer, offset, length); |
|||
} |
|||
|
|||
@Override |
|||
public void write(byte[] buffer) throws IOException { |
|||
write(buffer, 0, buffer.length); |
|||
} |
|||
|
|||
@Override |
|||
public void write(int b) { |
|||
throw new UnsupportedOperationException(); |
|||
} |
|||
}); |
|||
} |
|||
writeAction.write(nested); |
|||
nested.flush(); |
|||
writeSmallInt(0); |
|||
} |
|||
|
|||
/** |
|||
* Returns the total number of bytes written by this encoder, some of which may still be buffered. |
|||
*/ |
|||
public long getWritePosition() { |
|||
return output.total(); |
|||
} |
|||
|
|||
@Override |
|||
public void flush() { |
|||
output.flush(); |
|||
} |
|||
|
|||
@Override |
|||
public void close() { |
|||
output.close(); |
|||
} |
|||
} |
|||
@ -0,0 +1,188 @@ |
|||
/* |
|||
* Copyright 2018 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize.kryo; |
|||
|
|||
import com.esotericsoftware.kryo.KryoException; |
|||
import com.esotericsoftware.kryo.io.Input; |
|||
import seaweedfs.client.btree.serialize.AbstractDecoder; |
|||
import seaweedfs.client.btree.serialize.Decoder; |
|||
|
|||
import java.io.Closeable; |
|||
import java.io.EOFException; |
|||
import java.io.IOException; |
|||
import java.io.InputStream; |
|||
|
|||
/** |
|||
* Note that this decoder uses buffering, so will attempt to read beyond the end of the encoded data. This means you should use this type only when this decoder will be used to decode the entire |
|||
* stream. |
|||
*/ |
|||
public class StringDeduplicatingKryoBackedDecoder extends AbstractDecoder implements Decoder, Closeable { |
|||
public static final int INITIAL_CAPACITY = 32; |
|||
private final Input input; |
|||
private final InputStream inputStream; |
|||
private String[] strings; |
|||
private long extraSkipped; |
|||
|
|||
public StringDeduplicatingKryoBackedDecoder(InputStream inputStream) { |
|||
this(inputStream, 4096); |
|||
} |
|||
|
|||
public StringDeduplicatingKryoBackedDecoder(InputStream inputStream, int bufferSize) { |
|||
this.inputStream = inputStream; |
|||
input = new Input(this.inputStream, bufferSize); |
|||
} |
|||
|
|||
@Override |
|||
protected int maybeReadBytes(byte[] buffer, int offset, int count) { |
|||
return input.read(buffer, offset, count); |
|||
} |
|||
|
|||
@Override |
|||
protected long maybeSkip(long count) throws IOException { |
|||
// Work around some bugs in Input.skip() |
|||
int remaining = input.limit() - input.position(); |
|||
if (remaining == 0) { |
|||
long skipped = inputStream.skip(count); |
|||
if (skipped > 0) { |
|||
extraSkipped += skipped; |
|||
} |
|||
return skipped; |
|||
} else if (count <= remaining) { |
|||
input.setPosition(input.position() + (int) count); |
|||
return count; |
|||
} else { |
|||
input.setPosition(input.limit()); |
|||
return remaining; |
|||
} |
|||
} |
|||
|
|||
private RuntimeException maybeEndOfStream(KryoException e) throws EOFException { |
|||
if (e.getMessage().equals("Buffer underflow.")) { |
|||
throw (EOFException) (new EOFException().initCause(e)); |
|||
} |
|||
throw e; |
|||
} |
|||
|
|||
@Override |
|||
public byte readByte() throws EOFException { |
|||
try { |
|||
return input.readByte(); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public void readBytes(byte[] buffer, int offset, int count) throws EOFException { |
|||
try { |
|||
input.readBytes(buffer, offset, count); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public long readLong() throws EOFException { |
|||
try { |
|||
return input.readLong(); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public long readSmallLong() throws EOFException, IOException { |
|||
try { |
|||
return input.readLong(true); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public int readInt() throws EOFException { |
|||
try { |
|||
return input.readInt(); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public int readSmallInt() throws EOFException { |
|||
try { |
|||
return input.readInt(true); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public boolean readBoolean() throws EOFException { |
|||
try { |
|||
return input.readBoolean(); |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public String readString() throws EOFException { |
|||
return readNullableString(); |
|||
} |
|||
|
|||
@Override |
|||
public String readNullableString() throws EOFException { |
|||
try { |
|||
int idx = readInt(); |
|||
if (idx == -1) { |
|||
return null; |
|||
} |
|||
if (strings == null) { |
|||
strings = new String[INITIAL_CAPACITY]; |
|||
} |
|||
String string = null; |
|||
if (idx >= strings.length) { |
|||
String[] grow = new String[strings.length * 3 / 2]; |
|||
System.arraycopy(strings, 0, grow, 0, strings.length); |
|||
strings = grow; |
|||
} else { |
|||
string = strings[idx]; |
|||
} |
|||
if (string == null) { |
|||
string = input.readString(); |
|||
strings[idx] = string; |
|||
} |
|||
return string; |
|||
} catch (KryoException e) { |
|||
throw maybeEndOfStream(e); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* Returns the total number of bytes consumed by this decoder. Some additional bytes may also be buffered by this decoder but have not been consumed. |
|||
*/ |
|||
public long getReadPosition() { |
|||
return input.total() + extraSkipped; |
|||
} |
|||
|
|||
@Override |
|||
public void close() throws IOException { |
|||
strings = null; |
|||
input.close(); |
|||
} |
|||
} |
|||
@ -0,0 +1,128 @@ |
|||
/* |
|||
* Copyright 2018 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize.kryo; |
|||
|
|||
import com.esotericsoftware.kryo.io.Output; |
|||
import com.google.common.collect.Maps; |
|||
import seaweedfs.client.btree.serialize.AbstractEncoder; |
|||
import seaweedfs.client.btree.serialize.FlushableEncoder; |
|||
|
|||
import javax.annotation.Nullable; |
|||
import java.io.Closeable; |
|||
import java.io.OutputStream; |
|||
import java.util.Map; |
|||
|
|||
public class StringDeduplicatingKryoBackedEncoder extends AbstractEncoder implements FlushableEncoder, Closeable { |
|||
private Map<String, Integer> strings; |
|||
|
|||
private final Output output; |
|||
|
|||
public StringDeduplicatingKryoBackedEncoder(OutputStream outputStream) { |
|||
this(outputStream, 4096); |
|||
} |
|||
|
|||
public StringDeduplicatingKryoBackedEncoder(OutputStream outputStream, int bufferSize) { |
|||
output = new Output(outputStream, bufferSize); |
|||
} |
|||
|
|||
@Override |
|||
public void writeByte(byte value) { |
|||
output.writeByte(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeBytes(byte[] bytes, int offset, int count) { |
|||
output.writeBytes(bytes, offset, count); |
|||
} |
|||
|
|||
@Override |
|||
public void writeLong(long value) { |
|||
output.writeLong(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeSmallLong(long value) { |
|||
output.writeLong(value, true); |
|||
} |
|||
|
|||
@Override |
|||
public void writeInt(int value) { |
|||
output.writeInt(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeSmallInt(int value) { |
|||
output.writeInt(value, true); |
|||
} |
|||
|
|||
@Override |
|||
public void writeBoolean(boolean value) { |
|||
output.writeBoolean(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeString(CharSequence value) { |
|||
if (value == null) { |
|||
throw new IllegalArgumentException("Cannot encode a null string."); |
|||
} |
|||
writeNullableString(value); |
|||
} |
|||
|
|||
@Override |
|||
public void writeNullableString(@Nullable CharSequence value) { |
|||
if (value == null) { |
|||
output.writeInt(-1); |
|||
return; |
|||
} else { |
|||
if (strings == null) { |
|||
strings = Maps.newHashMapWithExpectedSize(1024); |
|||
} |
|||
} |
|||
String key = value.toString(); |
|||
Integer index = strings.get(key); |
|||
if (index == null) { |
|||
index = strings.size(); |
|||
output.writeInt(index); |
|||
strings.put(key, index); |
|||
output.writeString(key); |
|||
} else { |
|||
output.writeInt(index); |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* Returns the total number of bytes written by this encoder, some of which may still be buffered. |
|||
*/ |
|||
public long getWritePosition() { |
|||
return output.total(); |
|||
} |
|||
|
|||
@Override |
|||
public void flush() { |
|||
output.flush(); |
|||
} |
|||
|
|||
@Override |
|||
public void close() { |
|||
output.close(); |
|||
} |
|||
|
|||
public void done() { |
|||
strings = null; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,51 @@ |
|||
/* |
|||
* Copyright 2012 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
|
|||
package seaweedfs.client.btree.serialize.kryo; |
|||
|
|||
import seaweedfs.client.btree.serialize.*; |
|||
|
|||
public class TypeSafeSerializer<T> implements StatefulSerializer<Object> { |
|||
private final Class<T> type; |
|||
private final StatefulSerializer<T> serializer; |
|||
|
|||
public TypeSafeSerializer(Class<T> type, StatefulSerializer<T> serializer) { |
|||
this.type = type; |
|||
this.serializer = serializer; |
|||
} |
|||
|
|||
@Override |
|||
public ObjectReader<Object> newReader(Decoder decoder) { |
|||
final ObjectReader<T> reader = serializer.newReader(decoder); |
|||
return new ObjectReader<Object>() { |
|||
@Override |
|||
public Object read() throws Exception { |
|||
return reader.read(); |
|||
} |
|||
}; |
|||
} |
|||
|
|||
@Override |
|||
public ObjectWriter<Object> newWriter(Encoder encoder) { |
|||
final ObjectWriter<T> writer = serializer.newWriter(encoder); |
|||
return new ObjectWriter<Object>() { |
|||
@Override |
|||
public void write(Object value) throws Exception { |
|||
writer.write(type.cast(value)); |
|||
} |
|||
}; |
|||
} |
|||
} |
|||
@ -0,0 +1,476 @@ |
|||
/* |
|||
* Copyright 2010 the original author or authors. |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0 |
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package seaweedfs.client.btree; |
|||
|
|||
import seaweedfs.client.btree.serialize.DefaultSerializer; |
|||
import seaweedfs.client.btree.serialize.Serializer; |
|||
import org.junit.Before; |
|||
import org.junit.Rule; |
|||
import org.junit.Test; |
|||
|
|||
import java.io.File; |
|||
import java.io.IOException; |
|||
import java.io.RandomAccessFile; |
|||
import java.util.ArrayList; |
|||
import java.util.Arrays; |
|||
import java.util.Collection; |
|||
import java.util.Collections; |
|||
import java.util.Comparator; |
|||
import java.util.LinkedHashMap; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
import static org.hamcrest.CoreMatchers.*; |
|||
import static org.junit.Assert.assertNull; |
|||
import static org.hamcrest.MatcherAssert.assertThat; |
|||
import static org.junit.Assert.assertTrue; |
|||
|
|||
public class BTreePersistentIndexedCacheTest { |
|||
private final Serializer<String> stringSerializer = new DefaultSerializer<String>(); |
|||
private final Serializer<Integer> integerSerializer = new DefaultSerializer<Integer>(); |
|||
private BTreePersistentIndexedCache<String, Integer> cache; |
|||
private File cacheFile; |
|||
|
|||
@Before |
|||
public void setup() { |
|||
cacheFile = tmpDirFile("cache.bin"); |
|||
} |
|||
|
|||
public File tmpDirFile(String filename) { |
|||
File f = new File("/Users/chris/tmp/mm/dev/btree_test"); |
|||
// File f = new File("/tmp/btree_test"); |
|||
f.mkdirs(); |
|||
return new File(f, filename); |
|||
} |
|||
|
|||
private void createCache() { |
|||
cache = new BTreePersistentIndexedCache<String, Integer>(cacheFile, stringSerializer, integerSerializer, (short) 4, 100); |
|||
} |
|||
|
|||
private void verifyAndCloseCache() { |
|||
cache.verify(); |
|||
cache.close(); |
|||
} |
|||
|
|||
@Test |
|||
public void getReturnsNullWhenEntryDoesNotExist() { |
|||
createCache(); |
|||
assertNull(cache.get("unknown")); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void persistsAddedEntries() { |
|||
createCache(); |
|||
checkAdds(1, 2, 3, 4, 5); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void persistsAddedEntriesInReverseOrder() { |
|||
createCache(); |
|||
checkAdds(5, 4, 3, 2, 1); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void persistsAddedEntriesOverMultipleIndexBlocks() { |
|||
createCache(); |
|||
checkAdds(3, 2, 11, 5, 7, 1, 10, 8, 9, 4, 6, 0); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void persistsUpdates() { |
|||
createCache(); |
|||
checkUpdates(3, 2, 11, 5, 7, 1, 10, 8, 9, 4, 6, 0); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void handlesUpdatesWhenBlockSizeDecreases() { |
|||
BTreePersistentIndexedCache<String, List<Integer>> cache = |
|||
new BTreePersistentIndexedCache<String, List<Integer>>( |
|||
tmpDirFile("listcache.bin"), stringSerializer, |
|||
new DefaultSerializer<List<Integer>>(), (short) 4, 100); |
|||
|
|||
List<Integer> values = Arrays.asList(3, 2, 11, 5, 7, 1, 10, 8, 9, 4, 6, 0); |
|||
Map<Integer, List<Integer>> updated = new LinkedHashMap<Integer, List<Integer>>(); |
|||
|
|||
for (int i = 10; i > 0; i--) { |
|||
for (Integer value : values) { |
|||
String key = String.format("key_%d", value); |
|||
List<Integer> newValue = new ArrayList<Integer>(i); |
|||
for (int j = 0; j < i * 2; j++) { |
|||
newValue.add(j); |
|||
} |
|||
cache.put(key, newValue); |
|||
updated.put(value, newValue); |
|||
} |
|||
|
|||
checkListEntries(cache, updated); |
|||
} |
|||
|
|||
cache.reset(); |
|||
|
|||
checkListEntries(cache, updated); |
|||
|
|||
cache.verify(); |
|||
cache.close(); |
|||
} |
|||
|
|||
private void checkListEntries(BTreePersistentIndexedCache<String, List<Integer>> cache, Map<Integer, List<Integer>> updated) { |
|||
for (Map.Entry<Integer, List<Integer>> entry : updated.entrySet()) { |
|||
String key = String.format("key_%d", entry.getKey()); |
|||
assertThat(cache.get(key), equalTo(entry.getValue())); |
|||
} |
|||
} |
|||
|
|||
@Test |
|||
public void handlesUpdatesWhenBlockSizeIncreases() { |
|||
BTreePersistentIndexedCache<String, List<Integer>> cache = |
|||
new BTreePersistentIndexedCache<String, List<Integer>>( |
|||
tmpDirFile("listcache.bin"), stringSerializer, |
|||
new DefaultSerializer<List<Integer>>(), (short) 4, 100); |
|||
|
|||
List<Integer> values = Arrays.asList(3, 2, 11, 5, 7, 1, 10, 8, 9, 4, 6, 0); |
|||
Map<Integer, List<Integer>> updated = new LinkedHashMap<Integer, List<Integer>>(); |
|||
|
|||
for (int i = 1; i < 10; i++) { |
|||
for (Integer value : values) { |
|||
String key = String.format("key_%d", value); |
|||
List<Integer> newValue = new ArrayList<Integer>(i); |
|||
for (int j = 0; j < i * 2; j++) { |
|||
newValue.add(j); |
|||
} |
|||
cache.put(key, newValue); |
|||
updated.put(value, newValue); |
|||
} |
|||
|
|||
checkListEntries(cache, updated); |
|||
} |
|||
|
|||
cache.reset(); |
|||
|
|||
checkListEntries(cache, updated); |
|||
|
|||
cache.verify(); |
|||
cache.close(); |
|||
} |
|||
|
|||
@Test |
|||
public void persistsAddedEntriesAfterReopen() { |
|||
createCache(); |
|||
|
|||
checkAdds(1, 2, 3, 4); |
|||
|
|||
cache.reset(); |
|||
|
|||
checkAdds(5, 6, 7, 8); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void persistsReplacedEntries() { |
|||
createCache(); |
|||
|
|||
cache.put("key_1", 1); |
|||
cache.put("key_2", 2); |
|||
cache.put("key_3", 3); |
|||
cache.put("key_4", 4); |
|||
cache.put("key_5", 5); |
|||
|
|||
cache.put("key_1", 1); |
|||
cache.put("key_4", 12); |
|||
|
|||
assertThat(cache.get("key_1"), equalTo(1)); |
|||
assertThat(cache.get("key_2"), equalTo(2)); |
|||
assertThat(cache.get("key_3"), equalTo(3)); |
|||
assertThat(cache.get("key_4"), equalTo(12)); |
|||
assertThat(cache.get("key_5"), equalTo(5)); |
|||
|
|||
cache.reset(); |
|||
|
|||
assertThat(cache.get("key_1"), equalTo(1)); |
|||
assertThat(cache.get("key_2"), equalTo(2)); |
|||
assertThat(cache.get("key_3"), equalTo(3)); |
|||
assertThat(cache.get("key_4"), equalTo(12)); |
|||
assertThat(cache.get("key_5"), equalTo(5)); |
|||
|
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void reusesEmptySpaceWhenPuttingEntries() { |
|||
BTreePersistentIndexedCache<String, String> cache = new BTreePersistentIndexedCache<String, String>(cacheFile, stringSerializer, stringSerializer, (short) 4, 100); |
|||
|
|||
long beforeLen = cacheFile.length(); |
|||
if (beforeLen>0){ |
|||
System.out.println(String.format("cache %s: %s", "key_new", cache.get("key_new"))); |
|||
} |
|||
|
|||
cache.put("key_1", "abcd"); |
|||
cache.put("key_2", "abcd"); |
|||
cache.put("key_3", "abcd"); |
|||
cache.put("key_4", "abcd"); |
|||
cache.put("key_5", "abcd"); |
|||
|
|||
long len = cacheFile.length(); |
|||
assertTrue(len > 0L); |
|||
|
|||
System.out.println(String.format("cache file size %d => %d", beforeLen, len)); |
|||
|
|||
cache.put("key_1", "1234"); |
|||
assertThat(cacheFile.length(), equalTo(len)); |
|||
|
|||
cache.remove("key_1"); |
|||
cache.put("key_new", "a1b2"); |
|||
assertThat(cacheFile.length(), equalTo(len)); |
|||
|
|||
cache.put("key_new", "longer value assertThat(cacheFile.length(), equalTo(len))"); |
|||
System.out.println(String.format("cache file size %d beforeLen %d", cacheFile.length(), len)); |
|||
// assertTrue(cacheFile.length() > len); |
|||
len = cacheFile.length(); |
|||
|
|||
cache.put("key_1", "1234"); |
|||
assertThat(cacheFile.length(), equalTo(len)); |
|||
|
|||
cache.close(); |
|||
} |
|||
|
|||
@Test |
|||
public void canHandleLargeNumberOfEntries() { |
|||
createCache(); |
|||
int count = 2000; |
|||
List<Integer> values = new ArrayList<Integer>(); |
|||
for (int i = 0; i < count; i++) { |
|||
values.add(i); |
|||
} |
|||
|
|||
checkAddsAndRemoves(null, values); |
|||
|
|||
long len = cacheFile.length(); |
|||
|
|||
checkAddsAndRemoves(Collections.reverseOrder(), values); |
|||
|
|||
// need to make this better |
|||
assertTrue(cacheFile.length() < (long)(1.4 * len)); |
|||
|
|||
checkAdds(values); |
|||
|
|||
// need to make this better |
|||
assertTrue(cacheFile.length() < (long) (1.4 * 1.4 * len)); |
|||
|
|||
cache.close(); |
|||
} |
|||
|
|||
@Test |
|||
public void persistsRemovalOfEntries() { |
|||
createCache(); |
|||
checkAddsAndRemoves(1, 2, 3, 4, 5); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void persistsRemovalOfEntriesInReverse() { |
|||
createCache(); |
|||
checkAddsAndRemoves(Collections.<Integer>reverseOrder(), 1, 2, 3, 4, 5); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void persistsRemovalOfEntriesOverMultipleIndexBlocks() { |
|||
createCache(); |
|||
checkAddsAndRemoves(4, 12, 9, 1, 3, 10, 11, 7, 8, 2, 5, 6); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void removalRedistributesRemainingEntriesWithLeftSibling() { |
|||
createCache(); |
|||
// Ends up with: 1 2 3 -> 4 <- 5 6 |
|||
checkAdds(1, 2, 5, 6, 4, 3); |
|||
cache.verify(); |
|||
cache.remove("key_5"); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void removalMergesRemainingEntriesIntoLeftSibling() { |
|||
createCache(); |
|||
// Ends up with: 1 2 -> 3 <- 4 5 |
|||
checkAdds(1, 2, 4, 5, 3); |
|||
cache.verify(); |
|||
cache.remove("key_4"); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void removalRedistributesRemainingEntriesWithRightSibling() { |
|||
createCache(); |
|||
// Ends up with: 1 2 -> 3 <- 4 5 6 |
|||
checkAdds(1, 2, 4, 5, 3, 6); |
|||
cache.verify(); |
|||
cache.remove("key_2"); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void removalMergesRemainingEntriesIntoRightSibling() { |
|||
createCache(); |
|||
// Ends up with: 1 2 -> 3 <- 4 5 |
|||
checkAdds(1, 2, 4, 5, 3); |
|||
cache.verify(); |
|||
cache.remove("key_2"); |
|||
verifyAndCloseCache(); |
|||
} |
|||
|
|||
@Test |
|||
public void handlesOpeningATruncatedCacheFile() throws IOException { |
|||
BTreePersistentIndexedCache<String, Integer> cache = new BTreePersistentIndexedCache<String, Integer>(cacheFile, stringSerializer, integerSerializer); |
|||
|
|||
assertNull(cache.get("key_1")); |
|||
cache.put("key_1", 99); |
|||
|
|||
RandomAccessFile file = new RandomAccessFile(cacheFile, "rw"); |
|||
file.setLength(file.length() - 10); |
|||
file.close(); |
|||
|
|||
cache.reset(); |
|||
|
|||
assertNull(cache.get("key_1")); |
|||
cache.verify(); |
|||
|
|||
cache.close(); |
|||
} |
|||
|
|||
@Test |
|||
public void canUseFileAsKey() { |
|||
BTreePersistentIndexedCache<File, Integer> cache = new BTreePersistentIndexedCache<File, Integer>(cacheFile, new DefaultSerializer<File>(), integerSerializer); |
|||
|
|||
cache.put(new File("file"), 1); |
|||
cache.put(new File("dir/file"), 2); |
|||
cache.put(new File("File"), 3); |
|||
|
|||
assertThat(cache.get(new File("file")), equalTo(1)); |
|||
assertThat(cache.get(new File("dir/file")), equalTo(2)); |
|||
assertThat(cache.get(new File("File")), equalTo(3)); |
|||
|
|||
cache.close(); |
|||
} |
|||
|
|||
@Test |
|||
public void handlesKeysWithSameHashCode() { |
|||
createCache(); |
|||
|
|||
String key1 = new String(new byte[]{2, 31}); |
|||
String key2 = new String(new byte[]{1, 62}); |
|||
cache.put(key1, 1); |
|||
cache.put(key2, 2); |
|||
|
|||
assertThat(cache.get(key1), equalTo(1)); |
|||
assertThat(cache.get(key2), equalTo(2)); |
|||
|
|||
cache.close(); |
|||
} |
|||
|
|||
private void checkAdds(Integer... values) { |
|||
checkAdds(Arrays.asList(values)); |
|||
} |
|||
|
|||
private Map<String, Integer> checkAdds(Iterable<Integer> values) { |
|||
Map<String, Integer> added = new LinkedHashMap<String, Integer>(); |
|||
|
|||
for (Integer value : values) { |
|||
String key = String.format("key_%d", value); |
|||
cache.put(key, value); |
|||
added.put(String.format("key_%d", value), value); |
|||
} |
|||
|
|||
for (Map.Entry<String, Integer> entry : added.entrySet()) { |
|||
assertThat(cache.get(entry.getKey()), equalTo(entry.getValue())); |
|||
} |
|||
|
|||
cache.reset(); |
|||
|
|||
for (Map.Entry<String, Integer> entry : added.entrySet()) { |
|||
assertThat(cache.get(entry.getKey()), equalTo(entry.getValue())); |
|||
} |
|||
|
|||
return added; |
|||
} |
|||
|
|||
private void checkUpdates(Integer... values) { |
|||
checkUpdates(Arrays.asList(values)); |
|||
} |
|||
|
|||
private Map<Integer, Integer> checkUpdates(Iterable<Integer> values) { |
|||
Map<Integer, Integer> updated = new LinkedHashMap<Integer, Integer>(); |
|||
|
|||
for (int i = 0; i < 10; i++) { |
|||
for (Integer value : values) { |
|||
String key = String.format("key_%d", value); |
|||
int newValue = value + (i * 100); |
|||
cache.put(key, newValue); |
|||
updated.put(value, newValue); |
|||
} |
|||
|
|||
for (Map.Entry<Integer, Integer> entry : updated.entrySet()) { |
|||
String key = String.format("key_%d", entry.getKey()); |
|||
assertThat(cache.get(key), equalTo(entry.getValue())); |
|||
} |
|||
} |
|||
|
|||
cache.reset(); |
|||
|
|||
for (Map.Entry<Integer, Integer> entry : updated.entrySet()) { |
|||
String key = String.format("key_%d", entry.getKey()); |
|||
assertThat(cache.get(key), equalTo(entry.getValue())); |
|||
} |
|||
|
|||
return updated; |
|||
} |
|||
|
|||
private void checkAddsAndRemoves(Integer... values) { |
|||
checkAddsAndRemoves(null, values); |
|||
} |
|||
|
|||
private void checkAddsAndRemoves(Comparator<Integer> comparator, Integer... values) { |
|||
checkAddsAndRemoves(comparator, Arrays.asList(values)); |
|||
} |
|||
|
|||
private void checkAddsAndRemoves(Comparator<Integer> comparator, Collection<Integer> values) { |
|||
checkAdds(values); |
|||
|
|||
List<Integer> deleteValues = new ArrayList<Integer>(values); |
|||
Collections.sort(deleteValues, comparator); |
|||
for (Integer value : deleteValues) { |
|||
String key = String.format("key_%d", value); |
|||
assertThat(cache.get(key), notNullValue()); |
|||
cache.remove(key); |
|||
assertThat(cache.get(key), nullValue()); |
|||
} |
|||
|
|||
cache.reset(); |
|||
cache.verify(); |
|||
|
|||
for (Integer value : deleteValues) { |
|||
String key = String.format("key_%d", value); |
|||
assertThat(cache.get(key), nullValue()); |
|||
} |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,143 @@ |
|||
package seaweedfs.file; |
|||
|
|||
import org.junit.Test; |
|||
|
|||
import java.io.File; |
|||
import java.io.FileOutputStream; |
|||
import java.io.IOException; |
|||
import java.io.RandomAccessFile; |
|||
import java.nio.MappedByteBuffer; |
|||
import java.nio.channels.FileChannel; |
|||
|
|||
public class MmapFileTest { |
|||
|
|||
static File dir = new File("/Users/chris/tmp/mm/dev"); |
|||
|
|||
@Test |
|||
public void testMmap() { |
|||
try { |
|||
System.out.println("starting ..."); |
|||
|
|||
File f = new File(dir, "mmap_file.txt"); |
|||
RandomAccessFile raf = new RandomAccessFile(f, "rw"); |
|||
FileChannel fc = raf.getChannel(); |
|||
MappedByteBuffer mbf = fc.map(FileChannel.MapMode.READ_ONLY, 0, fc.size()); |
|||
fc.close(); |
|||
raf.close(); |
|||
|
|||
FileOutputStream fos = new FileOutputStream(f); |
|||
fos.write("abcdefg".getBytes()); |
|||
fos.close(); |
|||
System.out.println("completed!"); |
|||
} catch (Exception e) { |
|||
e.printStackTrace(); |
|||
} |
|||
} |
|||
|
|||
@Test |
|||
public void testBigMmap() throws IOException { |
|||
/* |
|||
|
|||
// new file |
|||
I0817 09:48:02 25175 dir.go:147] create /dev/mmap_big.txt: OpenReadWrite+OpenCreate |
|||
I0817 09:48:02 25175 wfs.go:116] AcquireHandle /dev/mmap_big.txt uid=502 gid=20 |
|||
I0817 09:48:02 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 0 |
|||
I0817 09:48:02 25175 meta_cache_subscribe.go:32] creating /dev/mmap_big.txt |
|||
|
|||
//get channel |
|||
I0817 09:48:26 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 0 |
|||
|
|||
I0817 09:48:32 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 0 |
|||
I0817 09:48:32 25175 wfs.go:116] AcquireHandle /dev/mmap_big.txt uid=0 gid=0 |
|||
I0817 09:48:32 25175 filehandle.go:160] Release /dev/mmap_big.txt fh 14968871991130164560 |
|||
|
|||
//fileChannel.map |
|||
I0817 09:49:18 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 0 |
|||
I0817 09:49:18 25175 file.go:112] /dev/mmap_big.txt file setattr set size=262144 chunks=0 |
|||
I0817 09:49:18 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 262144 |
|||
I0817 09:49:18 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 262144 |
|||
I0817 09:49:18 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 262144 |
|||
|
|||
// buffer.put |
|||
I0817 09:49:49 25175 filehandle.go:57] /dev/mmap_big.txt read fh 14968871991130164560: [0,32768) size 32768 resp.Data len=0 cap=32768 |
|||
I0817 09:49:49 25175 reader_at.go:113] zero2 [0,32768) |
|||
I0817 09:49:50 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 262144 |
|||
|
|||
I0817 09:49:53 25175 file.go:233] /dev/mmap_big.txt fsync file Fsync [ID=0x4 Node=0xe Uid=0 Gid=0 Pid=0] Handle 0x2 Flags 1 |
|||
|
|||
//close |
|||
I0817 09:50:14 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 262144 |
|||
I0817 09:50:14 25175 dirty_page.go:130] saveToStorage /dev/mmap_big.txt 1,315b69812039e5 [0,4096) of 262144 bytes |
|||
I0817 09:50:14 25175 file.go:274] /dev/mmap_big.txt existing 0 chunks adds 1 more |
|||
I0817 09:50:14 25175 filehandle.go:218] /dev/mmap_big.txt set chunks: 1 |
|||
I0817 09:50:14 25175 filehandle.go:220] /dev/mmap_big.txt chunks 0: 1,315b69812039e5 [0,4096) |
|||
I0817 09:50:14 25175 meta_cache_subscribe.go:23] deleting /dev/mmap_big.txt |
|||
I0817 09:50:14 25175 meta_cache_subscribe.go:32] creating /dev/mmap_big.txt |
|||
|
|||
// end of test |
|||
I0817 09:50:41 25175 file.go:62] file Attr /dev/mmap_big.txt, open:1, size: 262144 |
|||
I0817 09:50:41 25175 filehandle.go:160] Release /dev/mmap_big.txt fh 14968871991130164560 |
|||
|
|||
*/ |
|||
// Create file object |
|||
File file = new File(dir, "mmap_big.txt"); |
|||
|
|||
try (RandomAccessFile randomAccessFile = new RandomAccessFile(file, "rw")) { |
|||
// Get file channel in read-write mode |
|||
FileChannel fileChannel = randomAccessFile.getChannel(); |
|||
|
|||
// Get direct byte buffer access using channel.map() operation |
|||
MappedByteBuffer buffer = fileChannel.map(FileChannel.MapMode.READ_WRITE, 0, 4096 * 8 * 8); |
|||
|
|||
//Write the content using put methods |
|||
buffer.put("howtodoinjava.com".getBytes()); |
|||
} |
|||
|
|||
/* |
|||
> meta.cat /dev/mmap_big.txt |
|||
{ |
|||
"name": "mmap_big.txt", |
|||
"isDirectory": false, |
|||
"chunks": [ |
|||
{ |
|||
"fileId": "1,315b69812039e5", |
|||
"offset": "0", |
|||
"size": "4096", |
|||
"mtime": "1597683014026365000", |
|||
"eTag": "985ab0ac", |
|||
"sourceFileId": "", |
|||
"fid": { |
|||
"volumeId": 1, |
|||
"fileKey": "3234665", |
|||
"cookie": 2166372837 |
|||
}, |
|||
"sourceFid": null, |
|||
"cipherKey": null, |
|||
"isCompressed": true, |
|||
"isChunkManifest": false |
|||
} |
|||
], |
|||
"attributes": { |
|||
"fileSize": "262144", |
|||
"mtime": "1597683014", |
|||
"fileMode": 420, |
|||
"uid": 502, |
|||
"gid": 20, |
|||
"crtime": "1597682882", |
|||
"mime": "application/octet-stream", |
|||
"replication": "", |
|||
"collection": "", |
|||
"ttlSec": 0, |
|||
"userName": "", |
|||
"groupName": [ |
|||
], |
|||
"symlinkTarget": "", |
|||
"md5": null |
|||
}, |
|||
"extended": { |
|||
} |
|||
} |
|||
*/ |
|||
|
|||
} |
|||
} |
|||
@ -0,0 +1,70 @@ |
|||
package seaweedfs.file; |
|||
|
|||
import org.junit.Assert; |
|||
import org.junit.Test; |
|||
|
|||
import java.io.File; |
|||
import java.io.IOException; |
|||
import java.io.RandomAccessFile; |
|||
import java.nio.ByteBuffer; |
|||
import java.util.Random; |
|||
|
|||
public class RandomeAccessFileTest { |
|||
|
|||
@Test |
|||
public void testRandomWriteAndRead() throws IOException { |
|||
|
|||
File f = new File(MmapFileTest.dir, "mmap_file.txt"); |
|||
|
|||
RandomAccessFile af = new RandomAccessFile(f, "rw"); |
|||
af.setLength(0); |
|||
af.close(); |
|||
|
|||
Random r = new Random(); |
|||
|
|||
int maxLength = 5000; |
|||
|
|||
byte[] data = new byte[maxLength]; |
|||
byte[] readData = new byte[maxLength]; |
|||
|
|||
for (int i = 4096; i < maxLength; i++) { |
|||
|
|||
RandomAccessFile raf = new RandomAccessFile(f, "rw"); |
|||
long fileSize = raf.length(); |
|||
|
|||
raf.readFully(readData, 0, (int)fileSize); |
|||
|
|||
for (int x=0;x<fileSize;x++){ |
|||
Assert.assertEquals(data[x], readData[x]); |
|||
} |
|||
|
|||
int start = r.nextInt(i); |
|||
int stop = r.nextInt(i); |
|||
if (start > stop) { |
|||
int t = stop; |
|||
stop = start; |
|||
start = t; |
|||
} |
|||
if (stop > fileSize) { |
|||
fileSize = stop; |
|||
raf.setLength(fileSize); |
|||
} |
|||
|
|||
randomize(r, data, start, stop); |
|||
raf.seek(start); |
|||
raf.write(data, start, stop-start); |
|||
|
|||
raf.close(); |
|||
} |
|||
|
|||
} |
|||
|
|||
private static void randomize(Random r, byte[] bytes, int start, int stop) { |
|||
for (int i = start; i < stop; i++) { |
|||
int rnd = r.nextInt(); |
|||
bytes[i] = (byte) rnd; |
|||
} |
|||
} |
|||
|
|||
|
|||
} |
|||
@ -0,0 +1,175 @@ |
|||
package main |
|||
|
|||
// copied from https://github.com/apoorvam/aws-s3-multipart-upload
|
|||
|
|||
import ( |
|||
"bytes" |
|||
"flag" |
|||
"fmt" |
|||
"net/http" |
|||
"os" |
|||
|
|||
"github.com/aws/aws-sdk-go/aws" |
|||
"github.com/aws/aws-sdk-go/aws/awserr" |
|||
"github.com/aws/aws-sdk-go/aws/credentials" |
|||
"github.com/aws/aws-sdk-go/aws/session" |
|||
"github.com/aws/aws-sdk-go/service/s3" |
|||
) |
|||
|
|||
const ( |
|||
maxPartSize = int64(5 * 1024 * 1024) |
|||
maxRetries = 3 |
|||
awsAccessKeyID = "Your access key" |
|||
awsSecretAccessKey = "Your secret key" |
|||
awsBucketRegion = "S3 bucket region" |
|||
awsBucketName = "newBucket" |
|||
) |
|||
|
|||
var ( |
|||
filename = flag.String("f", "", "the file name") |
|||
) |
|||
|
|||
func main() { |
|||
flag.Parse() |
|||
|
|||
creds := credentials.NewStaticCredentials(awsAccessKeyID, awsSecretAccessKey, "") |
|||
_, err := creds.Get() |
|||
if err != nil { |
|||
fmt.Printf("bad credentials: %s", err) |
|||
} |
|||
cfg := aws.NewConfig().WithRegion(awsBucketRegion).WithCredentials(creds).WithDisableSSL(true).WithEndpoint("localhost:8333") |
|||
svc := s3.New(session.New(), cfg) |
|||
|
|||
file, err := os.Open(*filename) |
|||
if err != nil { |
|||
fmt.Printf("err opening file: %s", err) |
|||
return |
|||
} |
|||
defer file.Close() |
|||
fileInfo, _ := file.Stat() |
|||
size := fileInfo.Size() |
|||
buffer := make([]byte, size) |
|||
fileType := http.DetectContentType(buffer) |
|||
file.Read(buffer) |
|||
|
|||
path := "/media/" + file.Name() |
|||
input := &s3.CreateMultipartUploadInput{ |
|||
Bucket: aws.String(awsBucketName), |
|||
Key: aws.String(path), |
|||
ContentType: aws.String(fileType), |
|||
} |
|||
|
|||
resp, err := svc.CreateMultipartUpload(input) |
|||
if err != nil { |
|||
fmt.Println(err.Error()) |
|||
return |
|||
} |
|||
fmt.Println("Created multipart upload request") |
|||
|
|||
var curr, partLength int64 |
|||
var remaining = size |
|||
var completedParts []*s3.CompletedPart |
|||
partNumber := 1 |
|||
for curr = 0; remaining != 0; curr += partLength { |
|||
if remaining < maxPartSize { |
|||
partLength = remaining |
|||
} else { |
|||
partLength = maxPartSize |
|||
} |
|||
completedPart, err := uploadPart(svc, resp, buffer[curr:curr+partLength], partNumber) |
|||
if err != nil { |
|||
fmt.Println(err.Error()) |
|||
err := abortMultipartUpload(svc, resp) |
|||
if err != nil { |
|||
fmt.Println(err.Error()) |
|||
} |
|||
return |
|||
} |
|||
remaining -= partLength |
|||
partNumber++ |
|||
completedParts = append(completedParts, completedPart) |
|||
} |
|||
|
|||
// list parts
|
|||
parts, err := svc.ListParts(&s3.ListPartsInput{ |
|||
Bucket: input.Bucket, |
|||
Key: input.Key, |
|||
MaxParts: nil, |
|||
PartNumberMarker: nil, |
|||
RequestPayer: nil, |
|||
UploadId: resp.UploadId, |
|||
}) |
|||
if err != nil { |
|||
fmt.Println(err.Error()) |
|||
return |
|||
} |
|||
fmt.Printf("list parts: %d\n", len(parts.Parts)) |
|||
for i, part := range parts.Parts { |
|||
fmt.Printf("part %d: %v\n", i, part) |
|||
} |
|||
|
|||
|
|||
completeResponse, err := completeMultipartUpload(svc, resp, completedParts) |
|||
if err != nil { |
|||
fmt.Println(err.Error()) |
|||
return |
|||
} |
|||
|
|||
fmt.Printf("Successfully uploaded file: %s\n", completeResponse.String()) |
|||
} |
|||
|
|||
func completeMultipartUpload(svc *s3.S3, resp *s3.CreateMultipartUploadOutput, completedParts []*s3.CompletedPart) (*s3.CompleteMultipartUploadOutput, error) { |
|||
completeInput := &s3.CompleteMultipartUploadInput{ |
|||
Bucket: resp.Bucket, |
|||
Key: resp.Key, |
|||
UploadId: resp.UploadId, |
|||
MultipartUpload: &s3.CompletedMultipartUpload{ |
|||
Parts: completedParts, |
|||
}, |
|||
} |
|||
return svc.CompleteMultipartUpload(completeInput) |
|||
} |
|||
|
|||
func uploadPart(svc *s3.S3, resp *s3.CreateMultipartUploadOutput, fileBytes []byte, partNumber int) (*s3.CompletedPart, error) { |
|||
tryNum := 1 |
|||
partInput := &s3.UploadPartInput{ |
|||
Body: bytes.NewReader(fileBytes), |
|||
Bucket: resp.Bucket, |
|||
Key: resp.Key, |
|||
PartNumber: aws.Int64(int64(partNumber)), |
|||
UploadId: resp.UploadId, |
|||
ContentLength: aws.Int64(int64(len(fileBytes))), |
|||
} |
|||
|
|||
for tryNum <= maxRetries { |
|||
uploadResult, err := svc.UploadPart(partInput) |
|||
if err != nil { |
|||
if tryNum == maxRetries { |
|||
if aerr, ok := err.(awserr.Error); ok { |
|||
return nil, aerr |
|||
} |
|||
return nil, err |
|||
} |
|||
fmt.Printf("Retrying to upload part #%v\n", partNumber) |
|||
tryNum++ |
|||
} else { |
|||
fmt.Printf("Uploaded part #%v\n", partNumber) |
|||
return &s3.CompletedPart{ |
|||
ETag: uploadResult.ETag, |
|||
PartNumber: aws.Int64(int64(partNumber)), |
|||
}, nil |
|||
} |
|||
} |
|||
return nil, nil |
|||
} |
|||
|
|||
func abortMultipartUpload(svc *s3.S3, resp *s3.CreateMultipartUploadOutput) error { |
|||
fmt.Println("Aborting multipart upload for UploadId#" + *resp.UploadId) |
|||
abortInput := &s3.AbortMultipartUploadInput{ |
|||
Bucket: resp.Bucket, |
|||
Key: resp.Key, |
|||
UploadId: resp.UploadId, |
|||
} |
|||
_, err := svc.AbortMultipartUpload(abortInput) |
|||
return err |
|||
} |
|||
@ -0,0 +1,73 @@ |
|||
package main |
|||
|
|||
import ( |
|||
"github.com/aws/aws-sdk-go/aws" |
|||
"github.com/aws/aws-sdk-go/aws/session" |
|||
"github.com/aws/aws-sdk-go/service/s3" |
|||
"encoding/base64" |
|||
"fmt" |
|||
"crypto/md5" |
|||
"strings" |
|||
"time" |
|||
"net/http" |
|||
) |
|||
|
|||
// Downloads an item from an S3 Bucket in the region configured in the shared config
|
|||
// or AWS_REGION environment variable.
|
|||
//
|
|||
// Usage:
|
|||
// go run presigned_put.go
|
|||
// For this exampl to work, the domainName is needd
|
|||
// weed s3 -domainName=localhost
|
|||
func main() { |
|||
h := md5.New() |
|||
content := strings.NewReader(stringContent) |
|||
content.WriteTo(h) |
|||
|
|||
// Initialize a session in us-west-2 that the SDK will use to load
|
|||
// credentials from the shared credentials file ~/.aws/credentials.
|
|||
sess, err := session.NewSession(&aws.Config{ |
|||
Region: aws.String("us-east-1"), |
|||
Endpoint: aws.String("http://localhost:8333"), |
|||
}) |
|||
|
|||
// Create S3 service client
|
|||
svc := s3.New(sess) |
|||
|
|||
putRequest, output := svc.PutObjectRequest(&s3.PutObjectInput{ |
|||
Bucket: aws.String("dev"), |
|||
Key: aws.String("testKey"), |
|||
}) |
|||
fmt.Printf("output: %+v\n", output) |
|||
|
|||
md5s := base64.StdEncoding.EncodeToString(h.Sum(nil)) |
|||
putRequest.HTTPRequest.Header.Set("Content-MD5", md5s) |
|||
|
|||
url, err := putRequest.Presign(15 * time.Minute) |
|||
if err != nil { |
|||
fmt.Println("error presigning request", err) |
|||
return |
|||
} |
|||
|
|||
fmt.Println(url) |
|||
|
|||
req, err := http.NewRequest("PUT", url, strings.NewReader(stringContent)) |
|||
req.Header.Set("Content-MD5", md5s) |
|||
if err != nil { |
|||
fmt.Println("error creating request", url) |
|||
return |
|||
} |
|||
|
|||
resp, err := http.DefaultClient.Do(req) |
|||
if err != nil { |
|||
fmt.Printf("error put request: %v\n", err) |
|||
return |
|||
} |
|||
fmt.Printf("response: %+v\n", resp) |
|||
} |
|||
|
|||
var stringContent = `Generate a Pre-Signed URL for an Amazon S3 PUT Operation with a Specific Payload |
|||
You can generate a pre-signed URL for a PUT operation that checks whether users upload the correct content. When the SDK pre-signs a request, it computes the checksum of the request body and generates an MD5 checksum that is included in the pre-signed URL. Users must upload the same content that produces the same MD5 checksum generated by the SDK; otherwise, the operation fails. This is not the Content-MD5, but the signature. To enforce Content-MD5, simply add the header to the request. |
|||
|
|||
The following example adds a Body field to generate a pre-signed PUT operation that requires a specific payload to be uploaded by users. |
|||
` |
|||
@ -1,83 +0,0 @@ |
|||
package main |
|||
|
|||
import ( |
|||
"bytes" |
|||
"compress/gzip" |
|||
"crypto/md5" |
|||
"flag" |
|||
"io" |
|||
"io/ioutil" |
|||
"net/http" |
|||
"time" |
|||
"github.com/chrislusf/seaweedfs/weed/glog" |
|||
"github.com/chrislusf/seaweedfs/weed/storage" |
|||
"github.com/chrislusf/seaweedfs/weed/storage/needle" |
|||
"github.com/chrislusf/seaweedfs/weed/storage/super_block" |
|||
"github.com/chrislusf/seaweedfs/weed/util" |
|||
) |
|||
|
|||
type VolumeFileScanner4SeeDat struct { |
|||
version needle.Version |
|||
} |
|||
|
|||
func (scanner *VolumeFileScanner4SeeDat) VisitSuperBlock(superBlock super_block.SuperBlock) error { |
|||
scanner.version = superBlock.Version |
|||
return nil |
|||
} |
|||
|
|||
func (scanner *VolumeFileScanner4SeeDat) ReadNeedleBody() bool { |
|||
return true |
|||
} |
|||
|
|||
var ( |
|||
files = int64(0) |
|||
filebytes = int64(0) |
|||
diffbytes = int64(0) |
|||
) |
|||
|
|||
func Compresssion(data []byte) float64 { |
|||
if len(data) <= 128 { |
|||
return 100.0 |
|||
} |
|||
compressed, _ := util.GzipData(data[0:128]) |
|||
return float64(len(compressed)*10) / 1280.0 |
|||
} |
|||
|
|||
func (scanner *VolumeFileScanner4SeeDat) VisitNeedle(n *needle.Needle, offset int64, needleHeader, needleBody []byte) error { |
|||
t := time.Unix(int64(n.AppendAtNs)/int64(time.Second), int64(n.AppendAtNs)%int64(time.Second)) |
|||
glog.V(0).Info("----------------------------------------------------------------------------------") |
|||
glog.V(0).Infof("%d,%s%x offset %d size %d(%s) cookie %x appendedAt %v hasmime[%t] mime[%s] (len: %d)", |
|||
*volumeId, n.Id, n.Cookie, offset, n.Size, util.BytesToHumanReadable(uint64(n.Size)), n.Cookie, t, n.HasMime(), string(n.Mime), len(n.Mime)) |
|||
r, err := gzip.NewReader(bytes.NewReader(n.Data)) |
|||
if err == nil { |
|||
buf := bytes.Buffer{} |
|||
h := md5.New() |
|||
c, _ := io.Copy(&buf, r) |
|||
d := buf.Bytes() |
|||
io.Copy(h, bytes.NewReader(d)) |
|||
diff := (int64(n.DataSize) - int64(c)) |
|||
diffbytes += diff |
|||
glog.V(0).Infof("was gzip! stored_size: %d orig_size: %d diff: %d(%d) mime:%s compression-of-128: %.2f md5: %x", n.DataSize, c, diff, diffbytes, http.DetectContentType(d), Compresssion(d), h.Sum(nil)) |
|||
} else { |
|||
glog.V(0).Infof("no gzip!") |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
var ( |
|||
_ = ioutil.ReadAll |
|||
volumePath = flag.String("dir", "/tmp", "data directory to store files") |
|||
volumeCollection = flag.String("collection", "", "the volume collection name") |
|||
volumeId = flag.Int("volumeId", -1, "a volume id. The volume should already exist in the dir. The volume index file should not exist.") |
|||
) |
|||
|
|||
func main() { |
|||
flag.Parse() |
|||
vid := needle.VolumeId(*volumeId) |
|||
glog.V(0).Info("Starting") |
|||
scanner := &VolumeFileScanner4SeeDat{} |
|||
err := storage.ScanVolumeFile(*volumePath, *volumeCollection, vid, storage.NeedleMapInMemory, scanner) |
|||
if err != nil { |
|||
glog.Fatalf("Reading Volume File [ERROR] %s\n", err) |
|||
} |
|||
} |
|||
@ -0,0 +1,27 @@ |
|||
BINARY = weed |
|||
|
|||
SOURCE_DIR = . |
|||
|
|||
all: debug_mount |
|||
|
|||
.PHONY : clean debug_mount |
|||
|
|||
clean: |
|||
go clean $(SOURCE_DIR) |
|||
rm -f $(BINARY) |
|||
|
|||
debug_shell: |
|||
go build -gcflags="all=-N -l" |
|||
dlv --listen=:2345 --headless=true --api-version=2 --accept-multiclient exec weed -- shell |
|||
|
|||
debug_mount: |
|||
go build -gcflags="all=-N -l" |
|||
dlv --listen=:2345 --headless=true --api-version=2 --accept-multiclient exec weed -- mount -dir=~/tmp/mm |
|||
|
|||
debug_server: |
|||
go build -gcflags="all=-N -l" |
|||
dlv --listen=:2345 --headless=true --api-version=2 --accept-multiclient exec weed -- server -dir=/Volumes/mobile_disk/99 -filer -volume.port=8343 -s3 -volume.max=0 |
|||
|
|||
debug_volume: |
|||
go build -gcflags="all=-N -l" |
|||
dlv --listen=:2345 --headless=true --api-version=2 --accept-multiclient exec weed -- volume -dir=/Volumes/mobile_disk/100 -port 8564 -max=30 -preStopSeconds=2 |
|||
@ -0,0 +1,337 @@ |
|||
package command |
|||
|
|||
import ( |
|||
"context" |
|||
"errors" |
|||
"fmt" |
|||
"github.com/chrislusf/seaweedfs/weed/glog" |
|||
"github.com/chrislusf/seaweedfs/weed/pb" |
|||
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb" |
|||
"github.com/chrislusf/seaweedfs/weed/replication" |
|||
"github.com/chrislusf/seaweedfs/weed/replication/sink/filersink" |
|||
"github.com/chrislusf/seaweedfs/weed/replication/source" |
|||
"github.com/chrislusf/seaweedfs/weed/security" |
|||
"github.com/chrislusf/seaweedfs/weed/util" |
|||
"github.com/chrislusf/seaweedfs/weed/util/grace" |
|||
"google.golang.org/grpc" |
|||
"io" |
|||
"strings" |
|||
"time" |
|||
) |
|||
|
|||
type SyncOptions struct { |
|||
isActivePassive *bool |
|||
filerA *string |
|||
filerB *string |
|||
aPath *string |
|||
bPath *string |
|||
aReplication *string |
|||
bReplication *string |
|||
aCollection *string |
|||
bCollection *string |
|||
aTtlSec *int |
|||
bTtlSec *int |
|||
aDebug *bool |
|||
bDebug *bool |
|||
} |
|||
|
|||
var ( |
|||
syncOptions SyncOptions |
|||
syncCpuProfile *string |
|||
syncMemProfile *string |
|||
) |
|||
|
|||
func init() { |
|||
cmdFilerSynchronize.Run = runFilerSynchronize // break init cycle
|
|||
syncOptions.isActivePassive = cmdFilerSynchronize.Flag.Bool("isActivePassive", false, "one directional follow if true") |
|||
syncOptions.filerA = cmdFilerSynchronize.Flag.String("a", "", "filer A in one SeaweedFS cluster") |
|||
syncOptions.filerB = cmdFilerSynchronize.Flag.String("b", "", "filer B in the other SeaweedFS cluster") |
|||
syncOptions.aPath = cmdFilerSynchronize.Flag.String("a.path", "/", "directory to sync on filer A") |
|||
syncOptions.bPath = cmdFilerSynchronize.Flag.String("b.path", "/", "directory to sync on filer B") |
|||
syncOptions.aReplication = cmdFilerSynchronize.Flag.String("a.replication", "", "replication on filer A") |
|||
syncOptions.bReplication = cmdFilerSynchronize.Flag.String("b.replication", "", "replication on filer B") |
|||
syncOptions.aCollection = cmdFilerSynchronize.Flag.String("a.collection", "", "collection on filer A") |
|||
syncOptions.bCollection = cmdFilerSynchronize.Flag.String("b.collection", "", "collection on filer B") |
|||
syncOptions.aTtlSec = cmdFilerSynchronize.Flag.Int("a.ttlSec", 0, "ttl in seconds on filer A") |
|||
syncOptions.bTtlSec = cmdFilerSynchronize.Flag.Int("b.ttlSec", 0, "ttl in seconds on filer B") |
|||
syncOptions.aDebug = cmdFilerSynchronize.Flag.Bool("a.debug", false, "debug mode to print out filer A received files") |
|||
syncOptions.bDebug = cmdFilerSynchronize.Flag.Bool("b.debug", false, "debug mode to print out filer B received files") |
|||
syncCpuProfile = cmdFilerSynchronize.Flag.String("cpuprofile", "", "cpu profile output file") |
|||
syncMemProfile = cmdFilerSynchronize.Flag.String("memprofile", "", "memory profile output file") |
|||
} |
|||
|
|||
var cmdFilerSynchronize = &Command{ |
|||
UsageLine: "filer.sync -a=<oneFilerHost>:<oneFilerPort> -b=<otherFilerHost>:<otherFilerPort>", |
|||
Short: "continuously synchronize between two active-active or active-passive SeaweedFS clusters", |
|||
Long: `continuously synchronize file changes between two active-active or active-passive filers |
|||
|
|||
filer.sync listens on filer notifications. If any file is updated, it will fetch the updated content, |
|||
and write to the other destination. Different from filer.replicate: |
|||
|
|||
* filer.sync only works between two filers. |
|||
* filer.sync does not need any special message queue setup. |
|||
* filer.sync supports both active-active and active-passive modes. |
|||
|
|||
If restarted, the synchronization will resume from the previous checkpoints, persisted every minute. |
|||
A fresh sync will start from the earliest metadata logs. |
|||
|
|||
`, |
|||
} |
|||
|
|||
func runFilerSynchronize(cmd *Command, args []string) bool { |
|||
|
|||
grpcDialOption := security.LoadClientTLS(util.GetViper(), "grpc.client") |
|||
|
|||
grace.SetupProfiling(*syncCpuProfile, *syncMemProfile) |
|||
|
|||
go func() { |
|||
for { |
|||
err := doSubscribeFilerMetaChanges(grpcDialOption, *syncOptions.filerA, *syncOptions.aPath, *syncOptions.filerB, |
|||
*syncOptions.bPath, *syncOptions.bReplication, *syncOptions.bCollection, *syncOptions.bTtlSec, *syncOptions.bDebug) |
|||
if err != nil { |
|||
glog.Errorf("sync from %s to %s: %v", *syncOptions.filerA, *syncOptions.filerB, err) |
|||
time.Sleep(1747 * time.Millisecond) |
|||
} |
|||
} |
|||
}() |
|||
|
|||
if !*syncOptions.isActivePassive { |
|||
go func() { |
|||
for { |
|||
err := doSubscribeFilerMetaChanges(grpcDialOption, *syncOptions.filerB, *syncOptions.bPath, *syncOptions.filerA, |
|||
*syncOptions.aPath, *syncOptions.aReplication, *syncOptions.aCollection, *syncOptions.aTtlSec, *syncOptions.aDebug) |
|||
if err != nil { |
|||
glog.Errorf("sync from %s to %s: %v", *syncOptions.filerB, *syncOptions.filerA, err) |
|||
time.Sleep(2147 * time.Millisecond) |
|||
} |
|||
} |
|||
}() |
|||
} |
|||
|
|||
select {} |
|||
|
|||
return true |
|||
} |
|||
|
|||
func doSubscribeFilerMetaChanges(grpcDialOption grpc.DialOption, sourceFiler, sourcePath, targetFiler, targetPath string, |
|||
replicationStr, collection string, ttlSec int, debug bool) error { |
|||
|
|||
// read source filer signature
|
|||
sourceFilerSignature, sourceErr := replication.ReadFilerSignature(grpcDialOption, sourceFiler) |
|||
if sourceErr != nil { |
|||
return sourceErr |
|||
} |
|||
// read target filer signature
|
|||
targetFilerSignature, targetErr := replication.ReadFilerSignature(grpcDialOption, targetFiler) |
|||
if targetErr != nil { |
|||
return targetErr |
|||
} |
|||
|
|||
// if first time, start from now
|
|||
// if has previously synced, resume from that point of time
|
|||
sourceFilerOffsetTsNs, err := readSyncOffset(grpcDialOption, targetFiler, sourceFilerSignature) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
glog.V(0).Infof("start sync %s(%d) => %s(%d) from %v(%d)", sourceFiler, sourceFilerSignature, targetFiler, targetFilerSignature, time.Unix(0, sourceFilerOffsetTsNs), sourceFilerOffsetTsNs) |
|||
|
|||
// create filer sink
|
|||
filerSource := &source.FilerSource{} |
|||
filerSource.DoInitialize(pb.ServerToGrpcAddress(sourceFiler), sourcePath) |
|||
filerSink := &filersink.FilerSink{} |
|||
filerSink.DoInitialize(pb.ServerToGrpcAddress(targetFiler), targetPath, replicationStr, collection, ttlSec, grpcDialOption) |
|||
filerSink.SetSourceFiler(filerSource) |
|||
|
|||
processEventFn := func(resp *filer_pb.SubscribeMetadataResponse) error { |
|||
message := resp.EventNotification |
|||
|
|||
var sourceOldKey, sourceNewKey util.FullPath |
|||
if message.OldEntry != nil { |
|||
sourceOldKey = util.FullPath(resp.Directory).Child(message.OldEntry.Name) |
|||
} |
|||
if message.NewEntry != nil { |
|||
sourceNewKey = util.FullPath(message.NewParentPath).Child(message.NewEntry.Name) |
|||
} |
|||
|
|||
for _, sig := range message.Signatures { |
|||
if sig == targetFilerSignature && targetFilerSignature != 0 { |
|||
fmt.Printf("%s skipping %s change to %v\n", targetFiler, sourceFiler, message) |
|||
return nil |
|||
} |
|||
} |
|||
if debug { |
|||
fmt.Printf("%s check %s change %s,%s sig %v, target sig: %v\n", targetFiler, sourceFiler, sourceOldKey, sourceNewKey, message.Signatures, targetFilerSignature) |
|||
} |
|||
|
|||
if !strings.HasPrefix(resp.Directory, sourcePath) { |
|||
return nil |
|||
} |
|||
|
|||
// handle deletions
|
|||
if message.OldEntry != nil && message.NewEntry == nil { |
|||
if !strings.HasPrefix(string(sourceOldKey), sourcePath) { |
|||
return nil |
|||
} |
|||
key := util.Join(targetPath, string(sourceOldKey)[len(sourcePath):]) |
|||
return filerSink.DeleteEntry(key, message.OldEntry.IsDirectory, message.DeleteChunks, message.Signatures) |
|||
} |
|||
|
|||
// handle new entries
|
|||
if message.OldEntry == nil && message.NewEntry != nil { |
|||
if !strings.HasPrefix(string(sourceNewKey), sourcePath) { |
|||
return nil |
|||
} |
|||
key := util.Join(targetPath, string(sourceNewKey)[len(sourcePath):]) |
|||
return filerSink.CreateEntry(key, message.NewEntry, message.Signatures) |
|||
} |
|||
|
|||
// this is something special?
|
|||
if message.OldEntry == nil && message.NewEntry == nil { |
|||
return nil |
|||
} |
|||
|
|||
// handle updates
|
|||
if strings.HasPrefix(string(sourceOldKey), sourcePath) { |
|||
// old key is in the watched directory
|
|||
if strings.HasPrefix(string(sourceNewKey), sourcePath) { |
|||
// new key is also in the watched directory
|
|||
oldKey := util.Join(targetPath, string(sourceOldKey)[len(sourcePath):]) |
|||
message.NewParentPath = util.Join(targetPath, message.NewParentPath[len(sourcePath):]) |
|||
foundExisting, err := filerSink.UpdateEntry(string(oldKey), message.OldEntry, message.NewParentPath, message.NewEntry, message.DeleteChunks, message.Signatures) |
|||
if foundExisting { |
|||
return err |
|||
} |
|||
|
|||
// not able to find old entry
|
|||
if err = filerSink.DeleteEntry(string(oldKey), message.OldEntry.IsDirectory, false, message.Signatures); err != nil { |
|||
return fmt.Errorf("delete old entry %v: %v", oldKey, err) |
|||
} |
|||
|
|||
// create the new entry
|
|||
newKey := util.Join(targetPath, string(sourceNewKey)[len(sourcePath):]) |
|||
return filerSink.CreateEntry(newKey, message.NewEntry, message.Signatures) |
|||
|
|||
} else { |
|||
// new key is outside of the watched directory
|
|||
key := util.Join(targetPath, string(sourceOldKey)[len(sourcePath):]) |
|||
return filerSink.DeleteEntry(key, message.OldEntry.IsDirectory, message.DeleteChunks, message.Signatures) |
|||
} |
|||
} else { |
|||
// old key is outside of the watched directory
|
|||
if strings.HasPrefix(string(sourceNewKey), sourcePath) { |
|||
// new key is in the watched directory
|
|||
key := util.Join(targetPath, string(sourceNewKey)[len(sourcePath):]) |
|||
return filerSink.CreateEntry(key, message.NewEntry, message.Signatures) |
|||
} else { |
|||
// new key is also outside of the watched directory
|
|||
// skip
|
|||
} |
|||
} |
|||
|
|||
return nil |
|||
} |
|||
|
|||
return pb.WithFilerClient(sourceFiler, grpcDialOption, func(client filer_pb.SeaweedFilerClient) error { |
|||
|
|||
ctx, cancel := context.WithCancel(context.Background()) |
|||
defer cancel() |
|||
|
|||
stream, err := client.SubscribeMetadata(ctx, &filer_pb.SubscribeMetadataRequest{ |
|||
ClientName: "syncTo_" + targetFiler, |
|||
PathPrefix: sourcePath, |
|||
SinceNs: sourceFilerOffsetTsNs, |
|||
Signature: targetFilerSignature, |
|||
}) |
|||
if err != nil { |
|||
return fmt.Errorf("listen: %v", err) |
|||
} |
|||
|
|||
var counter int64 |
|||
var lastWriteTime time.Time |
|||
for { |
|||
resp, listenErr := stream.Recv() |
|||
if listenErr == io.EOF { |
|||
return nil |
|||
} |
|||
if listenErr != nil { |
|||
return listenErr |
|||
} |
|||
|
|||
if err := processEventFn(resp); err != nil { |
|||
return err |
|||
} |
|||
|
|||
counter++ |
|||
if lastWriteTime.Add(3 * time.Second).Before(time.Now()) { |
|||
glog.V(0).Infof("sync %s => %s progressed to %v %0.2f/sec", sourceFiler, targetFiler, time.Unix(0, resp.TsNs), float64(counter)/float64(3)) |
|||
counter = 0 |
|||
lastWriteTime = time.Now() |
|||
if err := writeSyncOffset(grpcDialOption, targetFiler, sourceFilerSignature, resp.TsNs); err != nil { |
|||
return err |
|||
} |
|||
} |
|||
|
|||
} |
|||
|
|||
}) |
|||
|
|||
} |
|||
|
|||
const ( |
|||
SyncKeyPrefix = "sync." |
|||
) |
|||
|
|||
func readSyncOffset(grpcDialOption grpc.DialOption, filer string, filerSignature int32) (lastOffsetTsNs int64, readErr error) { |
|||
|
|||
readErr = pb.WithFilerClient(filer, grpcDialOption, func(client filer_pb.SeaweedFilerClient) error { |
|||
syncKey := []byte(SyncKeyPrefix + "____") |
|||
util.Uint32toBytes(syncKey[len(SyncKeyPrefix):len(SyncKeyPrefix)+4], uint32(filerSignature)) |
|||
|
|||
resp, err := client.KvGet(context.Background(), &filer_pb.KvGetRequest{Key: syncKey}) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
if len(resp.Error) != 0 { |
|||
return errors.New(resp.Error) |
|||
} |
|||
if len(resp.Value) < 8 { |
|||
return nil |
|||
} |
|||
|
|||
lastOffsetTsNs = int64(util.BytesToUint64(resp.Value)) |
|||
|
|||
return nil |
|||
}) |
|||
|
|||
return |
|||
|
|||
} |
|||
|
|||
func writeSyncOffset(grpcDialOption grpc.DialOption, filer string, filerSignature int32, offsetTsNs int64) error { |
|||
return pb.WithFilerClient(filer, grpcDialOption, func(client filer_pb.SeaweedFilerClient) error { |
|||
|
|||
syncKey := []byte(SyncKeyPrefix + "____") |
|||
util.Uint32toBytes(syncKey[len(SyncKeyPrefix):len(SyncKeyPrefix)+4], uint32(filerSignature)) |
|||
|
|||
valueBuf := make([]byte, 8) |
|||
util.Uint64toBytes(valueBuf, uint64(offsetTsNs)) |
|||
|
|||
resp, err := client.KvPut(context.Background(), &filer_pb.KvPutRequest{ |
|||
Key: syncKey, |
|||
Value: valueBuf, |
|||
}) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
if len(resp.Error) != 0 { |
|||
return errors.New(resp.Error) |
|||
} |
|||
|
|||
return nil |
|||
|
|||
}) |
|||
|
|||
} |
|||
@ -0,0 +1,87 @@ |
|||
package abstract_sql |
|||
|
|||
import ( |
|||
"context" |
|||
"database/sql" |
|||
"fmt" |
|||
"strings" |
|||
|
|||
"github.com/chrislusf/seaweedfs/weed/filer" |
|||
"github.com/chrislusf/seaweedfs/weed/glog" |
|||
"github.com/chrislusf/seaweedfs/weed/util" |
|||
) |
|||
|
|||
func (store *AbstractSqlStore) KvPut(ctx context.Context, key []byte, value []byte) (err error) { |
|||
|
|||
dirStr, dirHash, name := genDirAndName(key) |
|||
|
|||
res, err := store.getTxOrDB(ctx).ExecContext(ctx, store.SqlInsert, dirHash, name, dirStr, value) |
|||
if err != nil { |
|||
if !strings.Contains(strings.ToLower(err.Error()), "duplicate") { |
|||
return fmt.Errorf("kv insert: %s", err) |
|||
} |
|||
} |
|||
|
|||
// now the insert failed possibly due to duplication constraints
|
|||
glog.V(1).Infof("kv insert falls back to update: %s", err) |
|||
|
|||
res, err = store.getTxOrDB(ctx).ExecContext(ctx, store.SqlUpdate, value, dirHash, name, dirStr) |
|||
if err != nil { |
|||
return fmt.Errorf("kv upsert: %s", err) |
|||
} |
|||
|
|||
_, err = res.RowsAffected() |
|||
if err != nil { |
|||
return fmt.Errorf("kv upsert no rows affected: %s", err) |
|||
} |
|||
return nil |
|||
|
|||
} |
|||
|
|||
func (store *AbstractSqlStore) KvGet(ctx context.Context, key []byte) (value []byte, err error) { |
|||
|
|||
dirStr, dirHash, name := genDirAndName(key) |
|||
row := store.getTxOrDB(ctx).QueryRowContext(ctx, store.SqlFind, dirHash, name, dirStr) |
|||
|
|||
err = row.Scan(&value) |
|||
|
|||
if err == sql.ErrNoRows { |
|||
return nil, filer.ErrKvNotFound |
|||
} |
|||
|
|||
if err != nil { |
|||
return nil, fmt.Errorf("kv get: %v", err) |
|||
} |
|||
|
|||
return |
|||
} |
|||
|
|||
func (store *AbstractSqlStore) KvDelete(ctx context.Context, key []byte) (err error) { |
|||
|
|||
dirStr, dirHash, name := genDirAndName(key) |
|||
|
|||
res, err := store.getTxOrDB(ctx).ExecContext(ctx, store.SqlDelete, dirHash, name, dirStr) |
|||
if err != nil { |
|||
return fmt.Errorf("kv delete: %s", err) |
|||
} |
|||
|
|||
_, err = res.RowsAffected() |
|||
if err != nil { |
|||
return fmt.Errorf("kv delete no rows affected: %s", err) |
|||
} |
|||
|
|||
return nil |
|||
|
|||
} |
|||
|
|||
func genDirAndName(key []byte) (dirStr string, dirHash int64, name string) { |
|||
for len(key) < 8 { |
|||
key = append(key, 0) |
|||
} |
|||
|
|||
dirHash = int64(util.BytesToUint64(key[:8])) |
|||
dirStr = string(key[:8]) |
|||
name = string(key[8:]) |
|||
|
|||
return |
|||
} |
|||
@ -0,0 +1,61 @@ |
|||
package cassandra |
|||
|
|||
import ( |
|||
"context" |
|||
"fmt" |
|||
"github.com/chrislusf/seaweedfs/weed/filer" |
|||
"github.com/gocql/gocql" |
|||
) |
|||
|
|||
func (store *CassandraStore) KvPut(ctx context.Context, key []byte, value []byte) (err error) { |
|||
dir, name := genDirAndName(key) |
|||
|
|||
if err := store.session.Query( |
|||
"INSERT INTO filemeta (directory,name,meta) VALUES(?,?,?) USING TTL ? ", |
|||
dir, name, value, 0).Exec(); err != nil { |
|||
return fmt.Errorf("kv insert: %s", err) |
|||
} |
|||
|
|||
return nil |
|||
} |
|||
|
|||
func (store *CassandraStore) KvGet(ctx context.Context, key []byte) (data []byte, err error) { |
|||
dir, name := genDirAndName(key) |
|||
|
|||
if err := store.session.Query( |
|||
"SELECT meta FROM filemeta WHERE directory=? AND name=?", |
|||
dir, name).Consistency(gocql.One).Scan(&data); err != nil { |
|||
if err != gocql.ErrNotFound { |
|||
return nil, filer.ErrKvNotFound |
|||
} |
|||
} |
|||
|
|||
if len(data) == 0 { |
|||
return nil, filer.ErrKvNotFound |
|||
} |
|||
|
|||
return data, nil |
|||
} |
|||
|
|||
func (store *CassandraStore) KvDelete(ctx context.Context, key []byte) (err error) { |
|||
dir, name := genDirAndName(key) |
|||
|
|||
if err := store.session.Query( |
|||
"DELETE FROM filemeta WHERE directory=? AND name=?", |
|||
dir, name).Exec(); err != nil { |
|||
return fmt.Errorf("kv delete: %v", err) |
|||
} |
|||
|
|||
return nil |
|||
} |
|||
|
|||
func genDirAndName(key []byte) (dir string, name string) { |
|||
for len(key) < 8 { |
|||
key = append(key, 0) |
|||
} |
|||
|
|||
dir = string(key[:8]) |
|||
name = string(key[8:]) |
|||
|
|||
return |
|||
} |
|||
@ -1,4 +1,4 @@ |
|||
package filer2 |
|||
package filer |
|||
|
|||
import ( |
|||
"os" |
|||
@ -0,0 +1,338 @@ |
|||
package elastic |
|||
|
|||
import ( |
|||
"context" |
|||
"fmt" |
|||
"math" |
|||
"strings" |
|||
|
|||
"github.com/chrislusf/seaweedfs/weed/filer" |
|||
"github.com/chrislusf/seaweedfs/weed/glog" |
|||
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb" |
|||
weed_util "github.com/chrislusf/seaweedfs/weed/util" |
|||
jsoniter "github.com/json-iterator/go" |
|||
elastic "github.com/olivere/elastic/v7" |
|||
) |
|||
|
|||
var ( |
|||
indexType = "_doc" |
|||
indexPrefix = ".seaweedfs_" |
|||
indexKV = ".seaweedfs_kv_entries" |
|||
kvMappings = ` { |
|||
"mappings": { |
|||
"enabled": false, |
|||
"properties": { |
|||
"Value":{ |
|||
"type": "binary" |
|||
} |
|||
} |
|||
} |
|||
}` |
|||
) |
|||
|
|||
type ESEntry struct { |
|||
ParentId string `json:"ParentId"` |
|||
Entry *filer.Entry |
|||
} |
|||
|
|||
type ESKVEntry struct { |
|||
Value []byte `json:"Value"` |
|||
} |
|||
|
|||
func init() { |
|||
filer.Stores = append(filer.Stores, &ElasticStore{}) |
|||
} |
|||
|
|||
type ElasticStore struct { |
|||
client *elastic.Client |
|||
maxPageSize int |
|||
} |
|||
|
|||
func (store *ElasticStore) GetName() string { |
|||
return "elastic7" |
|||
} |
|||
|
|||
func (store *ElasticStore) Initialize(configuration weed_util.Configuration, prefix string) (err error) { |
|||
options := []elastic.ClientOptionFunc{} |
|||
servers := configuration.GetStringSlice(prefix + "servers") |
|||
options = append(options, elastic.SetURL(servers...)) |
|||
username := configuration.GetString(prefix + "username") |
|||
password := configuration.GetString(prefix + "password") |
|||
if username != "" && password != "" { |
|||
options = append(options, elastic.SetBasicAuth(username, password)) |
|||
} |
|||
options = append(options, elastic.SetSniff(configuration.GetBool(prefix+"sniff_enabled"))) |
|||
options = append(options, elastic.SetHealthcheck(configuration.GetBool(prefix+"healthcheck_enabled"))) |
|||
store.maxPageSize = configuration.GetInt(prefix + "index.max_result_window") |
|||
if store.maxPageSize <= 0 { |
|||
store.maxPageSize = 10000 |
|||
} |
|||
glog.Infof("filer store elastic endpoints: %v.", servers) |
|||
return store.initialize(options) |
|||
} |
|||
|
|||
func (store *ElasticStore) initialize(options []elastic.ClientOptionFunc) (err error) { |
|||
ctx := context.Background() |
|||
store.client, err = elastic.NewClient(options...) |
|||
if err != nil { |
|||
return fmt.Errorf("init elastic %v.", err) |
|||
} |
|||
if ok, err := store.client.IndexExists(indexKV).Do(ctx); err == nil && !ok { |
|||
_, err = store.client.CreateIndex(indexKV).Body(kvMappings).Do(ctx) |
|||
if err != nil { |
|||
return fmt.Errorf("create index(%s) %v.", indexKV, err) |
|||
} |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (store *ElasticStore) BeginTransaction(ctx context.Context) (context.Context, error) { |
|||
return ctx, nil |
|||
} |
|||
func (store *ElasticStore) CommitTransaction(ctx context.Context) error { |
|||
return nil |
|||
} |
|||
func (store *ElasticStore) RollbackTransaction(ctx context.Context) error { |
|||
return nil |
|||
} |
|||
|
|||
func (store *ElasticStore) ListDirectoryPrefixedEntries(ctx context.Context, fullpath weed_util.FullPath, startFileName string, inclusive bool, limit int, prefix string) (entries []*filer.Entry, err error) { |
|||
return nil, filer.ErrUnsupportedListDirectoryPrefixed |
|||
} |
|||
|
|||
func (store *ElasticStore) InsertEntry(ctx context.Context, entry *filer.Entry) (err error) { |
|||
index := getIndex(entry.FullPath) |
|||
dir, _ := entry.FullPath.DirAndName() |
|||
id := weed_util.Md5String([]byte(entry.FullPath)) |
|||
esEntry := &ESEntry{ |
|||
ParentId: weed_util.Md5String([]byte(dir)), |
|||
Entry: entry, |
|||
} |
|||
value, err := jsoniter.Marshal(esEntry) |
|||
if err != nil { |
|||
glog.Errorf("insert entry(%s) %v.", string(entry.FullPath), err) |
|||
return fmt.Errorf("insert entry %v.", err) |
|||
} |
|||
_, err = store.client.Index(). |
|||
Index(index). |
|||
Type(indexType). |
|||
Id(id). |
|||
BodyJson(string(value)). |
|||
Do(ctx) |
|||
if err != nil { |
|||
glog.Errorf("insert entry(%s) %v.", string(entry.FullPath), err) |
|||
return fmt.Errorf("insert entry %v.", err) |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (store *ElasticStore) UpdateEntry(ctx context.Context, entry *filer.Entry) (err error) { |
|||
return store.InsertEntry(ctx, entry) |
|||
} |
|||
|
|||
func (store *ElasticStore) FindEntry(ctx context.Context, fullpath weed_util.FullPath) (entry *filer.Entry, err error) { |
|||
index := getIndex(fullpath) |
|||
id := weed_util.Md5String([]byte(fullpath)) |
|||
searchResult, err := store.client.Get(). |
|||
Index(index). |
|||
Type(indexType). |
|||
Id(id). |
|||
Do(ctx) |
|||
if elastic.IsNotFound(err) { |
|||
return nil, filer_pb.ErrNotFound |
|||
} |
|||
if searchResult != nil && searchResult.Found { |
|||
esEntry := &ESEntry{ |
|||
ParentId: "", |
|||
Entry: &filer.Entry{}, |
|||
} |
|||
err := jsoniter.Unmarshal(searchResult.Source, esEntry) |
|||
return esEntry.Entry, err |
|||
} |
|||
glog.Errorf("find entry(%s),%v.", string(fullpath), err) |
|||
return nil, filer_pb.ErrNotFound |
|||
} |
|||
|
|||
func (store *ElasticStore) DeleteEntry(ctx context.Context, fullpath weed_util.FullPath) (err error) { |
|||
index := getIndex(fullpath) |
|||
id := weed_util.Md5String([]byte(fullpath)) |
|||
if strings.Count(string(fullpath), "/") == 1 { |
|||
return store.deleteIndex(ctx, index) |
|||
} |
|||
return store.deleteEntry(ctx, index, id) |
|||
} |
|||
|
|||
func (store *ElasticStore) deleteIndex(ctx context.Context, index string) (err error) { |
|||
deleteResult, err := store.client.DeleteIndex(index).Do(ctx) |
|||
if elastic.IsNotFound(err) || (err == nil && deleteResult.Acknowledged) { |
|||
return nil |
|||
} |
|||
glog.Errorf("delete index(%s) %v.", index, err) |
|||
return err |
|||
} |
|||
|
|||
func (store *ElasticStore) deleteEntry(ctx context.Context, index, id string) (err error) { |
|||
deleteResult, err := store.client.Delete(). |
|||
Index(index). |
|||
Type(indexType). |
|||
Id(id). |
|||
Do(ctx) |
|||
if err == nil { |
|||
if deleteResult.Result == "deleted" || deleteResult.Result == "not_found" { |
|||
return nil |
|||
} |
|||
} |
|||
glog.Errorf("delete entry(index:%s,_id:%s) %v.", index, id, err) |
|||
return fmt.Errorf("delete entry %v.", err) |
|||
} |
|||
|
|||
func (store *ElasticStore) DeleteFolderChildren(ctx context.Context, fullpath weed_util.FullPath) (err error) { |
|||
if entries, err := store.ListDirectoryEntries(ctx, fullpath, "", false, math.MaxInt32); err == nil { |
|||
for _, entry := range entries { |
|||
store.DeleteEntry(ctx, entry.FullPath) |
|||
} |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func (store *ElasticStore) ListDirectoryEntries( |
|||
ctx context.Context, fullpath weed_util.FullPath, startFileName string, inclusive bool, limit int, |
|||
) (entries []*filer.Entry, err error) { |
|||
if string(fullpath) == "/" { |
|||
return store.listRootDirectoryEntries(ctx, startFileName, inclusive, limit) |
|||
} |
|||
return store.listDirectoryEntries(ctx, fullpath, startFileName, inclusive, limit) |
|||
} |
|||
|
|||
func (store *ElasticStore) listRootDirectoryEntries(ctx context.Context, startFileName string, inclusive bool, limit int) (entries []*filer.Entry, err error) { |
|||
indexResult, err := store.client.CatIndices().Do(ctx) |
|||
if err != nil { |
|||
glog.Errorf("list indices %v.", err) |
|||
return entries, err |
|||
} |
|||
for _, index := range indexResult { |
|||
if index.Index == indexKV { |
|||
continue |
|||
} |
|||
if strings.HasPrefix(index.Index, indexPrefix) { |
|||
if entry, err := store.FindEntry(ctx, |
|||
weed_util.FullPath("/"+strings.Replace(index.Index, indexPrefix, "", 1))); err == nil { |
|||
fileName := getFileName(entry.FullPath) |
|||
if fileName == startFileName && !inclusive { |
|||
continue |
|||
} |
|||
limit-- |
|||
if limit < 0 { |
|||
break |
|||
} |
|||
entries = append(entries, entry) |
|||
} |
|||
} |
|||
} |
|||
return entries, nil |
|||
} |
|||
|
|||
func (store *ElasticStore) listDirectoryEntries( |
|||
ctx context.Context, fullpath weed_util.FullPath, startFileName string, inclusive bool, limit int, |
|||
) (entries []*filer.Entry, err error) { |
|||
first := true |
|||
index := getIndex(fullpath) |
|||
nextStart := "" |
|||
parentId := weed_util.Md5String([]byte(fullpath)) |
|||
if _, err := store.client.Refresh(index).Do(ctx); err != nil { |
|||
if elastic.IsNotFound(err) { |
|||
store.client.CreateIndex(index).Do(ctx) |
|||
return entries, nil |
|||
} |
|||
} |
|||
for { |
|||
result := &elastic.SearchResult{} |
|||
if (startFileName == "" && first) || inclusive { |
|||
if result, err = store.search(ctx, index, parentId); err != nil { |
|||
glog.Errorf("search (%s,%s,%t,%d) %v.", string(fullpath), startFileName, inclusive, limit, err) |
|||
return entries, err |
|||
} |
|||
} else { |
|||
fullPath := string(fullpath) + "/" + startFileName |
|||
if !first { |
|||
fullPath = nextStart |
|||
} |
|||
after := weed_util.Md5String([]byte(fullPath)) |
|||
if result, err = store.searchAfter(ctx, index, parentId, after); err != nil { |
|||
glog.Errorf("searchAfter (%s,%s,%t,%d) %v.", string(fullpath), startFileName, inclusive, limit, err) |
|||
return entries, err |
|||
} |
|||
} |
|||
first = false |
|||
for _, hit := range result.Hits.Hits { |
|||
esEntry := &ESEntry{ |
|||
ParentId: "", |
|||
Entry: &filer.Entry{}, |
|||
} |
|||
if err := jsoniter.Unmarshal(hit.Source, esEntry); err == nil { |
|||
limit-- |
|||
if limit < 0 { |
|||
return entries, nil |
|||
} |
|||
nextStart = string(esEntry.Entry.FullPath) |
|||
fileName := getFileName(esEntry.Entry.FullPath) |
|||
if fileName == startFileName && !inclusive { |
|||
continue |
|||
} |
|||
entries = append(entries, esEntry.Entry) |
|||
} |
|||
} |
|||
if len(result.Hits.Hits) < store.maxPageSize { |
|||
break |
|||
} |
|||
} |
|||
return entries, nil |
|||
} |
|||
|
|||
func (store *ElasticStore) search(ctx context.Context, index, parentId string) (result *elastic.SearchResult, err error) { |
|||
if count, err := store.client.Count(index).Do(ctx); err == nil && count == 0 { |
|||
return &elastic.SearchResult{ |
|||
Hits: &elastic.SearchHits{ |
|||
Hits: make([]*elastic.SearchHit, 0)}, |
|||
}, nil |
|||
} |
|||
queryResult, err := store.client.Search(). |
|||
Index(index). |
|||
Query(elastic.NewMatchQuery("ParentId", parentId)). |
|||
Size(store.maxPageSize). |
|||
Sort("_id", false). |
|||
Do(ctx) |
|||
return queryResult, err |
|||
} |
|||
|
|||
func (store *ElasticStore) searchAfter(ctx context.Context, index, parentId, after string) (result *elastic.SearchResult, err error) { |
|||
queryResult, err := store.client.Search(). |
|||
Index(index). |
|||
Query(elastic.NewMatchQuery("ParentId", parentId)). |
|||
SearchAfter(after). |
|||
Size(store.maxPageSize). |
|||
Sort("_id", false). |
|||
Do(ctx) |
|||
return queryResult, err |
|||
|
|||
} |
|||
|
|||
func (store *ElasticStore) Shutdown() { |
|||
store.client.Stop() |
|||
} |
|||
|
|||
func getIndex(fullpath weed_util.FullPath) string { |
|||
path := strings.Split(string(fullpath), "/") |
|||
if len(path) > 1 { |
|||
return indexPrefix + path[1] |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
func getFileName(fullpath weed_util.FullPath) string { |
|||
path := strings.Split(string(fullpath), "/") |
|||
if len(path) > 1 { |
|||
return path[len(path)-1] |
|||
} |
|||
return "" |
|||
} |
|||
Some files were not shown because too many files changed in this diff
Write
Preview
Loading…
Cancel
Save
Reference in new issue