com.netflix.zuul.stats.ZuulEvent Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zuul-core Show documentation
zuul-core
There is a newer version: 2.5.13
/*
 * Copyright 2018 Netflix, Inc.
 *
 *      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 com.netflix.zuul.stats;

import org.json.JSONException;
import org.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

/**
 * An event that can be published to an NFEventPublisher and consumed by an NFEventListener,
 * defined by String key-value pairs.
 *
 * @author mhawthorne
 */
public class ZuulEvent {


    private static final Logger LOG = LoggerFactory.getLogger(ZuulEvent.class);


    private static final String TYPE_KEY = "type";

    private final JSONObject json;

    public ZuulEvent() {
        this(new JSONObject());
    }

    public ZuulEvent(JSONObject json) {
        this.json = json;
    }

    @Deprecated
    public ZuulEvent setAttribute(String key, String value) {
        return this.set(key, value);
    }

    public ZuulEvent set(String key, Object value) {
        try {
            this.json.put(key, value);
        } catch (JSONException e) {
            throw new IllegalStateException(e);
        }
        return this;
    }

    public Object get(String key) {
        if(!this.has(key)) {
            return null;
        }

        try {
            return this.json.get(key);
        } catch (JSONException e) {
            LOG.debug(e.getMessage(),e);
            return null;
        }
    }

    public boolean has(String key) {
        return this.json.has(key);
    }

    public Iterator keys() {
        return this.json.keys();
    }

    public JSONObject toJson() {
        return this.json;
    }

    public Map toMap() {
        final Map m = new HashMap();
        for(final Iterator i = this.keys(); i.hasNext();) {
            final String key = i.next();
            final Object val = this.get(key);
            if (val != null) m.put(key, val);
        }
        return m;
    }

    public Map toStringMap() {
        final Map m = new HashMap();
        for(final Iterator i = this.keys(); i.hasNext();) {
            final String key = i.next();
            final Object val = this.get(key);
            if (val != null) m.put(key, val.toString());
        }
        return m;
    }

    @Override
    public String toString() {
        return String.format("%s%s", this.getClass().getSimpleName(), this.json);
    }

}