开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 网格计算 > 查看源码
NewMapReduceProjectWizard.java
资源名称：hadoop-0.20.0.tar.gz　[点击查看]
上传用户：quxuerui
上传日期：2018-01-08
资源大小：41811k
文件大小：14k
源码类别：

网格计算

开发平台：

Java

NewMapReduceProjectWizard.java：源码内容
  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  19. package org.apache.hadoop.eclipse;
  21. import java.io.File;
  22. import java.io.FilenameFilter;
  23. import java.lang.reflect.InvocationTargetException;
  24. import java.util.logging.Level;
  25. import java.util.logging.Logger;
  27. import org.apache.hadoop.eclipse.preferences.MapReducePreferencePage;
  28. import org.apache.hadoop.eclipse.preferences.PreferenceConstants;
  29. import org.eclipse.core.resources.IProject;
  30. import org.eclipse.core.resources.IProjectDescription;
  31. import org.eclipse.core.resources.ResourcesPlugin;
  32. import org.eclipse.core.runtime.CoreException;
  33. import org.eclipse.core.runtime.IConfigurationElement;
  34. import org.eclipse.core.runtime.IExecutableExtension;
  35. import org.eclipse.core.runtime.IProgressMonitor;
  36. import org.eclipse.core.runtime.NullProgressMonitor;
  37. import org.eclipse.core.runtime.Path;
  38. import org.eclipse.core.runtime.QualifiedName;
  39. import org.eclipse.core.runtime.SubProgressMonitor;
  40. import org.eclipse.jdt.ui.wizards.NewJavaProjectWizardPage;
  41. import org.eclipse.jface.dialogs.IDialogConstants;
  42. import org.eclipse.jface.operation.IRunnableWithProgress;
  43. import org.eclipse.jface.preference.PreferenceDialog;
  44. import org.eclipse.jface.preference.PreferenceManager;
  45. import org.eclipse.jface.preference.PreferenceNode;
  46. import org.eclipse.jface.viewers.IStructuredSelection;
  47. import org.eclipse.jface.wizard.IWizardPage;
  48. import org.eclipse.jface.wizard.Wizard;
  49. import org.eclipse.swt.SWT;
  50. import org.eclipse.swt.events.SelectionEvent;
  51. import org.eclipse.swt.events.SelectionListener;
  52. import org.eclipse.swt.layout.GridData;
  53. import org.eclipse.swt.layout.GridLayout;
  54. import org.eclipse.swt.widgets.Button;
  55. import org.eclipse.swt.widgets.Composite;
  56. import org.eclipse.swt.widgets.DirectoryDialog;
  57. import org.eclipse.swt.widgets.Group;
  58. import org.eclipse.swt.widgets.Link;
  59. import org.eclipse.swt.widgets.Text;
  60. import org.eclipse.ui.IWorkbench;
  61. import org.eclipse.ui.IWorkbenchWizard;
  62. import org.eclipse.ui.PlatformUI;
  63. import org.eclipse.ui.dialogs.WizardNewProjectCreationPage;
  64. import org.eclipse.ui.wizards.newresource.BasicNewProjectResourceWizard;
  66. /**
  67.  * Wizard for creating a new MapReduce Project
  68.  * 
  69.  */
  71. public class NewMapReduceProjectWizard extends Wizard implements
  72.     IWorkbenchWizard, IExecutableExtension {
  73.   static Logger log =
  74.       Logger.getLogger(NewMapReduceProjectWizard.class.getName());
  76.   private HadoopFirstPage firstPage;
  78.   private NewJavaProjectWizardPage javaPage;
  80.   public NewDriverWizardPage newDriverPage;
  82.   private IConfigurationElement config;
  84.   public NewMapReduceProjectWizard() {
  85.     setWindowTitle("New MapReduce Project Wizard");
  86.   }
  88.   public void init(IWorkbench workbench, IStructuredSelection selection) {
  90.   }
  92.   @Override
  93.   public boolean canFinish() {
  94.     return firstPage.isPageComplete() && javaPage.isPageComplete()
  95.     // && ((!firstPage.generateDriver.getSelection())
  96.     // || newDriverPage.isPageComplete()
  97.     ;
  98.   }
  100.   @Override
  101.   public IWizardPage getNextPage(IWizardPage page) {
  102.     // if (page == firstPage
  103.     // && firstPage.generateDriver.getSelection()
  104.     // )
  105.     // {
  106.     // return newDriverPage; // if "generate mapper" checked, second page is
  107.     // new driver page
  108.     // }
  109.     // else
  110.     // {
  111.     IWizardPage answer = super.getNextPage(page);
  112.     if (answer == newDriverPage) {
  113.       return null; // dont flip to new driver page unless "generate
  114.       // driver" is checked
  115.     } else if (answer == javaPage) {
  116.       return answer;
  117.     } else {
  118.       return answer;
  119.     }
  120.     // }
  121.   }
  123.   @Override
  124.   public IWizardPage getPreviousPage(IWizardPage page) {
  125.     if (page == newDriverPage) {
  126.       return firstPage; // newDriverPage, if it appears, is the second
  127.       // page
  128.     } else {
  129.       return super.getPreviousPage(page);
  130.     }
  131.   }
  133.   static class HadoopFirstPage extends WizardNewProjectCreationPage
  134.       implements SelectionListener {
  135.     public HadoopFirstPage() {
  136.       super("New Hadoop Project");
  137.       setImageDescriptor(ImageLibrary.get("wizard.mapreduce.project.new"));
  138.     }
  140.     private Link openPreferences;
  142.     private Button workspaceHadoop;
  144.     private Button projectHadoop;
  146.     private Text location;
  148.     private Button browse;
  150.     private String path;
  152.     public String currentPath;
  154.     // private Button generateDriver;
  156.     @Override
  157.     public void createControl(Composite parent) {
  158.       super.createControl(parent);
  160.       setTitle("MapReduce Project");
  161.       setDescription("Create a MapReduce project.");
  163.       Group group = new Group((Composite) getControl(), SWT.NONE);
  164.       group.setLayoutData(new GridData(GridData.FILL_HORIZONTAL));
  165.       group.setText("Hadoop MapReduce Library Installation Path");
  166.       GridLayout layout = new GridLayout(3, true);
  167.       layout.marginLeft =
  168.           convertHorizontalDLUsToPixels(IDialogConstants.HORIZONTAL_MARGIN);
  169.       layout.marginRight =
  170.           convertHorizontalDLUsToPixels(IDialogConstants.HORIZONTAL_MARGIN);
  171.       layout.marginTop =
  172.           convertHorizontalDLUsToPixels(IDialogConstants.VERTICAL_MARGIN);
  173.       layout.marginBottom =
  174.           convertHorizontalDLUsToPixels(IDialogConstants.VERTICAL_MARGIN);
  175.       group.setLayout(layout);
  177.       workspaceHadoop = new Button(group, SWT.RADIO);
  178.       GridData d =
  179.           new GridData(GridData.BEGINNING, GridData.BEGINNING, false, false);
  180.       d.horizontalSpan = 2;
  181.       workspaceHadoop.setLayoutData(d);
  182.       // workspaceHadoop.setText("Use default workbench Hadoop library
  183.       // location");
  184.       workspaceHadoop.setSelection(true);
  186.       updateHadoopDirLabelFromPreferences();
  188.       openPreferences = new Link(group, SWT.NONE);
  189.       openPreferences
  190.           .setText("<a>Configure Hadoop install directory...</a>");
  191.       openPreferences.setLayoutData(new GridData(GridData.END,
  192.           GridData.CENTER, false, false));
  193.       openPreferences.addSelectionListener(this);
  195.       projectHadoop = new Button(group, SWT.RADIO);
  196.       projectHadoop.setLayoutData(new GridData(GridData.BEGINNING,
  197.           GridData.CENTER, false, false));
  198.       projectHadoop.setText("Specify Hadoop library location");
  200.       location = new Text(group, SWT.SINGLE | SWT.BORDER);
  201.       location.setText("");
  202.       d = new GridData(GridData.END, GridData.CENTER, true, false);
  203.       d.horizontalSpan = 1;
  204.       d.widthHint = 250;
  205.       d.grabExcessHorizontalSpace = true;
  206.       location.setLayoutData(d);
  207.       location.setEnabled(false);
  209.       browse = new Button(group, SWT.NONE);
  210.       browse.setText("Browse...");
  211.       browse.setLayoutData(new GridData(GridData.BEGINNING, GridData.CENTER,
  212.           false, false));
  213.       browse.setEnabled(false);
  214.       browse.addSelectionListener(this);
  216.       projectHadoop.addSelectionListener(this);
  217.       workspaceHadoop.addSelectionListener(this);
  219.       // generateDriver = new Button((Composite) getControl(), SWT.CHECK);
  220.       // generateDriver.setText("Generate a MapReduce driver");
  221.       // generateDriver.addListener(SWT.Selection, new Listener()
  222.       // {
  223.       // public void handleEvent(Event event) {
  224.       // getContainer().updateButtons(); }
  225.       // });
  226.     }
  228.     @Override
  229.     public boolean isPageComplete() {
  230.       boolean validHadoop = validateHadoopLocation();
  232.       if (!validHadoop && isCurrentPage()) {
  233.         setErrorMessage("Invalid Hadoop Runtime specified; please click 'Configure Hadoop install directory' or fill in library location input field");
  234.       } else {
  235.         setErrorMessage(null);
  236.       }
  238.       return super.isPageComplete() && validHadoop;
  239.     }
  241.     private boolean validateHadoopLocation() {
  242.       FilenameFilter gotHadoopJar = new FilenameFilter() {
  243.         public boolean accept(File dir, String name) {
  244.           return (name.startsWith("hadoop") && name.endsWith(".jar")
  245.               && (name.indexOf("test") == -1) && (name.indexOf("examples") == -1));
  246.         }
  247.       };
  249.       if (workspaceHadoop.getSelection()) {
  250.         this.currentPath = path;
  251.         return new Path(path).toFile().exists()
  252.             && (new Path(path).toFile().list(gotHadoopJar).length > 0);
  253.       } else {
  254.         this.currentPath = location.getText();
  255.         File file = new Path(location.getText()).toFile();
  256.         return file.exists()
  257.             && (new Path(location.getText()).toFile().list(gotHadoopJar).length > 0);
  258.       }
  259.     }
  261.     private void updateHadoopDirLabelFromPreferences() {
  262.       path =
  263.           Activator.getDefault().getPreferenceStore().getString(
  264.               PreferenceConstants.P_PATH);
  266.       if ((path != null) && (path.length() > 0)) {
  267.         workspaceHadoop.setText("Use default Hadoop");
  268.       } else {
  269.         workspaceHadoop.setText("Use default Hadoop (currently not set)");
  270.       }
  271.     }
  273.     public void widgetDefaultSelected(SelectionEvent e) {
  274.     }
  276.     public void widgetSelected(SelectionEvent e) {
  277.       if (e.getSource() == openPreferences) {
  278.         PreferenceManager manager = new PreferenceManager();
  279.         manager.addToRoot(new PreferenceNode(
  280.             "Hadoop Installation Directory", new MapReducePreferencePage()));
  281.         PreferenceDialog dialog =
  282.             new PreferenceDialog(this.getShell(), manager);
  283.         dialog.create();
  284.         dialog.setMessage("Select Hadoop Installation Directory");
  285.         dialog.setBlockOnOpen(true);
  286.         dialog.open();
  288.         updateHadoopDirLabelFromPreferences();
  289.       } else if (e.getSource() == browse) {
  290.         DirectoryDialog dialog = new DirectoryDialog(this.getShell());
  291.         dialog
  292.             .setMessage("Select a hadoop installation, containing hadoop-X-core.jar");
  293.         dialog.setText("Select Hadoop Installation Directory");
  294.         String directory = dialog.open();
  296.         if (directory != null) {
  297.           location.setText(directory);
  299.           if (!validateHadoopLocation()) {
  300.             setErrorMessage("No Hadoop jar found in specified directory");
  301.           } else {
  302.             setErrorMessage(null);
  303.           }
  304.         }
  305.       } else if (projectHadoop.getSelection()) {
  306.         location.setEnabled(true);
  307.         browse.setEnabled(true);
  308.       } else {
  309.         location.setEnabled(false);
  310.         browse.setEnabled(false);
  311.       }
  313.       getContainer().updateButtons();
  314.     }
  315.   }
  317.   @Override
  318.   public void addPages() {
  319.     /*
  320.      * firstPage = new HadoopFirstPage(); addPage(firstPage ); addPage( new
  321.      * JavaProjectWizardSecondPage(firstPage) );
  322.      */
  324.     firstPage = new HadoopFirstPage();
  325.     javaPage =
  326.         new NewJavaProjectWizardPage(ResourcesPlugin.getWorkspace()
  327.             .getRoot(), firstPage);
  328.     // newDriverPage = new NewDriverWizardPage(false);
  329.     // newDriverPage.setPageComplete(false); // ensure finish button
  330.     // initially disabled
  331.     addPage(firstPage);
  332.     addPage(javaPage);
  334.     // addPage(newDriverPage);
  335.   }
  337.   @Override
  338.   public boolean performFinish() {
  339.     try {
  340.       PlatformUI.getWorkbench().getProgressService().runInUI(
  341.           this.getContainer(), new IRunnableWithProgress() {
  342.             public void run(IProgressMonitor monitor) {
  343.               try {
  344.                 monitor.beginTask("Create Hadoop Project", 300);
  346.                 javaPage.getRunnable().run(
  347.                     new SubProgressMonitor(monitor, 100));
  349.                 // if( firstPage.generateDriver.getSelection())
  350.                 // {
  351.                 // newDriverPage.setPackageFragmentRoot(javaPage.getNewJavaProject().getAllPackageFragmentRoots()[0],
  352.                 // false);
  353.                 // newDriverPage.getRunnable().run(new
  354.                 // SubProgressMonitor(monitor,100));
  355.                 // }
  357.                 IProject project =
  358.                     javaPage.getNewJavaProject().getResource().getProject();
  359.                 IProjectDescription description = project.getDescription();
  360.                 String[] existingNatures = description.getNatureIds();
  361.                 String[] natures = new String[existingNatures.length + 1];
  362.                 for (int i = 0; i < existingNatures.length; i++) {
  363.                   natures[i + 1] = existingNatures[i];
  364.                 }
  366.                 natures[0] = MapReduceNature.ID;
  367.                 description.setNatureIds(natures);
  369.                 project.setPersistentProperty(new QualifiedName(
  370.                     Activator.PLUGIN_ID, "hadoop.runtime.path"),
  371.                     firstPage.currentPath);
  372.                 project.setDescription(description,
  373.                     new NullProgressMonitor());
  375.                 String[] natureIds = project.getDescription().getNatureIds();
  376.                 for (int i = 0; i < natureIds.length; i++) {
  377.                   log.fine("Nature id # " + i + " > " + natureIds[i]);
  378.                 }
  380.                 monitor.worked(100);
  381.                 monitor.done();
  383.                 BasicNewProjectResourceWizard.updatePerspective(config);
  384.               } catch (CoreException e) {
  385.                 // TODO Auto-generated catch block
  386.                 log.log(Level.SEVERE, "CoreException thrown.", e);
  387.               } catch (InvocationTargetException e) {
  388.                 // TODO Auto-generated catch block
  389.                 e.printStackTrace();
  390.               } catch (InterruptedException e) {
  391.                 // TODO Auto-generated catch block
  392.                 e.printStackTrace();
  393.               }
  394.             }
  395.           }, null);
  396.     } catch (InvocationTargetException e) {
  397.       // TODO Auto-generated catch block
  398.       e.printStackTrace();
  399.     } catch (InterruptedException e) {
  400.       // TODO Auto-generated catch block
  401.       e.printStackTrace();
  402.     }
  404.     return true;
  405.   }
  407.   public void setInitializationData(IConfigurationElement config,
  408.       String propertyName, Object data) throws CoreException {
  409.     this.config = config;
  410.   }
  411. }