iextensionunit类_Java ICompilationUnit.reconcile方法代码示例

import org.eclipse.jdt.core.ICompilationUnit; //导入方法依赖的package包/类

/**

* @param file

* @param info

* @param monitor

* @throws MalformedTreeException

* @throws BadLocationException

* @throws CoreException

*/

@SuppressWarnings("deprecation")

public static void addGeneratedAnnotation(IFile file, IFile graphFile, IProgressMonitor monitor)

throws MalformedTreeException, BadLocationException, CoreException {

ICompilationUnit compilationUnit = JavaCore.createCompilationUnitFrom(file);

try {

String source = compilationUnit.getSource();

Document document = new Document(source);

compilationUnit.becomeWorkingCopy(new SubProgressMonitor(monitor, 1));

ASTParser parser = ASTParser.newParser(AST.JLS8);

parser.setSource(compilationUnit);

parser.setResolveBindings(true);

CompilationUnit astRoot = (CompilationUnit) parser.createAST(new SubProgressMonitor(monitor, 1));

astRoot.recordModifications();

final ImportRewrite importRewrite = ImportRewrite.create(astRoot, true);

importRewrite.addImport("javax.annotation.Generated");

astRoot.accept(new ASTVisitor() {

@SuppressWarnings("unchecked")

@Override

public boolean visit(TypeDeclaration node) {

ASTNode copiedNode = null;

// Add Generated annotation

ClassExtension ce;

try {

ce = new ClassExtension(false, false, false, false, false, false, "", "", null, false, false,

"", "", "", graphFile);

NormalAnnotation annotation = ce.getGeneratedClassAnnotation();

if (annotation != null) {

copiedNode = ASTNode.copySubtree(node.getAST(), annotation);

node.modifiers().add(0, copiedNode);

}

} catch (JavaModelException e) {

ResourceManager.logException(e);

}

return super.visit(node);

}

});

TextEdit rewrite = astRoot.rewrite(document, compilationUnit.getJavaProject().getOptions(true));

rewrite.apply(document);

TextEdit rewriteImports = importRewrite.rewriteImports(new SubProgressMonitor(monitor, 1));

rewriteImports.apply(document);

String newSource = document.get();

compilationUnit.getBuffer().setContents(newSource);

compilationUnit.reconcile(ICompilationUnit.NO_AST, false, null, new SubProgressMonitor(monitor, 1));

compilationUnit.commitWorkingCopy(false, new SubProgressMonitor(monitor, 1));

} finally {

compilationUnit.discardWorkingCopy();

monitor.done();

}

// WorkbenchFacade.JDTManager.reorganizeImport(compilationUnit);

}