com.palantir.giraffe.file.base.attribute.DynamicAttributeAccessor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of giraffe-fs-base Show documentation
Palantir open source project
There is a newer version: 0.10.1
/**
 * Copyright 2015 Palantir Technologies, Inc.
 *
 * 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 com.palantir.giraffe.file.base.attribute;

import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import com.google.common.base.Throwables;
import com.google.common.collect.ImmutableSet;

/**
 * Reads annotations on {@link AnnotatedFileAttributeView}s to provide dynamic
 * access to file attributes.
 *
 * @author bkeyes
 */
public final class DynamicAttributeAccessor {

    private final AnnotatedFileAttributeView view;

    public DynamicAttributeAccessor(AnnotatedFileAttributeView view) {
        this.view = view;
    }

    public Map readAttributes(String attributes) throws IOException {
        ImmutableSet attributeSet = ImmutableSet.copyOf(attributes.split(","));
        if (attributeSet.contains("*")) {
            return readAllAttributes();
        } else {
            return readAttributes(attributeSet);
        }
    }

    public Map readAllAttributes() throws IOException {
        AnnotatedFileAttributes attributes = view.readAttributes();
        Map getters = AttributeMethodExtractor.getters(attributes.getClass());

        Map result = new HashMap<>();
        for (Entry e : getters.entrySet()) {
            result.put(e.getKey(), invoke(e.getValue(), attributes));
        }
        return result;
    }

    public Map readAttributes(Set names) throws IOException {
        AnnotatedFileAttributes attributes = view.readAttributes();
        Map getters = AttributeMethodExtractor.getters(attributes.getClass());

        Map result = new HashMap<>();
        for (String attribute : names) {
            Method getter = getters.get(attribute);
            if (getter != null) {
                result.put(attribute, invoke(getter, attributes));
            } else {
                throw new IllegalArgumentException("unrecognized attribute: '" + attribute + "'");
            }
        }
        return result;
    }

    public void setAttribute(String name, Object value) throws IOException {
        Map setters = AttributeMethodExtractor.setters(view.getClass());

        Method setter = setters.get(name);
        if (setter != null) {
            invoke(setter, view, value);
        } else {
            throw new IllegalArgumentException("unrecognized attribute: '" + name + "'");
        }
    }

    private static Object invoke(Method m, Object obj, Object... args) throws IOException {
        try {
            return m.invoke(obj, args);
        } catch (IllegalAccessException e) {
            throw new IllegalStateException(methodString(m) + " is not accessible", e);
        } catch (InvocationTargetException e) {
            Throwables.propagateIfInstanceOf(e.getCause(), IOException.class);
            throw new IllegalStateException("error invoking " + methodString(m), e.getCause());
        }
    }

    private static String methodString(Method m) {
        return m.getDeclaringClass().getName() + "." + m.getName();
    }
}