com.cloudera.exhibit.core.composite.UpdatableExhibit.java Source code

Introduction

Here is the source code for com.cloudera.exhibit.core.composite.UpdatableExhibit.java

Source

/*
 * Copyright (c) 2015, Cloudera, Inc. All Rights Reserved.
 *
 * Cloudera, Inc. 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
 *
 * This software 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.cloudera.exhibit.core.composite;

import com.cloudera.exhibit.core.Exhibit;
import com.cloudera.exhibit.core.ExhibitDescriptor;
import com.cloudera.exhibit.core.Frame;
import com.cloudera.exhibit.core.Obs;
import com.google.common.collect.Maps;

import java.util.Map;

public class UpdatableExhibit implements Exhibit {

    private final Exhibit base;
    private final Map<String, Frame> frames;
    private UpdatableExhibitDescriptor descriptor;

    public UpdatableExhibit(Exhibit base) {
        this.base = base;
        this.frames = Maps.newHashMap();
        this.descriptor = new UpdatableExhibitDescriptor(base.descriptor());
    }

    public UpdatableExhibit add(String name, Frame frame) {
        this.frames.put(name, frame);
        this.descriptor.add(name, frame.descriptor());
        return this;
    }

    public UpdatableExhibit addAll(Map<String, Frame> frames) {
        for (Map.Entry<String, Frame> e : frames.entrySet()) {
            add(e.getKey(), e.getValue());
        }
        return this;
    }

    @Override
    public ExhibitDescriptor descriptor() {
        return descriptor;
    }

    @Override
    public Obs attributes() {
        return base.attributes();
    }

    @Override
    public Map<String, Frame> frames() {
        Map<String, Frame> union = Maps.newHashMap();
        union.putAll(base.frames());
        union.putAll(frames);
        return union;
    }
}