content: /*
   * Licensed to the Apache Software Foundation (ASF) under one
   * or more contributor license agreements.  See the NOTICE file
   * distributed with this work for additional information
   * regarding copyright ownership.  The ASF licenses this file
   * to you 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 org.apache.asterix.external.operators;
  
  import java.io.File;
  import java.io.IOException;
  import java.nio.file.Files;
  import java.nio.file.Path;
  import java.nio.file.StandardCopyOption;
  
  import org.apache.asterix.common.api.INcApplicationContext;
  import org.apache.asterix.common.library.ILibraryManager;
  import org.apache.asterix.common.metadata.Namespace;
  import org.apache.asterix.external.library.ExternalLibraryManager;
  import org.apache.hyracks.api.comm.IFrameWriter;
  import org.apache.hyracks.api.context.IHyracksTaskContext;
  import org.apache.hyracks.api.dataflow.value.RecordDescriptor;
  import org.apache.hyracks.api.exceptions.HyracksDataException;
  import org.apache.hyracks.api.io.FileReference;
  import org.apache.hyracks.api.io.IIOManager;
  import org.apache.hyracks.api.job.IOperatorDescriptorRegistry;
  import org.apache.hyracks.api.util.IoUtil;
  import org.apache.hyracks.cloud.io.ICloudIOManager;
  import org.apache.hyracks.dataflow.std.base.AbstractOperatorNodePushable;
  import org.apache.hyracks.dataflow.std.base.AbstractSingleActivityOperatorDescriptor;
  
  abstract class AbstractLibraryOperatorDescriptor extends AbstractSingleActivityOperatorDescriptor {
  
      private static final long serialVersionUID = 1L;
  
      protected final Namespace namespace;
  
      protected final String libraryName;
  
      public AbstractLibraryOperatorDescriptor(IOperatorDescriptorRegistry spec, Namespace namespace,
              String libraryName) {
          super(spec, 0, 0);
          this.namespace = namespace;
          this.libraryName = libraryName;
      }
  
      protected abstract class AbstractLibraryNodePushable extends AbstractOperatorNodePushable {
  
          protected final IHyracksTaskContext ctx;
  
          protected IIOManager ioManager;
  
          protected ICloudIOManager cloudIoManager;
  
          protected ILibraryManager libraryManager;
  
          private FileReference libraryDir;
  
          protected boolean cloudMode = false;
  
          protected AbstractLibraryNodePushable(IHyracksTaskContext ctx) {
              this.ctx = ctx;
          }
  
          protected abstract void execute() throws IOException;
  
          @Override
          public final void initialize() throws HyracksDataException {
              INcApplicationContext runtimeCtx =
                      (INcApplicationContext) ctx.getJobletContext().getServiceContext().getApplicationContext();
              ioManager = runtimeCtx.getPersistenceIoManager();
              libraryManager = runtimeCtx.getLibraryManager();
              libraryDir = libraryManager.getLibraryDir(namespace, libraryName);
              if (runtimeCtx.isCloudDeployment()) {
                  cloudMode = true;
                  cloudIoManager = (ICloudIOManager) ioManager;
              }
              try {
                  execute();
              } catch (IOException e) {
                  throw HyracksDataException.create(e);
              }
          }
  
          protected FileReference getLibraryDir() {
              return libraryDir;
          }
  
          protected FileReference getRev0Dir() {
              return libraryDir.getChild(ExternalLibraryManager.REV_0_DIR_NAME);
          }
  
          protected FileReference getRev1Dir() {
              return libraryDir.getChild(ExternalLibraryManager.REV_1_DIR_NAME);
          }
  
          protected FileReference getStageDir() {
              return libraryDir.getChild(ExternalLibraryManager.STAGE_DIR_NAME);
          }
  
          // does not flush any directories
          protected void dropIfExists(FileReference fileRef) throws HyracksDataException {
              if (fileRef.getFile().exists()) {
                  libraryManager.dropLibraryPath(fileRef);
              }
          }
  
          protected void move(FileReference src, FileReference dest) throws IOException {
              dropIfExists(dest);
              Files.move(src.getFile().toPath(), dest.getFile().toPath(), StandardCopyOption.ATOMIC_MOVE);
          }
  
          protected void mkdir(FileReference dir) throws IOException {
              Files.createDirectory(dir.getFile().toPath());
          }
  
          protected void flushDirectory(FileReference dir) throws IOException {
              flushDirectory(dir.getFile());
          }
  
          protected void flushDirectory(File dir) throws IOException {
              IoUtil.flushDirectory(dir);
          }
  
          protected void flushDirectory(Path dir) throws IOException {
              IoUtil.flushDirectory(dir);
          }
  
          protected void closeLibrary() throws HyracksDataException {
              libraryManager.closeLibrary(namespace, libraryName);
          }
  
          @Override
          public final void deinitialize() {
          }
  
          @Override
          public final int getInputArity() {
              return 0;
          }
  
          @Override
          public final void setOutputFrameWriter(int index, IFrameWriter writer, RecordDescriptor recordDesc) {
          }
  
          @Override
          public IFrameWriter getInputFrameWriter(int index) {
              return null;
          }
      }
  }
  