me.phoboslabs.illuminati.common.dto.RequestGeneralModel Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2017 Phoboslabs.me
 *
 * 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 me.phoboslabs.illuminati.common.dto;

import com.google.gson.annotations.Expose;
import me.phoboslabs.illuminati.common.util.StringObjectUtils;
import org.apache.commons.collections.MapUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.*;

/**
 * Created by leekyoungil ([email protected]) on 10/07/2017.
 */
public class RequestGeneralModel {

    private static final Logger REQUEST_GENERAL_MODEL_LOGGER = LoggerFactory.getLogger(RequestGeneralModel.class);

    @Expose private String clientIp;
    @Expose private String methodName;
    @Expose private Map methodParams;
    @Expose private String path;
    @Expose private String anotherPath;
    @Expose private String queryString;

    private transient Method method;

    private static final List CLIENT_INFO_KEY_LIST = Collections.unmodifiableList(Arrays.asList("path", "queryString", "clientIp", "anotherPath"));

    public RequestGeneralModel () {}

    public RequestGeneralModel (final Map requestMap) throws Exception {
        if (MapUtils.isEmpty(requestMap)) {
            final String errorMessage = "Sorry. check your requestMap variable.";
            REQUEST_GENERAL_MODEL_LOGGER.error(errorMessage);
            throw new Exception(errorMessage);
        }

        for (Map.Entry entry : requestMap.entrySet() ) {
            if (entry.getValue() == null) {
                REQUEST_GENERAL_MODEL_LOGGER.debug("Sorry. check your requestMap "+entry.getKey()+" key variable.");
                continue;
            }

            try {
                final Field field = this.getClass().getDeclaredField(entry.getKey());
                field.setAccessible(true);

                if ("serverPort".equals(entry.getKey())) {
                    field.setInt(this, (Integer) entry.getValue());
                } else {
                    field.set(this, entry.getValue().toString());
                }
            } catch (NoSuchFieldException ex) {
                final String errorMessage = "Sorry. check your class field. ("+ex.toString()+")";
                REQUEST_GENERAL_MODEL_LOGGER.error(errorMessage, ex);
                throw new Exception(errorMessage);
            } catch (IllegalAccessException ex) {
                final String errorMessage = "Sorry. check your class field permission. ("+ex.toString()+")";
                REQUEST_GENERAL_MODEL_LOGGER.error(errorMessage, ex);
                throw new Exception(errorMessage);
            }
        }
    }

    private static final String CLIENT_IP_KEYWORD = "clientIp";
    private static final String REMOTE_ADDR_KEYWORD = "remoteAddr";

    public void initClientInfo (final Map clientInfoMap) {
        if (MapUtils.isEmpty(clientInfoMap)) {
            return;
        }

        for (String key : CLIENT_INFO_KEY_LIST) {
            if (!clientInfoMap.containsKey(key)) {
                continue;
            }
            try {
                String value = clientInfoMap.get(key);

                if (CLIENT_IP_KEYWORD.equals(key) && !StringObjectUtils.isValid(value)) {
                    value = clientInfoMap.get(REMOTE_ADDR_KEYWORD);
                }

                if (StringObjectUtils.isValid(value)) {
                    final Field field = this.getClass().getDeclaredField(key);
                    field.setAccessible(true);
                    field.set(this, value);
                }
            } catch (Exception ignore) {}
        }
    }

    public String getServerIp() {
        return this.getServerIp();
    }

    private boolean isParamValidated(final String[] paramNames, final Object[] paramValues) {
        return paramNames != null && paramNames.length > 0 && paramValues != null && paramValues.length > 0;
    }

    public void setMethod(final Method method, final String[] paramNames, final Object[] paramValues) {
        this.method = method;
        this.methodName = this.method.toString();

        try {
            if (this.isParamValidated(paramNames, paramValues)) {
                Map paramMap = new HashMap<>();
                for (int i=0; i -1
                && (this.path.indexOf(".dispatch") + 9) == this.path.length()) {
            this.path = this.path.replace("/grails", "");
            this.path = this.path.substring(0, this.path.indexOf(".dispatch"));
        }
    }

    public String getMethodName () {
        return this.methodName;
    }
}