scouterx.webapp.framework.util.ZZ Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scouter-webapp Show documentation
There is a newer version: 2.20.0
/*
 *  Copyright 2015 the original author or authors.
 *  @https://github.com/scouter-project/scouter
 *
 *  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 scouterx.webapp.framework.util;

import org.apache.commons.lang3.StringUtils;
import scouter.util.StringUtil;
import scouterx.webapp.framework.configure.ConfigureAdaptor;
import scouterx.webapp.framework.configure.ConfigureManager;

import javax.servlet.http.HttpServletRequest;
import java.util.Arrays;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @author Gun Lee ([email protected]) on 2017. 8. 27.
 */
public class ZZ {
    private static final ConfigureAdaptor conf = ConfigureManager.getConfigure();
    private static final char BRACKET = '[';
    private static final char COMMA = ',';

    public static String getRequestIp(HttpServletRequest request) {
        if (StringUtils.isNotBlank(conf.getNetHttpApiAuthIpHeaderKey())) {
            String ipString = request.getHeader(conf.getNetHttpApiAuthIpHeaderKey());
            if (StringUtil.isEmpty(ipString)) {
                return request.getRemoteAddr();
            }
            if (ipString.contains(",")) {
                return StringUtil.split(ipString, ',')[0].trim();
            }

            StringUtil.split(ipString, ',');
            return ipString;
        } else {
            return request.getRemoteAddr();
        }
    }

    public static String ipByteToString(byte[] ip) {
        if (ip == null)
            return "0.0.0.0";
        try {
            StringBuilder sb = new StringBuilder();
            sb.append(ip[0] & 0xff);
            sb.append(".");
            sb.append(ip[1] & 0xff);
            sb.append(".");
            sb.append(ip[2] & 0xff);
            sb.append(".");
            sb.append(ip[3] & 0xff);
            return sb.toString();
        } catch (Throwable e) {
            return "0.0.0.0";
        }
    }

    public static String stripFirstLastBracket(String org) {
        if (org.charAt(0) == BRACKET) {
            return org.substring(1, org.length() - 1);
        } else {
            return org;
        }
    }

    /**
     * split string by comma and strip array bracket if exists
     * eg) String : "[aaa, bbb]" -> List["aaa", "bbb"]
     * eg) String : "aaa, bbb" -> List["aaa", "bbb"]
     * @param org
     * @return
     */
    public static List splitParam(String org) {
        org = stripFirstLastBracket(org);
        return Arrays.asList(StringUtils.split(org, COMMA));
    }

    public static Set splitParamStringSet(String org) {
        org = stripFirstLastBracket(org);
        String[] items = StringUtils.split(org, COMMA);
        return Arrays.stream(items).collect(Collectors.toSet());
    }

    public static List splitParamAsInteger(String org) {
        org = stripFirstLastBracket(org);
        String[] items = StringUtils.split(org, COMMA);
        return Arrays.stream(items).map(Integer::parseInt).collect(Collectors.toList());
    }

    public static List splitParamAsLong(String org) {
        org = stripFirstLastBracket(org);
        String[] items = StringUtils.split(org, COMMA);
        return Arrays.stream(items).map(Long::parseLong).collect(Collectors.toList());
    }

    public static Set splitParamAsIntegerSet(String org) {
        org = stripFirstLastBracket(org);
        String[] items = StringUtils.split(org, COMMA);
        return Arrays.stream(items).map(Integer::parseInt).collect(Collectors.toSet());
    }
}