CalciteParserFactory.java

Go to the documentation of this file.

/*
 * Copyright 2022 HEAVY.AI, 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 com.mapd.parser.server;

import com.mapd.calcite.parser.HeavyDBParser;
import com.mapd.calcite.parser.HeavyDBSqlOperatorTable;
import com.mapd.common.SockTransportProperties;

import org.apache.calcite.sql.fun.SqlStdOperatorTable;
import org.apache.calcite.util.ConversionUtil;
import org.apache.commons.pool.PoolableObjectFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Map;
import java.util.function.Supplier;

class CalciteParserFactory implements PoolableObjectFactory {
  final static Logger HEAVYDBLOGGER = LoggerFactory.getLogger(CalciteParserFactory.class);

  private final String dataDir;
  private final Map<String, ExtensionFunction> extSigs;
  private final int dbPort;
  private final SockTransportProperties socket_transport_properties;
  private volatile HeavyDBSqlOperatorTable tableOperator;
  private final Supplier<HeavyDBSqlOperatorTable> tableOperatorSupplier =
          new Supplier<HeavyDBSqlOperatorTable>() {
            @Override
            public HeavyDBSqlOperatorTable get() {
              return tableOperator;
            }
          };

  static {
    System.setProperty(
            "saffron.default.charset", ConversionUtil.NATIVE_UTF16_CHARSET_NAME);
    System.setProperty(
            "saffron.default.nationalcharset", ConversionUtil.NATIVE_UTF16_CHARSET_NAME);
    System.setProperty("saffron.default.collation.name",
            ConversionUtil.NATIVE_UTF16_CHARSET_NAME + "$en_US");
  }

  public CalciteParserFactory(String dataDir,
          final Map<String, ExtensionFunction> extSigs,
          int dbPort,
          SockTransportProperties skT) {
    this.dataDir = dataDir;
    this.extSigs = extSigs;
    this.dbPort = dbPort;
    this.socket_transport_properties = skT;

    updateOperatorTable();
  }

  public void updateOperatorTable() {
    HeavyDBSqlOperatorTable tableOperator =
            new HeavyDBSqlOperatorTable(SqlStdOperatorTable.instance());
    tableOperator.addUDF(extSigs);
    this.tableOperator = tableOperator;
  }

  @Override
  public Object makeObject() throws Exception {
    HeavyDBParser obj = new HeavyDBParser(
            dataDir, tableOperatorSupplier, dbPort, socket_transport_properties);
    return obj;
  }

  @Override
  public void destroyObject(Object obj) throws Exception {
    // no need to do anything
  }

  @Override
  public boolean validateObject(Object obj) {
    HeavyDBParser mdp = (HeavyDBParser) obj;
    if (mdp.getCallCount() < 1000) {
      return true;
    } else {
      HEAVYDBLOGGER.debug(" invalidating object due to max use count");
      return false;
    }
  }

  @Override
  public void activateObject(Object obj) throws Exception {
    // don't need to do anything
  }

  @Override
  public void passivateObject(Object obj) throws Exception {
    // nothing to currently do here
  }
}