Add EXPORT MODEL syntax

Author: taewhi

traindb-core/src/main/antlr4/traindb/sql/TrainDBSql.g4

@@ -41,6 +41,7 @@ traindbStmts
  | bypassDdlStmt
  | deleteQueryLogs
  | deleteTasks
+ | exportModel
  ;
 
 createModeltype
@@ -114,6 +115,10 @@ optionValue
  | NUMERIC_LITERAL
  ;
 
+exportModel
+ : K_EXPORT K_MODEL modelName
+ ;
+
 showStmt
  : K_SHOW showTargets showWhereClause?
  ;
@@ -219,6 +224,7 @@ K_DELETE : D E L E T E ;
 K_DESC : D E S C ;
 K_DESCRIBE : D E S C R I B E ;
 K_DROP : D R O P ;
+K_EXPORT : E X P O R T ;
 K_FOR : F O R ;
 K_FROM : F R O M ;
 K_HYPERPARAMETERS : H Y P E R P A R A M E T E R S ;

traindb-core/src/main/java/traindb/engine/TrainDBQueryEngine.java

@@ -681,6 +681,11 @@ public void deleteTasks(Integer cnt) throws Exception {
  catalogContext.deleteTasks(cnt);
  }
 
+ @Override
+ public TrainDBListResultSet exportModel(String modelName) throws Exception {
+ throw new TrainDBException("Not supported yet");
+ }
+
  private void checkShowWhereColumns(Map<String, Object> patterns, List<String> columns)
  throws TrainDBException {
  for (String key : patterns.keySet()) {

traindb-core/src/main/java/traindb/sql/TrainDBSql.java

@@ -135,6 +135,9 @@ public static TrainDBListResultSet run(TrainDBSqlCommand command, TrainDBSqlRunn
  TrainDBSqlDeleteTasks deleteTasks = (TrainDBSqlDeleteTasks) command;
  runner.deleteTasks(deleteTasks.getRowCount());
  break;
+ case EXPORT_MODEL:
+ TrainDBSqlExportModel exportModel = (TrainDBSqlExportModel) command;
+ return runner.exportModel(exportModel.getModelName());
  default:
  throw new RuntimeException("invalid TrainDB SQL command");
  }
@@ -292,6 +295,13 @@ public void exitDescribeTable(TrainDBSqlParser.DescribeTableContext ctx) {
  commands.add(new TrainDBSqlDescribeTable(schemaName, tableName));
  }
 
+ @Override
+ public void exitExportModel(TrainDBSqlParser.ExportModelContext ctx) {
+ String modelName = ctx.modelName().getText();
+ LOG.debug("EXPORT MODEL: name=" + modelName);
+ commands.add(new TrainDBSqlExportModel(modelName));
+ }
+
  @Override
  public void exitBypassDdlStmt(TrainDBSqlParser.BypassDdlStmtContext ctx) {
  int start = ctx.ddlString().getStart().getStartIndex();

traindb-core/src/main/java/traindb/sql/TrainDBSqlCommand.java

@@ -38,6 +38,7 @@ public enum Type {
  SHOW_TASKS,
  DELETE_QUERY_LOGS,
  DELETE_TASKS,
+ EXPORT_MODEL,
  OTHER
  }
 }

traindb-core/src/main/java/traindb/sql/TrainDBSqlExportModel.java

@@ -0,0 +1,32 @@
+/*
+ * 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 traindb.sql;
+
+class TrainDBSqlExportModel extends TrainDBSqlCommand {
+ private final String modelName;
+
+ TrainDBSqlExportModel(String modelName) {
+ this.modelName = modelName;
+ }
+
+ String getModelName() {
+ return modelName;
+ }
+
+ @Override
+ public Type getType() {
+ return Type.EXPORT_MODEL;
+ }
+}

traindb-core/src/main/java/traindb/sql/TrainDBSqlRunner.java

@@ -65,5 +65,6 @@ void trainModel(String modeltypeName, String modelName, String schemaName, Strin
  void deleteQueryLogs(Integer cnt) throws Exception;
 
  void deleteTasks(Integer cnt) throws Exception;
-}
 
+ TrainDBListResultSet exportModel(String modelName) throws Exception;
+}