/*
 * 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.facebook.presto;

public final class SystemSessionProperties
{
    public static final String BIG_QUERY = "experimental_big_query";
    private static final String OPTIMIZE_HASH_GENERATION = "optimize_hash_generation";
    private static final String DISTRIBUTED_JOIN = "distributed_join";
    private static final String HASH_PARTITION_COUNT = "hash_partition_count";

    private SystemSessionProperties() {}

    public static boolean isBigQueryEnabled(Session session, boolean defaultValue)
    {
        return isEnabled(BIG_QUERY, session, defaultValue);
    }

    private static boolean isEnabled(String propertyName, Session session, boolean defaultValue)
    {
        String enabled = session.getSystemProperties().get(propertyName);
        if (enabled == null) {
            return defaultValue;
        }

        return Boolean.valueOf(enabled);
    }

    private static int getNumber(String propertyName, Session session, int defaultValue)
    {
        String count = session.getSystemProperties().get(propertyName);
        if (count != null) {
            try {
                return Integer.valueOf(count);
            }
            catch (NumberFormatException ignored) {
            }
        }

        return defaultValue;
    }

    public static boolean isOptimizeHashGenerationEnabled(Session session, boolean defaultValue)
    {
        return isEnabled(OPTIMIZE_HASH_GENERATION, session, defaultValue);
    }

    public static boolean isDistributedJoinEnabled(Session session, boolean defaultValue)
    {
        return isEnabled(DISTRIBUTED_JOIN, session, defaultValue);
    }

    public static int getHashPartitionCount(Session session, int defaultValue)
    {
        return getNumber(HASH_PARTITION_COUNT, session, defaultValue);
    }
}

/*
 * 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.facebook.presto;

public final class SystemSessionProperties
{
    public static final String BIG_QUERY = "experimental_big_query";
    private static final String OPTIMIZE_HASH_GENERATION = "optimize_hash_generation";
    private static final String DISTRIBUTED_JOIN = "distributed_join";
    private static final String HASH_PARTITION_COUNT = "hash_partition_count";
    private static final String PREFER_STREAMING_OPERATORS = "prefer_streaming_operators";

    private SystemSessionProperties() {}

    public static boolean isBigQueryEnabled(Session session, boolean defaultValue)
    {
        return isEnabled(BIG_QUERY, session, defaultValue);
    }

    private static boolean isEnabled(String propertyName, Session session, boolean defaultValue)
    {
        String enabled = session.getSystemProperties().get(propertyName);
        if (enabled == null) {
            return defaultValue;
        }

        return Boolean.valueOf(enabled);
    }

    private static int getNumber(String propertyName, Session session, int defaultValue)
    {
        String count = session.getSystemProperties().get(propertyName);
        if (count != null) {
            try {
                return Integer.valueOf(count);
            }
            catch (NumberFormatException ignored) {
            }
        }

        return defaultValue;
    }

    public static boolean isOptimizeHashGenerationEnabled(Session session, boolean defaultValue)
    {
        return isEnabled(OPTIMIZE_HASH_GENERATION, session, defaultValue);
    }

    public static boolean isDistributedJoinEnabled(Session session, boolean defaultValue)
    {
        return isEnabled(DISTRIBUTED_JOIN, session, defaultValue);
    }

    public static int getHashPartitionCount(Session session, int defaultValue)
    {
        return getNumber(HASH_PARTITION_COUNT, session, defaultValue);
    }

    public static boolean preferStreamingOperators(Session session, boolean defaultValue)
    {
        return isEnabled(PREFER_STREAMING_OPERATORS, session, defaultValue);
    }
}

Add session property to prefer streaming operators