فهرست منبع

HUE-6841 [oozie] Copy examples to HDFS as the hue user

Romain Rigaux 8 سال پیش
والد
کامیت
410e714c50
2فایلهای تغییر یافته به همراه5 افزوده شده و 7 حذف شده
  1. 2 2
      apps/oozie/src/oozie/management/commands/oozie_setup.py
  2. 3 5
      apps/pig/src/pig/management/commands/pig_setup.py

+ 2 - 2
apps/oozie/src/oozie/management/commands/oozie_setup.py

@@ -342,14 +342,14 @@ class Command(NoArgsCommand):
       remote_data_dir = self.fs.join(remote_dir, name)
       LOG.info(_('Copying examples %(local_dir)s to %(remote_data_dir)s\n') % {
                   'local_dir': local_dir, 'remote_data_dir': remote_data_dir})
-      self.fs.do_as_user(self.fs.DEFAULT_USER, self.fs.copyFromLocal, local_dir, remote_data_dir)
+      self.fs.do_as_user(self.user.username, self.fs.copyFromLocal, local_dir, remote_data_dir)
 
     # Copy sample data
     local_dir = LOCAL_SAMPLE_DATA_DIR.get()
     remote_data_dir = self.fs.join(remote_dir, 'data')
     LOG.info(_('Copying data %(local_dir)s to %(remote_data_dir)s\n') % {
                 'local_dir': local_dir, 'remote_data_dir': remote_data_dir})
-    self.fs.do_as_user(self.fs.DEFAULT_USER, self.fs.copyFromLocal, local_dir, remote_data_dir)
+    self.fs.do_as_user(self.user.username, self.fs.copyFromLocal, local_dir, remote_data_dir)
 
     # Load jobs
     LOG.info(_("Installing examples..."))

+ 3 - 5
apps/pig/src/pig/management/commands/pig_setup.py

@@ -100,6 +100,7 @@ STORE upper_case INTO '$output';
     fs = cluster.get_hdfs()
     create_directories(fs, [REMOTE_SAMPLE_DIR.get()])
     remote_dir = REMOTE_SAMPLE_DIR.get()
+    sample_user = install_sample_user()
 
     # Copy examples binaries
     for name in os.listdir(LOCAL_SAMPLE_DIR.get()):
@@ -107,17 +108,14 @@ STORE upper_case INTO '$output';
       remote_data_dir = fs.join(remote_dir, name)
       LOG.info(_('Copying examples %(local_dir)s to %(remote_data_dir)s\n') % {
                   'local_dir': local_dir, 'remote_data_dir': remote_data_dir})
-      fs.do_as_user(fs.DEFAULT_USER, fs.copyFromLocal, local_dir, remote_data_dir)
+      fs.do_as_user(sample_user.username, fs.copyFromLocal, local_dir, remote_data_dir)
 
     # Copy sample data
     local_dir = paths.get_thirdparty_root("sample_data")
     remote_data_dir = fs.join(remote_dir, 'data')
     LOG.info(_('Copying data %(local_dir)s to %(remote_data_dir)s\n') % {
                 'local_dir': local_dir, 'remote_data_dir': remote_data_dir})
-    fs.do_as_user(fs.DEFAULT_USER, fs.copyFromLocal, local_dir, remote_data_dir)
-
-    # Load jobs
-    sample_user = install_sample_user()
+    fs.do_as_user(sample_user.username, fs.copyFromLocal, local_dir, remote_data_dir)
 
     # Initialize doc2, whether editor script or link
     doc2 = None