io.activej.fs.BlockingFs Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of activej-fs Show documentation

Provides tools for building efficient, scalable local, remote or clustered file servers. It utilizes ActiveJ CSP for fast and reliable file transfer.

There is a newer version: 6.0-beta2

Show newest version

package io.activej.fs;

import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;

import static io.activej.common.Checks.checkArgument;
import static io.activej.common.collection.CollectionUtils.isBijection;
import static io.activej.fs.util.RemoteFsUtils.escapeGlob;
import static java.util.stream.Collectors.toSet;

public interface BlockingFs {
	String SEPARATOR = ActiveFs.SEPARATOR;

	OutputStream upload(@NotNull String name) throws IOException;

	OutputStream upload(@NotNull String name, long size) throws IOException;

	OutputStream append(@NotNull String name, long offset) throws IOException;

	InputStream download(@NotNull String name, long offset, long limit) throws IOException;

	default InputStream download(@NotNull String name) throws IOException {
		return download(name, 0, Long.MAX_VALUE);
	}

	void delete(@NotNull String name) throws IOException;

	default void deleteAll(Set toDelete) throws IOException {
		for (String file : toDelete) {
			delete(file);
		}
	}

	/**
	 * Duplicates a file
	 *
	 * @param name   file to be copied
	 * @param target file name of copy
	 */
	default void copy(@NotNull String name, @NotNull String target) throws IOException {
		try (InputStream from = download(name)) {
			OutputStream to = upload(target);
			//noinspection TryFinallyCanBeTryWithResources
			try {
				LocalFileUtils.copy(from, to);
			} finally {
				from.close();
				to.close();
			}
		}
	}

	default void copyAll(Map sourceToTarget) throws IOException {
		checkArgument(isBijection(sourceToTarget), "Targets must be unique");
		for (Map.Entry entry : sourceToTarget.entrySet()) {
			copy(entry.getKey(), entry.getValue());
		}
	}

	default void move(@NotNull String name, @NotNull String target) throws IOException {
		copy(name, target);
		if (!name.equals(target)) {
			delete(name);
		}
	}

	default void moveAll(Map sourceToTarget) throws IOException {
		checkArgument(isBijection(sourceToTarget), "Targets must be unique");
		for (Map.Entry entry : sourceToTarget.entrySet()) {
			copy(entry.getKey(), entry.getValue());
		}
		deleteAll(sourceToTarget.entrySet().stream()
				.filter(entry -> !entry.getKey().equals(entry.getValue()))
				.map(Map.Entry::getKey)
				.collect(toSet()));
	}

	Map list(@NotNull String glob) throws IOException;

	@Nullable
	default FileMetadata info(@NotNull String name) throws IOException {
		return list(escapeGlob(name)).get(name);
	}

	default Map infoAll(@NotNull Set names) throws IOException {
		Map result = new LinkedHashMap<>();
		for (String name : names) {
			FileMetadata info = info(name);
			if (info != null) {
				result.put(name, info);
			}
		}
		return result;
	}

	default void ping() throws IOException {
		list("");
	}

}