104 lines
3.2 KiB
Java
104 lines
3.2 KiB
Java
/*
|
|
* FileSponge
|
|
* Copyright (C) 2021 Andrea Cavalli
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package org.warp.filesponge;
|
|
|
|
import it.cavallium.dbengine.database.serialization.Serializer;
|
|
import it.unimi.dsi.fastutil.booleans.BooleanArrayList;
|
|
import java.io.ByteArrayInputStream;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.DataInputStream;
|
|
import java.io.DataOutputStream;
|
|
import lombok.Data;
|
|
import lombok.SneakyThrows;
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
|
@Data
|
|
public class DiskMetadata {
|
|
|
|
/**
|
|
* -1 = unknown size
|
|
*/
|
|
private final int size;
|
|
|
|
private final BooleanArrayList downloadedBlocks;
|
|
|
|
private Boolean downloadedFully;
|
|
|
|
public boolean isDownloadedFully() {
|
|
if (downloadedFully == null) {
|
|
// Ensure blocks count is valid by calling getBlocksCount()
|
|
getBlocksCount();
|
|
// It's fully downloaded if every block is true
|
|
downloadedFully = !this.getDownloadedBlocks().contains(false);
|
|
}
|
|
return downloadedFully;
|
|
}
|
|
|
|
private int getBlocksCount() {
|
|
var expectedBlocksCount = getBlocksCount(size, FileSponge.BLOCK_SIZE);
|
|
if (this.getDownloadedBlocks().size() != expectedBlocksCount) {
|
|
throw new IllegalStateException(
|
|
"Blocks array length (" + this.getDownloadedBlocks().size()
|
|
+ ") != expected blocks count (" + expectedBlocksCount + ")");
|
|
}
|
|
return expectedBlocksCount;
|
|
}
|
|
|
|
public static int getBlocksCount(int size, int blockSize) {
|
|
return (size + (blockSize - size % blockSize)) / blockSize;
|
|
}
|
|
|
|
public Metadata asMetadata() {
|
|
return new Metadata(size);
|
|
}
|
|
|
|
public static class DiskMetadataSerializer implements Serializer<DiskMetadata, byte[]> {
|
|
|
|
@SneakyThrows
|
|
@Override
|
|
public @NotNull DiskMetadata deserialize(byte @NotNull [] serialized) {
|
|
var bais = new ByteArrayInputStream(serialized);
|
|
var dis = new DataInputStream(bais);
|
|
int size = dis.readInt();
|
|
int blocksCount = getBlocksCount(size, FileSponge.BLOCK_SIZE);
|
|
var downloadedBlocks = new BooleanArrayList(blocksCount);
|
|
for (int i = 0; i < blocksCount; i++) {
|
|
downloadedBlocks.add(dis.readBoolean());
|
|
}
|
|
return new DiskMetadata(size, downloadedBlocks);
|
|
}
|
|
|
|
@SneakyThrows
|
|
@Override
|
|
public byte @NotNull [] serialize(@NotNull DiskMetadata deserialized) {
|
|
try (var bos = new ByteArrayOutputStream()) {
|
|
try (var dos = new DataOutputStream(bos)) {
|
|
dos.writeInt(deserialized.getSize());
|
|
deserialized.getBlocksCount();
|
|
for (boolean downloadedBlock : deserialized.getDownloadedBlocks()) {
|
|
dos.writeBoolean(downloadedBlock);
|
|
}
|
|
}
|
|
return bos.toByteArray();
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|