io.gravitee.gateway.policy.impl.PolicyMetadataBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gravitee-gateway-policy Show documentation
The newest version!
/**
 * Copyright (C) 2015 The Gravitee team (http://gravitee.io)
 *
 * 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 io.gravitee.gateway.policy.impl;

import io.gravitee.gateway.policy.PolicyMetadata;
import io.gravitee.gateway.policy.StreamType;
import io.gravitee.policy.api.PolicyConfiguration;
import io.gravitee.policy.api.PolicyContext;
import io.gravitee.policy.api.annotations.OnRequest;
import io.gravitee.policy.api.annotations.OnRequestContent;
import io.gravitee.policy.api.annotations.OnResponse;
import io.gravitee.policy.api.annotations.OnResponseContent;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.Map;

/**
 * @author David BRASSELY (david.brassely at graviteesource.com)
 * @author GraviteeSource Team
 */
public class PolicyMetadataBuilder {

    private String id;

    private Class configuration;

    private PolicyContext context;

    private Class policy;

    private Map, Method> methods;

    private ClassLoader classLoader;

    public PolicyMetadataBuilder setId(String id) {
        this.id = id;
        return this;
    }

    public PolicyMetadataBuilder setConfiguration(Class configuration) {
        this.configuration = configuration;
        return this;
    }

    public PolicyMetadataBuilder setContext(PolicyContext context) {
        this.context = context;
        return this;
    }

    public PolicyMetadataBuilder setPolicy(Class policy) {
        this.policy = policy;
        return this;
    }

    public PolicyMetadataBuilder setMethods(Map, Method> methods) {
        this.methods = methods;
        return this;
    }

    public PolicyMetadataBuilder setClassLoader(ClassLoader classLoader) {
        this.classLoader = classLoader;
        return this;
    }

    public PolicyMetadata build() {
        return new PolicyMetadata() {
            @Override
            public String id() {
                return id;
            }

            @Override
            public Class policy() {
                return policy;
            }

            @Override
            public Class configuration() {
                return configuration;
            }

            @Override
            public PolicyContext context() {
                return context;
            }

            @Override
            public ClassLoader classloader() {
                return classLoader;
            }

            @Override
            public Method method(Class type) {
                return methods.get(type);
            }

            @Override
            public boolean accept(StreamType stream) {
                return (
                    (stream == StreamType.ON_REQUEST && (method(OnRequest.class) != null || method(OnRequestContent.class) != null)) ||
                    (stream == StreamType.ON_RESPONSE && (method(OnResponse.class) != null || method(OnResponseContent.class) != null))
                );
            }
        };
    }
}