com.lindar.wellrested.util.BasicExclusionStrategy Maven / Gradle / Ivy

Go to download
package com.lindar.wellrested.util;

import com.google.gson.ExclusionStrategy;
import com.google.gson.FieldAttributes;

import java.util.List;
import java.util.Set;

public class BasicExclusionStrategy implements ExclusionStrategy {

    private List fieldNames;
    private Set classNames;

    public BasicExclusionStrategy (List fieldNames) {
        this.fieldNames = fieldNames;
    }

    public BasicExclusionStrategy (Set classNames) {
        this.classNames = classNames;
    }

    public BasicExclusionStrategy (Set classNames, List fieldNames) {
        this.classNames = classNames;
        this.fieldNames = fieldNames;
    }

    public boolean shouldSkipClass(Class arg0) {
        return classNames != null && classNames.contains(arg0.getSimpleName());
    }

    public boolean shouldSkipField(FieldAttributes f) {
        return fieldNames != null && fieldNames.contains(f.getName());
    }
}