org.apache.asterix.api.common.SessionConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of asterix-app Show documentation
There is a newer version: 0.9.9
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF 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
 *
 * 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 org.apache.asterix.api.common;

import java.io.PrintWriter;
import java.util.HashMap;
import java.util.Map;

/**
 * SessionConfig captures several different parameters for controlling
 * the execution of an APIFramework call.
 *  It specifies how the execution will proceed (for instance,
 * whether to optimize, or whether to execute at all).
 * 
 It allows you specify where the primary execution output will
 * be sent.
 * 
 It also allows you to request additional output for optional
 * out-of-band data about the execution (query plan, etc).
 * 
 It allows you to specify the output format for the primary
 * execution output - JSON, CSV, etc.
 *  It allows you to specify output format-specific parameters.
 */

public class SessionConfig {
    /**
     * Used to specify the output format for the primary execution.
     */
    public enum OutputFormat {
        ADM,
        JSON,
        CSV
    };

    /**
     * Produce out-of-band output for Hyracks Job.
     */
    public static final String OOB_HYRACKS_JOB = "oob-hyracks-job";

    /**
     * Produce out-of-band output for Expression Tree.
     */
    public static final String OOB_EXPR_TREE = "oob-expr-tree";

    /**
     * Produce out-of-band output for Rewritten Expression Tree.
     */
    public static final String OOB_REWRITTEN_EXPR_TREE = "oob-rewritten-expr-tree";

    /**
     * Produce out-of-band output for Logical Plan.
     */
    public static final String OOB_LOGICAL_PLAN = "oob-logical-plan";

    /**
     * Produce out-of-band output for Optimized Logical Plan.
     */
    public static final String OOB_OPTIMIZED_LOGICAL_PLAN = "oob-optimized-logical-plan";

    /**
     * Format flag: print only physical ops (for optimizer tests).
     */
    public static final String FORMAT_ONLY_PHYSICAL_OPS = "format-only-physical-ops";

    /**
     * Format flag: wrap out-of-band data in HTML.
     */
    public static final String FORMAT_HTML = "format-html";

    /**
     * Format flag: print CSV header line.
     */
    public static final String FORMAT_CSV_HEADER = "format-csv-header";

    // Standard execution flags.
    private final boolean executeQuery;
    private final boolean generateJobSpec;
    private final boolean optimize;

    // Output path for primary execution.
    private final PrintWriter out;

    // Output format.
    private final OutputFormat fmt;

    // Flags.
    private final Map flags;

    /**
     * Create a SessionConfig object with all default values:
     *
     * - All format flags set to "false".
     * - All out-of-band outputs set to "null".
     * - "Optimize" set to "true".
     * - "Execute Query" set to "true".
     * - "Generate Job Spec" set to "true".
     * @param out PrintWriter for execution output.
     * @param fmt Output format for execution output.
     */
    public SessionConfig(PrintWriter out, OutputFormat fmt) {
        this(out, fmt, true, true, true);
    }

    /**
     * Create a SessionConfig object with all optional values set to defaults:
     *
     * - All format flags set to "false".
     * - All out-of-band outputs set to "false".
     * @param out PrintWriter for execution output.
     * @param fmt Output format for execution output.
     * @param optimize Whether to optimize the execution.
     * @param executeQuery Whether to execute the query or not.
     * @param generateJobSpec Whether to generate the Hyracks job specification (if
     *    false, job cannot be executed).
     */
    public SessionConfig(PrintWriter out, OutputFormat fmt, boolean optimize, boolean executeQuery, boolean generateJobSpec) {
        this.out = out;
        this.fmt = fmt;
        this.optimize = optimize;
        this.executeQuery = executeQuery;
        this.generateJobSpec = generateJobSpec;
        this.flags = new HashMap();
    }

    /**
     * Retrieve the PrintWriter to produce output to.
     */
    public PrintWriter out() {
        return this.out;
    }

    /**
     * Retrieve the OutputFormat for this execution.
     */
    public OutputFormat fmt() {
        return this.fmt;
    }

    /**
     * Retrieve the value of the "execute query" flag.
     */
    public boolean isExecuteQuery() {
        return executeQuery;
    }

    /**
     * Retrieve the value of the "optimize" flag.
     */
    public boolean isOptimize() {
        return optimize;
    }

    /**
     * Retrieve the value of the "generate job spec" flag.
     */
    public boolean isGenerateJobSpec() {
        return generateJobSpec;
    }

    /**
     * Specify all out-of-band settings at once. For convenience of older code.
     */
    public void setOOBData(boolean expr_tree, boolean rewritten_expr_tree,
                           boolean logical_plan, boolean optimized_logical_plan,
                           boolean hyracks_job) {
        this.set(OOB_EXPR_TREE, expr_tree);
        this.set(OOB_REWRITTEN_EXPR_TREE, rewritten_expr_tree);
        this.set(OOB_LOGICAL_PLAN, logical_plan);
        this.set(OOB_OPTIMIZED_LOGICAL_PLAN, optimized_logical_plan);
        this.set(OOB_HYRACKS_JOB, hyracks_job);
    }

    /**
     * Specify a flag.
     * @param flag One of the OOB_ or FORMAT_ constants from this class.
     * @param value Value for the flag (all flags default to "false").
     */
    public void set(String flag, boolean value) {
        flags.put(flag, Boolean.valueOf(value));
    }

    /**
     * Retrieve the setting of a format-specific flag.
     * @param flag One of the FORMAT_ constants from this class.
     * @returns true or false (all flags default to "false").
     */
    public boolean is(String flag) {
        Boolean value = flags.get(flag);
        return value == null ? false : value.booleanValue();
    }
}