From ab66003c2820b338a7315dcd45653d8f7dfe1f4d Mon Sep 17 00:00:00 2001
From: Prasad Wagle <pwagle@twitter.com>
Date: Mon, 25 Apr 2016 16:50:14 -0700
Subject: [PATCH] Run scheduled notes with a user

---
 zeppelin-web/src/app/notebook/notebook-actionBar.html      | 7 +++++++
 zeppelin-web/src/app/notebook/notebook.controller.js       | 6 ++++++
 .../src/main/java/org/apache/zeppelin/notebook/Note.java   | 5 +++++
 3 files changed, 18 insertions(+)

diff --git a/zeppelin-web/src/app/notebook/notebook-actionBar.html b/zeppelin-web/src/app/notebook/notebook-actionBar.html
index f371bd318a8..21f7e9754d8 100644
--- a/zeppelin-web/src/app/notebook/notebook-actionBar.html
+++ b/zeppelin-web/src/app/notebook/notebook-actionBar.html
@@ -131,6 +131,13 @@ <h3>
                   {{note.info.cron}}
                 </p>
               </div>
+              <div>
+                <span>- Cron executing user (click enter in field to submit)</span>
+                <input type="text"
+                       ng-model="note.config.cronExecutingUser"
+                       ng-enter="setCronExecutingUser(note.config.cronExecutingUser)"
+                       dropdown-input />
+              </div>
               <div>
                 <span>- auto-restart interpreter on cron execution </span>
                 <input type="checkbox"
diff --git a/zeppelin-web/src/app/notebook/notebook.controller.js b/zeppelin-web/src/app/notebook/notebook.controller.js
index 3656394d787..1783d059ea2 100644
--- a/zeppelin-web/src/app/notebook/notebook.controller.js
+++ b/zeppelin-web/src/app/notebook/notebook.controller.js
@@ -295,6 +295,12 @@ angular.module('zeppelinWebApp').controller('NotebookCtrl',
     $scope.setConfig();
   };
 
+  /** Set the username of the user to be used to execute all notes in notebook **/
+  $scope.setCronExecutingUser = function(cronExecutingUser) {
+    $scope.note.config.cronExecutingUser = cronExecutingUser;
+    $scope.setConfig();
+  };
+
   /** Set release resource for this note **/
   $scope.setReleaseResource = function(value) {
     $scope.note.config.releaseresource = value;
diff --git a/zeppelin-zengine/src/main/java/org/apache/zeppelin/notebook/Note.java b/zeppelin-zengine/src/main/java/org/apache/zeppelin/notebook/Note.java
index 6a097355386..903c51c8f34 100644
--- a/zeppelin-zengine/src/main/java/org/apache/zeppelin/notebook/Note.java
+++ b/zeppelin-zengine/src/main/java/org/apache/zeppelin/notebook/Note.java
@@ -39,6 +39,7 @@
 import org.apache.zeppelin.search.SearchService;
 
 import com.google.gson.Gson;
+import org.apache.zeppelin.user.AuthenticationInfo;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -350,11 +351,15 @@ public List<Map<String, String>> generateParagraphsInfo (){
    * Run all paragraphs sequentially.
    */
   public void runAll() {
+    String cronExecutingUser = (String) getConfig().get("cronExecutingUser");
     synchronized (paragraphs) {
       for (Paragraph p : paragraphs) {
         if (!p.isEnabled()) {
           continue;
         }
+        AuthenticationInfo authenticationInfo = new AuthenticationInfo();
+        authenticationInfo.setUser(cronExecutingUser);
+        p.setAuthenticationInfo(authenticationInfo);
         p.setNoteReplLoader(replLoader);
         p.setListener(jobListenerFactory.getParagraphJobListener(this));
         Interpreter intp = replLoader.get(p.getRequiredReplName());