CalciteServerWrapper.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.common.SockTransportProperties;

import org.apache.thrift.server.TServer;
import org.apache.thrift.server.TThreadPoolServer;
import org.apache.thrift.transport.TSSLTransportFactory;
import org.apache.thrift.transport.TSSLTransportFactory.TSSLTransportParameters;
import org.apache.thrift.transport.TServerSocket;
import org.apache.thrift.transport.TServerTransport;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import ai.heavy.thrift.calciteserver.CalciteServer.Processor;

public class CalciteServerWrapper implements Runnable {
  private final static Logger HEAVYDBLOGGER =
          LoggerFactory.getLogger(CalciteServerWrapper.class);
  private final CalciteServerHandler handler;
  private final Processor processor;
  private TServer server;
  private int heavyDBPort = 6274;
  private String dataDir = ("data/");
  private int calcitePort = 6279;
  private boolean shutdown = false;
  private SockTransportProperties server_skT_;

  public CalciteServerWrapper() {
    handler = new CalciteServerHandler(heavyDBPort, dataDir, null, null, "");
    processor = new ai.heavy.thrift.calciteserver.CalciteServer.Processor(handler);
  }

  public CalciteServerWrapper(int calcitePort,
          int heavyDBPort,
          String dataDir,
          String extensionFunctionsAstFile,
          SockTransportProperties client_skT,
          SockTransportProperties server_skT) {
    handler = new CalciteServerHandler(
            heavyDBPort, dataDir, extensionFunctionsAstFile, client_skT, "");
    processor = new ai.heavy.thrift.calciteserver.CalciteServer.Processor(handler);
    this.calcitePort = calcitePort;
    this.heavyDBPort = heavyDBPort;
    this.server_skT_ = server_skT;
  }

  public CalciteServerWrapper(int calcitePort,
          int heavyDBPort,
          String dataDir,
          String extensionFunctionsAstFile,
          SockTransportProperties client_skT,
          SockTransportProperties server_skT,
          String userDefinedFunctionsFile) {
    handler = new CalciteServerHandler(heavyDBPort,
            dataDir,
            extensionFunctionsAstFile,
            client_skT,
            userDefinedFunctionsFile);
    processor = new ai.heavy.thrift.calciteserver.CalciteServer.Processor(handler);
    this.calcitePort = calcitePort;
    this.heavyDBPort = heavyDBPort;
    this.server_skT_ = server_skT;
    try {
    } catch (Exception e) {
      e.printStackTrace();
    }
  }

  private void startServer(
          ai.heavy.thrift.calciteserver.CalciteServer.Processor processor) {
    try {
      TServerTransport serverTransport = server_skT_.openServerTransport(calcitePort);
      server = new TThreadPoolServer(
              new TThreadPoolServer.Args(serverTransport).processor(processor));

      HEAVYDBLOGGER.debug("Starting a threaded pool server... Listening on port "
              + calcitePort + " HEAVY.AI on port " + heavyDBPort);
      handler.setServer(server);
      server.serve();
      // we have been told to shut down (only way to get to this piece of code
      shutdown = true;

    } catch (Exception e) {
      e.printStackTrace();
      HEAVYDBLOGGER.error(" Calcite server Failed to start ");
      shutdown = true;
    }
  }

  public void stopServer() {
    server.stop();
    shutdown = true;
  }

  @Override
  public void run() {
    startServer(processor);
  }

  boolean shutdown() {
    return shutdown;
  }
}