Feat: IMPORT SYNOPSIS ~ FROM FILE

Author: taewhi

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

@@ -166,7 +166,7 @@ exportToClause
  ;
 
 importSynopsis
- : K_IMPORT K_SYNOPSIS synopsisName K_FROM synopsisBinaryString
+ : K_IMPORT K_SYNOPSIS synopsisName K_FROM ( synopsisBinaryString | K_FILE filenameString )?
  ;
 
 analyzeSynopsis

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

@@ -1356,8 +1356,8 @@ public TrainDBListResultSet exportSynopsis(String synopsisName, String exportFil
  }
 
  @Override
- public TrainDBListResultSet importSynopsis(String synopsisName, String synopsisBinaryString)
- throws Exception {
+ public TrainDBListResultSet importSynopsis(String synopsisName, String synopsisBinaryString,
+  String importFilename) throws Exception {
  T_tracer.startTaskTracer("import synopsis " + synopsisName);
 
  T_tracer.openTaskTime("find : synopsis");
@@ -1372,7 +1372,12 @@ public TrainDBListResultSet importSynopsis(String synopsisName, String synopsisB
  T_tracer.closeTaskTime("SUCCESS");
 
  T_tracer.openTaskTime("decode synopsis binary string");
- byte[] zipBytes = Hex.decodeHex(synopsisBinaryString);
+ byte[] zipBytes;
+ if (importFilename != null) {
+ zipBytes = Files.readAllBytes(Paths.get(importFilename).toAbsolutePath());
+ } else {
+ zipBytes = Hex.decodeHex(synopsisBinaryString);
+ }
  byte[] exportMetadata = ZipUtils.extractZipEntry(zipBytes, "export_synopsis.json");
  if (exportMetadata == null) {
  throw new TrainDBException(

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

@@ -168,7 +168,7 @@ public static TrainDBListResultSet run(TrainDBSqlCommand command, TrainDBSqlRunn
  case IMPORT_SYNOPSIS:
  TrainDBSqlImportSynopsis importSynopsis = (TrainDBSqlImportSynopsis) command;
  return runner.importSynopsis(importSynopsis.getSynopsisName(),
- importSynopsis.getSynopsisBinaryString());
+ importSynopsis.getSynopsisBinaryString(), importSynopsis.getImportFilename());
  case ALTER_SYNOPSIS_RENAME: {
  TrainDBSqlAlterSynopsis alterSynopsis = (TrainDBSqlAlterSynopsis) command;
  runner.renameSynopsis(alterSynopsis.getSynopsisName(), alterSynopsis.getNewSynopsisName());
@@ -407,9 +407,16 @@ public void exitExportSynopsis(TrainDBSqlParser.ExportSynopsisContext ctx) {
  @Override
  public void exitImportSynopsis(TrainDBSqlParser.ImportSynopsisContext ctx) {
  String synopsisName = ctx.synopsisName().getText();
- String synopsisBinaryString = ctx.synopsisBinaryString().getText();
+ String synopsisBinaryString = null;
+ String importFilename = null;
+ if (ctx.synopsisBinaryString() != null) {
+ synopsisBinaryString = ctx.synopsisBinaryString().getText();
+ } else if (ctx.filenameString() != null) {
+ importFilename = ctx.filenameString().getText();
+ }
  LOG.debug("IMPORT SYNOPSIS: name=" + synopsisName);
- commands.add(new TrainDBSqlImportSynopsis(synopsisName, synopsisBinaryString));
+ commands.add(new TrainDBSqlImportSynopsis(synopsisName, synopsisBinaryString,
+ importFilename));
  }
 
  @Override

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

@@ -17,10 +17,13 @@
 class TrainDBSqlImportSynopsis extends TrainDBSqlCommand {
  private final String synopsisName;
  private final String synopsisBinaryString;
+ private final String importFilename;
 
- TrainDBSqlImportSynopsis(String synopsisName, String synopsisBinaryString) {
+ TrainDBSqlImportSynopsis(String synopsisName, String synopsisBinaryString,
+ String importFilename) {
  this.synopsisName = synopsisName;
  this.synopsisBinaryString = synopsisBinaryString;
+ this.importFilename = importFilename;
  }
 
  String getSynopsisName() {
@@ -31,6 +34,10 @@ String getSynopsisBinaryString() {
  return synopsisBinaryString;
  }
 
+ String getImportFilename() {
+ return importFilename;
+ }
+
  @Override
  public Type getType() {
  return Type.IMPORT_SYNOPSIS;

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

@@ -82,8 +82,8 @@ void createSynopsis(String synopsisName, String modelName, int limitRows, float
 
  TrainDBListResultSet exportSynopsis(String synopsisName, String exportFilename) throws Exception;
 
- TrainDBListResultSet importSynopsis(String synopsisName, String synopsisBinaryString)
- throws Exception;
+ TrainDBListResultSet importSynopsis(String synopsisName, String synopsisBinaryString,
+  String importFilename) throws Exception;
 
  void renameSynopsis(String synopsisName, String newSynopsisName) throws Exception;