io.gravitee.gateway.handlers.sharedpolicygroup.policy.SharedPolicyGroupPolicy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gravitee-apim-gateway-handlers-shared-policy-group Show documentation
The newest version!
/*
 * Copyright © 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.handlers.sharedpolicygroup.policy;

import io.gravitee.gateway.handlers.sharedpolicygroup.reactor.SharedPolicyGroupReactor;
import io.gravitee.gateway.handlers.sharedpolicygroup.registry.SharedPolicyGroupRegistry;
import io.gravitee.gateway.reactive.api.context.ContextAttributes;
import io.gravitee.gateway.reactive.api.context.base.BaseExecutionContext;
import io.gravitee.gateway.reactive.api.context.http.HttpPlainExecutionContext;
import io.gravitee.gateway.reactive.api.policy.http.HttpPolicy;
import io.gravitee.gateway.reactive.policy.HttpPolicyChain;
import io.reactivex.rxjava3.core.Completable;
import java.util.Optional;
import java.util.function.Supplier;
import lombok.extern.slf4j.Slf4j;

/**
 * @author Yann TAVERNIER (yann.tavernier at graviteesource.com)
 * @author GraviteeSource Team
 */
@Slf4j
public class SharedPolicyGroupPolicy implements HttpPolicy {

    public static final String POLICY_ID = "shared-policy-group-policy";

    private final String id;
    public final SharedPolicyGroupPolicyConfiguration policyConfiguration;

    public SharedPolicyGroupPolicy(String id, SharedPolicyGroupPolicyConfiguration policyConfiguration) {
        this.id = id;
        this.policyConfiguration = policyConfiguration;
    }

    @Override
    public String id() {
        return id;
    }

    @Override
    public Completable onRequest(HttpPlainExecutionContext ctx) {
        return getPolicyChain(ctx)
            .map(policyChain -> policyChain.execute(ctx))
            .orElseGet(warnNotFoundAndComplete(ctx.getAttribute(ContextAttributes.ATTR_ENVIRONMENT)));
    }

    @Override
    public Completable onResponse(HttpPlainExecutionContext ctx) {
        return getPolicyChain(ctx)
            .map(policyChain -> policyChain.execute(ctx))
            .orElseGet(warnNotFoundAndComplete(ctx.getAttribute(ContextAttributes.ATTR_ENVIRONMENT)));
    }

    protected Supplier warnNotFoundAndComplete(String environmentId) {
        return () -> {
            log.warn(
                "No Shared Policy Group found for id {} on environment {}",
                policyConfiguration.getSharedPolicyGroupId(),
                environmentId
            );
            return Completable.complete();
        };
    }

    private Optional getPolicyChain(BaseExecutionContext ctx) {
        final SharedPolicyGroupRegistry sharedPolicyGroupRegistry = ctx.getComponent(SharedPolicyGroupRegistry.class);
        return Optional
            .ofNullable(
                sharedPolicyGroupRegistry.get(
                    policyConfiguration.getSharedPolicyGroupId(),
                    ctx.getAttribute(ContextAttributes.ATTR_ENVIRONMENT)
                )
            )
            .map(SharedPolicyGroupReactor::policyChain);
    }
}