com.jetdrone.vertx.yoke.core.Context Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yoke Show documentation
There is a newer version: 3.0.0
/**
 * Copyright 2011-2014 the original author or authors.
 */
package com.jetdrone.vertx.yoke.core;

import org.jetbrains.annotations.NotNull;

import java.util.*;

/** # Context
 *
 * Context represents the state of a request. Internally it is a Map that allows read write operations and under it
 * there is the application context that only allows read operations. Adding elements to the context with the same
 * name as entries in the application context, shadows the application context entries.
 */
public final class Context implements Map {

    /**
     * ReadOnly sub context (Application Context)
     */
    private final Map ro;
    private Map rw;

    /**
     * Create a new Context with the given Read Only sub context
     *
     * @param ro parent context
     */
    public Context(@NotNull Map ro) {
        this.ro = ro;
    }

    @Override
    public int size() {
        return ro.size() + (rw == null ? 0 : rw.size());
    }

    @Override
    public boolean isEmpty() {
        return ro.isEmpty() || rw == null || rw.isEmpty();
    }

    @Override
    public boolean containsKey(@NotNull Object o) {
        if (rw != null) {
            if (rw.containsKey(o)) {
                return true;
            }
        }

        return ro.containsKey(o);
    }

    @Override
    public boolean containsValue(Object o) {
        if (rw != null) {
            if (rw.containsValue(o)) {
                return true;
            }
        }

        return ro.containsValue(o);
    }

    @Override
    public Object get(@NotNull Object o) {
        if (rw != null) {
            if (rw.containsKey(o)) {
                return rw.get(o);
            }
        }

        return ro.get(o);
    }

    @Override
    public Object put(@NotNull String s, Object o) {
        if (rw == null) {
            rw = new LinkedHashMap<>();
        }
        return rw.put(s, o);
    }

    @Override
    public Object remove(@NotNull Object o) {
        if (rw == null) {
            return null;
        }
        return rw.remove(o);
    }

    @Override
    public void putAll(@NotNull Map map) {
        if (rw == null) {
            rw = new LinkedHashMap<>();
        }
        rw.putAll(map);
    }

    @Override
    public void clear() {
        if (rw != null) {
            rw.clear();
        }
    }

    @Override
    public @NotNull Set keySet() {
        if (rw != null) {
            Set keys = new HashSet<>(ro.keySet());
            for (String entry : rw.keySet()) {
                // if already present replace
                keys.add(entry);
            }
            return keys;
        }

        return ro.keySet();
    }

    @Override
    public @NotNull Collection