ZEPPELIN-598 ] Dynamic loading for Interpreter and API

conf/zeppelin-env.cmd.template

@@ -34,6 +34,7 @@ REM set ZEPPELIN_NOTEBOOK_S3_USER              REM User in bucket where notebook
 REM set ZEPPELIN_IDENT_STRING   		REM A string representing this instance of zeppelin. $USER by default.
 REM set ZEPPELIN_NICENESS       		REM The scheduling priority for daemons. Defaults to 0.
 REM set ZEPPELIN_INTERPRETER_LOCALREPO         REM Local repository for interpreter's additional dependency loading
+REM set ZEPPELIN_INTERPRETER_DOWNLOAD_DIR          REM download interpreter for directory path
 
 
 REM Spark interpreter configuration

conf/zeppelin-env.sh.template

@@ -35,6 +35,7 @@
 # export ZEPPELIN_IDENT_STRING   		# A string representing this instance of zeppelin. $USER by default.
 # export ZEPPELIN_NICENESS       		# The scheduling priority for daemons. Defaults to 0.
 # export ZEPPELIN_INTERPRETER_LOCALREPO         # Local repository for interpreter's additional dependency loading
+# export ZEPPELIN_INTERPRETER_DOWNLOAD_DIR          # download interpreter for directory path
 
 
 #### Spark interpreter configuration ####

zeppelin-server/src/main/java/org/apache/zeppelin/rest/InterpreterRestApi.java

@@ -36,6 +36,7 @@
 import org.apache.zeppelin.dep.Repository;
 import org.apache.zeppelin.interpreter.*;
 import org.apache.zeppelin.interpreter.Interpreter.RegisteredInterpreter;
+import org.apache.zeppelin.rest.message.LoadDynamicInterpreterRequest;
 import org.apache.zeppelin.rest.message.NewInterpreterSettingRequest;
 import org.apache.zeppelin.rest.message.UpdateInterpreterSettingRequest;
 import org.apache.zeppelin.server.JsonResponse;
@@ -239,4 +240,46 @@ public Response removeRepository(@PathParam("repoId") String repoId) {
     }
     return new JsonResponse(Status.OK).build();
   }
+
+  /**
+   * load a downloaded interpreter via external repository
+   */
+  @POST
+  @Path("load/{interpreterGroupName}/{interpreterName}")
+  public Response loadDynamicInterpreter(
+      @PathParam("interpreterGroupName") String interpreterGroupName,
+      @PathParam("interpreterName") String interpreterName, String message) {
+    logger.info("dynamic load interpreter interpreterGroupName [{}] name [{}]",
+      interpreterGroupName, interpreterName);
+    try {
+      LoadDynamicInterpreterRequest request = gson.fromJson(
+        message, LoadDynamicInterpreterRequest.class
+      );
+
+      if (request.getClassName() == null
+          || request.getArtifact() == null
+          || request.getUrl() == null) {
+        throw new Exception("invalid request data");
+      }
+
+      boolean result = interpreterFactory.loadDynamicInterpreter(
+        interpreterGroupName,
+        interpreterName,
+        request.getArtifact(),
+        request.getClassName(),
+        request.getUrl(),
+        request.isSnapshot()
+      );
+
+      if (result == false) {
+        throw new Exception("can't not found artifact");
+      }
+
+    } catch (Exception e) {
+      logger.error("Exception in InterpreterRestApi while adding repository - load failed ", e);
+      return new JsonResponse(
+        Status.NOT_FOUND, e.getMessage(), ExceptionUtils.getStackTrace(e)).build();
+    }
+    return new JsonResponse(Status.OK).build();
+  }
 }

zeppelin-server/src/main/java/org/apache/zeppelin/rest/message/LoadDynamicInterpreterRequest.java

@@ -0,0 +1,56 @@
+/*
+ * 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.zeppelin.rest.message;
+
+import java.util.Map;
+
+/**
+ * LoadDynamicInterpreterRequest rest api request message
+ */
+
+public class LoadDynamicInterpreterRequest {
+  String artifact;
+  String className;
+  Map<String, Object> repository;
+
+  public LoadDynamicInterpreterRequest() {
+
+  }
+
+  public String getArtifact() { return artifact; }
+
+  public String getClassName() { return className; }
+
+  public Map<String, Object> getRepository() { return repository; }
+
+  public String getUrl() throws ClassCastException {
+    Object urlObj = repository.get("url");
+    if (urlObj == null) {
+      return null;
+    }
+    return (String) urlObj;
+  }
+
+  public Boolean isSnapshot() throws ClassCastException {
+    Object snapshotFlagObj = repository.get("snapshot");
+    if (snapshotFlagObj == null) {
+      return false;
+    }
+    return (Boolean) snapshotFlagObj;
+  }
+}

zeppelin-zengine/src/main/java/org/apache/zeppelin/conf/ZeppelinConfiguration.java

@@ -503,6 +503,7 @@ public static enum ConfVars {
     // Decide when new note is created, interpreter settings will be binded automatically or not.
     ZEPPELIN_NOTEBOOK_AUTO_INTERPRETER_BINDING("zeppelin.notebook.autoInterpreterBinding", true),
     ZEPPELIN_CONF_DIR("zeppelin.conf.dir", "conf"),
+    ZEPPELIN_INTERPRETER_DOWNLOAD_DIR("zeppelin.interpreter.download.dir", "interpreter"),
     ZEPPELIN_DEP_LOCALREPO("zeppelin.dep.localrepo", "local-repo"),
     // Allows a way to specify a ',' separated list of allowed origins for rest and websockets
     // i.e. http://localhost:8080

zeppelin-zengine/src/main/java/org/apache/zeppelin/interpreter/InterpreterFactory.java

@@ -60,7 +60,7 @@ public class InterpreterFactory implements InterpreterGroupFactory {
       .synchronizedMap(new HashMap<String, URLClassLoader>());
 
   private ZeppelinConfiguration conf;
-  String[] interpreterClassList;
+  List<String> interpreterClassList;
 
   private Map<String, InterpreterSetting> interpreterSettings =
       new HashMap<String, InterpreterSetting>();
@@ -99,7 +99,10 @@ public InterpreterFactory(ZeppelinConfiguration conf, InterpreterOption defaultO
     this.interpreterRepositories = depResolver.getRepos();
     this.remoteInterpreterProcessListener = remoteInterpreterProcessListener;
     String replsConf = conf.getString(ConfVars.ZEPPELIN_INTERPRETERS);
-    interpreterClassList = replsConf.split(",");
+    interpreterClassList = new ArrayList<String>();
+    for (String className : replsConf.split(",")) {
+      interpreterClassList.add(className);
+    }
 
     GsonBuilder builder = new GsonBuilder();
     builder.setPrettyPrinting();
@@ -110,6 +113,84 @@ public InterpreterFactory(ZeppelinConfiguration conf, InterpreterOption defaultO
     init();
   }
 
+  public boolean loadDynamicInterpreter(String intpGroupName, String intpName, String artifact,
+      String intpClassName) {
+    return loadDynamicInterpreter(intpGroupName, intpName, artifact, intpClassName, null, false);
+  }
+
+  public boolean loadDynamicInterpreter(String intpGroupName, String intpName, String artifact,
+      String intpClassName, String repositoryUrl, boolean isSnapShotRepo) {
+    String[] artifactItem = artifact.split(":");
+    String zepInterpreterRepoDir = conf.getString(ConfVars.ZEPPELIN_INTERPRETER_DOWNLOAD_DIR);
+    String zepInterpreterRepoFullPath = conf.getRelativeDir(
+      ConfVars.ZEPPELIN_INTERPRETER_DOWNLOAD_DIR
+    );
+    String interpreterDesPath = String.format("%s/%s/%s/", zepInterpreterRepoDir,
+      intpGroupName, intpName);
+    String interpreterLoadPath = String.format("%s/%s/%s", zepInterpreterRepoFullPath,
+      intpGroupName, intpName);
+
+    if (artifactItem.length <= 0) {
+      logger.error("Failed load dynamic interpreter - invalid artifact : {}", artifact);
+      return false;
+    }
+
+    try {
+      if (repositoryUrl != null) {
+        depResolver.addRepo("dyInterpreterRepo", repositoryUrl, isSnapShotRepo);
+      }
+      logger.info("interpreter path : {}", interpreterLoadPath);
+      depResolver.load(artifact, interpreterDesPath);
+      setDynamicInterpreter(intpClassName, interpreterLoadPath);
+    } catch (Exception e) {
+      logger.error("Failed load dynamic interpreter : ", e);
+      return false;
+    }
+    return true;
+  }
+
+  protected void setDynamicInterpreter(String interpreterClassName, String fileDirPath)
+      throws InterpreterException, IOException {
+    logger.info("load Dynamic Interpreter ClassName : {}", interpreterClassName);
+    logger.info("load Dynamic Interpreter FilePath : {}", interpreterClassName);
+
+    ClassLoader oldcl = Thread.currentThread().getContextClassLoader();
+    interpreterClassList.add(interpreterClassName);
+    // Load classes
+    File interpreterDir = new File(fileDirPath);
+
+    if (interpreterDir != null) {
+      URL[] urls = null;
+      try {
+        urls = recursiveBuildLibList(interpreterDir);
+      } catch (MalformedURLException e1) {
+        logger.error("Can't load jars ", e1);
+      }
+      URLClassLoader ccl = new URLClassLoader(urls, oldcl);
+
+      try {
+        Class.forName(interpreterClassName, true, ccl);
+        Set<String> keys = Interpreter.registeredInterpreters.keySet();
+        for (String intName : keys) {
+          if (interpreterClassName.equals(
+              Interpreter.registeredInterpreters.get(intName).getClassName())) {
+            Interpreter.registeredInterpreters.get(intName).setPath(fileDirPath);
+            logger.info("Interpreter {} found. class={}", intName, fileDirPath);
+            cleanCl.put(fileDirPath, ccl);
+          }
+        }
+      } catch (ClassNotFoundException e) {
+        logger.error("Load error : ", e);
+      }
+    }
+
+    for (String settingId : interpreterSettings.keySet()) {
+      InterpreterSetting setting = interpreterSettings.get(settingId);
+      logger.info("Interpreter setting group {} : id={}, name={}",
+        setting.getGroup(), settingId, setting.getName());
+    }
+  }
+
   private void init() throws InterpreterException, IOException, RepositoryException {
     ClassLoader oldcl = Thread.currentThread().getContextClassLoader();