modules/core/src/test/java/org/apache/ignite/internal/processors/igfs/IgfsLocalSecondaryFileSystemTestAdapter.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.ignite.internal.processors.igfs;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.file.DirectoryStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.StandardOpenOption;
import java.nio.file.attribute.BasicFileAttributeView;
import java.nio.file.attribute.BasicFileAttributes;
import java.nio.file.attribute.PosixFileAttributeView;
import java.nio.file.attribute.PosixFileAttributes;
import java.nio.file.attribute.PosixFilePermission;
import java.util.HashMap;
import java.util.Map;
import org.apache.ignite.internal.util.typedef.T2;

/**
 * Adapter for local secondary file system.
 */
public class IgfsLocalSecondaryFileSystemTestAdapter implements IgfsSecondaryFileSystemTestAdapter {
    /** */
    private final String workDir;

    /**
     * @param workDir Work dir.
     */
    public IgfsLocalSecondaryFileSystemTestAdapter(final File workDir) {
        this.workDir = workDir.getAbsolutePath();
    }

    /** {@inheritDoc} */
    @Override public String name() throws IOException {
        return "local";
    }

    /** {@inheritDoc} */
    @Override public boolean exists(final String path) throws IOException {
        return Files.exists(path(path));
    }

    /** {@inheritDoc} */
    @Override public boolean delete(final String path, final boolean recursive) throws IOException {
        if (recursive)
            return deleteRecursively(path(path));
        else
            return path(path).toFile().delete();
    }

    /** {@inheritDoc} */
    @Override public void mkdirs(final String path) throws IOException {
        Files.createDirectory(path(path));
    }

    /** {@inheritDoc} */
    @Override public void format() throws IOException {
        try (DirectoryStream<Path> stream = Files.newDirectoryStream(Paths.get(workDir))) {
            for (Path innerPath : stream)
                deleteRecursively(innerPath);
        }
    }

    /** {@inheritDoc} */
    @Override public Map<String, String> properties(final String path) throws IOException {
        Path p = path(path);
        PosixFileAttributes attrs = Files.getFileAttributeView(p, PosixFileAttributeView.class).readAttributes();

        Map<String, String> props = new HashMap<>();
        props.put(IgfsUtils.PROP_USER_NAME, attrs.owner().getName());
        props.put(IgfsUtils.PROP_GROUP_NAME, attrs.group().getName());
        props.put(IgfsUtils.PROP_PERMISSION, permissions(path));

        return props;
    }

    /** {@inheritDoc} */
    @Override public String permissions(String path) throws IOException {
        Path p = path(path);
        PosixFileAttributeView attrView = Files.getFileAttributeView(p, PosixFileAttributeView.class);

        if (attrView == null)
            throw new UnsupportedOperationException("Posix file attributes not available");

        int perm = 0;
        for (PosixFilePermission pfp : attrView.readAttributes().permissions())
            perm |= (1 << 8 - pfp.ordinal());

        return '0' + Integer.toOctalString(perm);
    }

    /** {@inheritDoc} */
    @Override public InputStream openInputStream(final String path) throws IOException {
        return Files.newInputStream(path(path));
    }

    /** {@inheritDoc} */
    @Override public OutputStream openOutputStream(final String path, final boolean append) throws IOException {
        if (append)
            return Files.newOutputStream(path(path), StandardOpenOption.APPEND);
        else
            return Files.newOutputStream(path(path));
    }

    /** {@inheritDoc} */
    @Override public T2<Long, Long> times(String path) throws IOException {
        BasicFileAttributes attrs = Files.getFileAttributeView(path(path), BasicFileAttributeView.class)
            .readAttributes();

        return new T2<>(attrs.lastModifiedTime().toMillis(), attrs.lastAccessTime().toMillis());
    }

    /** {@inheritDoc} */
    @Override public IgfsEx igfs() {
        return null;
    }

    /**
     * Convert path.
     *
     * @param path String path.
     * @return Java File API path.
     */
    private Path path(String path) {
        return Paths.get(workDir + path);
    }

    /**
     * Delete recursively.
     *
     * @param path Path.
     * @throws IOException If failed.
     * @return {@code true} if the file is deleted successfully. {@code false} otherwise.
     */
    private boolean deleteRecursively(Path path) throws IOException {
        if (Files.isDirectory(path)) {
            try (DirectoryStream<Path> stream = Files.newDirectoryStream(path)) {
                for (Path innerPath : stream) {
                    boolean res = deleteRecursively(innerPath);

                    if (!res)
                        return false;
                }
            }
        }

        return path.toFile().delete();
    }
}