com.alipay.sofa.koupleless.ext.web.gateway.ForwardItems Maven / Gradle / Ivy

Go to download
/*
 * 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 com.alipay.sofa.koupleless.ext.web.gateway;

import lombok.AccessLevel;
import lombok.NoArgsConstructor;
import org.springframework.util.CollectionUtils;

import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;

/**
 * ForwardItems class.
 *
 * @author zzl_i
 * @version 1.0.0
 */
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class ForwardItems {
    private static final String EMPTY               = "";
    private static final String CONTEXT_PATH_PREFIX = "/";

    private static final String PATH_PREFIX         = "/";

    /**
     * init.
     *
     * @param bean a {@link com.alipay.sofa.koupleless.ext.web.gateway.Forwards} object
     * @param gatewayProperties a {@link com.alipay.sofa.koupleless.ext.web.gateway.GatewayProperties} object
     */
    public static void init(Forwards bean, GatewayProperties gatewayProperties) {
        //load conf
        List forwards = gatewayProperties.getForwards();
        List items = toForwardItems(forwards);
        items.sort(ForwardItemComparator.getInstance());
        //make forwards bean and set conf
        bean.setItems(items);
    }

    private static List toForwardItems(List forwards) {
        if (CollectionUtils.isEmpty(forwards)) {
            return Collections.emptyList();
        }
        List items = new LinkedList<>();
        for (Forward forward : forwards) {
            items.addAll(toForwardItems(forward));
        }
        return items;
    }

    private static List toForwardItems(Forward forward) {
        Set hosts = forward.getHosts();
        if (CollectionUtils.isEmpty(hosts)) {
            hosts = Collections.singleton(EMPTY);
        }
        String contextPath = forward.getContextPath();
        if (!contextPath.startsWith(CONTEXT_PATH_PREFIX)) {
            contextPath = CONTEXT_PATH_PREFIX + contextPath;
        }
        Set paths = forward.getPaths();
        if (CollectionUtils.isEmpty(paths)) {
            ForwardPath path = new ForwardPath();
            path.setFrom(PATH_PREFIX);
            path.setTo(PATH_PREFIX);
            paths = Collections.singleton(path);
        } else {
            for (ForwardPath path : paths) {
                String from = path.getFrom();
                if (!from.startsWith(CONTEXT_PATH_PREFIX)) {
                    path.setFrom(CONTEXT_PATH_PREFIX + from);
                }
                String to = path.getTo();
                if (to == null) {
                    path.setTo(path.getFrom());
                    continue;
                }
                if (!to.startsWith(CONTEXT_PATH_PREFIX)) {
                    path.setTo(CONTEXT_PATH_PREFIX + to);
                }
            }
        }
        List items = new LinkedList<>();
        for (String host : hosts) {
            for (ForwardPath path : paths) {
                ForwardItem item = new ForwardItem(contextPath, host, path.getFrom(), path.getTo());
                items.add(item);
            }
        }
        return items;
    }
}