io.druid.query.BaseQuery.java Source code

Introduction

Here is the source code for io.druid.query.BaseQuery.java

Source

/*
 * Druid - a distributed column store.
 * Copyright 2012 - 2015 Metamarkets Group 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 io.druid.query;

import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Maps;
import com.metamx.common.ISE;
import com.metamx.common.guava.Sequence;
import io.druid.query.spec.QuerySegmentSpec;
import org.joda.time.Duration;
import org.joda.time.Interval;

import java.util.List;
import java.util.Map;

/**
 */
public abstract class BaseQuery<T> implements Query<T> {
    public static final String QUERYID = "queryId";
    private final DataSource dataSource;
    private final Map<String, Object> context;
    private final QuerySegmentSpec querySegmentSpec;
    private volatile Duration duration;

    public BaseQuery(DataSource dataSource, QuerySegmentSpec querySegmentSpec, Map<String, Object> context) {
        Preconditions.checkNotNull(dataSource, "dataSource can't be null");
        Preconditions.checkNotNull(querySegmentSpec, "querySegmentSpec can't be null");

        this.dataSource = dataSource;
        this.context = context;
        this.querySegmentSpec = querySegmentSpec;
    }

    @JsonProperty
    @Override
    public DataSource getDataSource() {
        return dataSource;
    }

    @JsonProperty("intervals")
    public QuerySegmentSpec getQuerySegmentSpec() {
        return querySegmentSpec;
    }

    @Override
    public Sequence<T> run(QuerySegmentWalker walker, Map<String, Object> context) {
        return run(querySegmentSpec.lookup(this, walker), context);
    }

    public Sequence<T> run(QueryRunner<T> runner, Map<String, Object> context) {
        return runner.run(this, context);
    }

    @Override
    public List<Interval> getIntervals() {
        return querySegmentSpec.getIntervals();
    }

    @Override
    public Duration getDuration() {
        if (duration == null) {
            Duration totalDuration = new Duration(0);
            for (Interval interval : querySegmentSpec.getIntervals()) {
                if (interval != null) {
                    totalDuration = totalDuration.plus(interval.toDuration());
                }
            }
            duration = totalDuration;
        }

        return duration;
    }

    @Override
    @JsonProperty
    public Map<String, Object> getContext() {
        return context;
    }

    @Override
    public <ContextType> ContextType getContextValue(String key) {
        return context == null ? null : (ContextType) context.get(key);
    }

    @Override
    public <ContextType> ContextType getContextValue(String key, ContextType defaultValue) {
        ContextType retVal = getContextValue(key);
        return retVal == null ? defaultValue : retVal;
    }

    @Override
    public int getContextPriority(int defaultValue) {
        if (context == null) {
            return defaultValue;
        }
        Object val = context.get("priority");
        if (val == null) {
            return defaultValue;
        }
        if (val instanceof String) {
            return Integer.parseInt((String) val);
        } else if (val instanceof Integer) {
            return (int) val;
        } else {
            throw new ISE("Unknown type [%s]", val.getClass());
        }
    }

    @Override
    public boolean getContextBySegment(boolean defaultValue) {
        return parseBoolean("bySegment", defaultValue);
    }

    @Override
    public boolean getContextPopulateCache(boolean defaultValue) {
        return parseBoolean("populateCache", defaultValue);
    }

    @Override
    public boolean getContextUseCache(boolean defaultValue) {
        return parseBoolean("useCache", defaultValue);
    }

    @Override
    public boolean getContextFinalize(boolean defaultValue) {
        return parseBoolean("finalize", defaultValue);
    }

    private boolean parseBoolean(String key, boolean defaultValue) {
        if (context == null) {
            return defaultValue;
        }
        Object val = context.get(key);
        if (val == null) {
            return defaultValue;
        }
        if (val instanceof String) {
            return Boolean.parseBoolean((String) val);
        } else if (val instanceof Boolean) {
            return (boolean) val;
        } else {
            throw new ISE("Unknown type [%s]. Cannot parse!", val.getClass());
        }
    }

    protected Map<String, Object> computeOverridenContext(Map<String, Object> overrides) {
        Map<String, Object> overridden = Maps.newTreeMap();
        final Map<String, Object> context = getContext();
        if (context != null) {
            overridden.putAll(context);
        }
        overridden.putAll(overrides);

        return overridden;
    }

    @Override
    public String getId() {
        return (String) getContextValue(QUERYID);
    }

    @Override
    public Query withId(String id) {
        return withOverriddenContext(ImmutableMap.<String, Object>of(QUERYID, id));
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }

        BaseQuery baseQuery = (BaseQuery) o;

        if (context != null ? !context.equals(baseQuery.context) : baseQuery.context != null) {
            return false;
        }
        if (dataSource != null ? !dataSource.equals(baseQuery.dataSource) : baseQuery.dataSource != null) {
            return false;
        }
        if (duration != null ? !duration.equals(baseQuery.duration) : baseQuery.duration != null) {
            return false;
        }
        if (querySegmentSpec != null ? !querySegmentSpec.equals(baseQuery.querySegmentSpec)
                : baseQuery.querySegmentSpec != null) {
            return false;
        }

        return true;
    }

    @Override
    public int hashCode() {
        int result = dataSource != null ? dataSource.hashCode() : 0;
        result = 31 * result + (context != null ? context.hashCode() : 0);
        result = 31 * result + (querySegmentSpec != null ? querySegmentSpec.hashCode() : 0);
        result = 31 * result + (duration != null ? duration.hashCode() : 0);
        return result;
    }
}