forked from elastic/elasticsearch
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Port krb5kdc to test container and rework hdfs handling (elastic#106228)
This ports our krb5kdc test fixture to test container and reworks hdfs handling to also be based on test containers. The yaml rest tests that are using hdfs required introducing variable substitution in yamlresttestparser handling.
- Loading branch information
Showing
42 changed files
with
1,912 additions
and
1,479 deletions.
There are no files selected for viewing
142 changes: 142 additions & 0 deletions
142
...src/main/java/org/elasticsearch/gradle/internal/shadow/XmlClassRelocationTransformer.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,142 @@ | ||
| /* | ||
| * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one | ||
| * or more contributor license agreements. Licensed under the Elastic License | ||
| * 2.0 and the Server Side Public License, v 1; you may not use this file except | ||
| * in compliance with, at your election, the Elastic License 2.0 or the Server | ||
| * Side Public License, v 1. | ||
| */ | ||
|
|
||
| package org.elasticsearch.gradle.internal.shadow; | ||
|
|
||
| import com.github.jengelman.gradle.plugins.shadow.ShadowStats; | ||
| import com.github.jengelman.gradle.plugins.shadow.relocation.RelocateClassContext; | ||
| import com.github.jengelman.gradle.plugins.shadow.relocation.Relocator; | ||
| import com.github.jengelman.gradle.plugins.shadow.transformers.Transformer; | ||
| import com.github.jengelman.gradle.plugins.shadow.transformers.TransformerContext; | ||
|
|
||
| import org.apache.commons.io.IOUtils; | ||
| import org.apache.tools.zip.ZipEntry; | ||
| import org.apache.tools.zip.ZipOutputStream; | ||
| import org.gradle.api.file.FileTreeElement; | ||
| import org.w3c.dom.Document; | ||
| import org.w3c.dom.Node; | ||
| import org.w3c.dom.NodeList; | ||
|
|
||
| import java.io.BufferedInputStream; | ||
| import java.io.ByteArrayOutputStream; | ||
| import java.io.IOException; | ||
| import java.util.List; | ||
|
|
||
| import javax.xml.parsers.DocumentBuilder; | ||
| import javax.xml.parsers.DocumentBuilderFactory; | ||
| import javax.xml.transform.TransformerException; | ||
| import javax.xml.transform.TransformerFactory; | ||
| import javax.xml.transform.dom.DOMSource; | ||
| import javax.xml.transform.stream.StreamResult; | ||
|
|
||
| public class XmlClassRelocationTransformer implements Transformer { | ||
|
|
||
| boolean hasTransformedResource = false; | ||
|
|
||
| private Document doc; | ||
|
|
||
| private String resource; | ||
|
|
||
| @Override | ||
| public boolean canTransformResource(FileTreeElement element) { | ||
| String path = element.getRelativePath().getPathString(); | ||
| if (resource != null && resource.equals(path)) { | ||
| return true; | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| @Override | ||
| public void transform(TransformerContext context) { | ||
| try { | ||
| BufferedInputStream bis = new BufferedInputStream(context.getIs()); | ||
| DocumentBuilderFactory dbFactory = DocumentBuilderFactory.newInstance(); | ||
| DocumentBuilder dBuilder = dbFactory.newDocumentBuilder(); | ||
| doc = dBuilder.parse(bis); | ||
| doc.getDocumentElement().normalize(); | ||
| Node root = doc.getDocumentElement(); | ||
| walkThroughNodes(root, context); | ||
| if (hasTransformedResource == false) { | ||
| this.doc = null; | ||
| } | ||
| } catch (Exception e) { | ||
| throw new RuntimeException("Error parsing xml file in " + context.getIs(), e); | ||
| } | ||
| } | ||
|
|
||
| private static String getRelocatedClass(String className, TransformerContext context) { | ||
| List<Relocator> relocators = context.getRelocators(); | ||
| ShadowStats stats = context.getStats(); | ||
| if (className != null && className.length() > 0 && relocators != null) { | ||
| for (Relocator relocator : relocators) { | ||
| if (relocator.canRelocateClass(className)) { | ||
| RelocateClassContext relocateClassContext = new RelocateClassContext(className, stats); | ||
| return relocator.relocateClass(relocateClassContext); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| return className; | ||
| } | ||
|
|
||
| private void walkThroughNodes(Node node, TransformerContext context) { | ||
| if (node.getNodeType() == Node.TEXT_NODE) { | ||
| String nodeValue = node.getNodeValue(); | ||
| if (nodeValue.isBlank() == false) { | ||
| String relocatedClass = getRelocatedClass(nodeValue, context); | ||
| if (relocatedClass.equals(nodeValue) == false) { | ||
| node.setNodeValue(relocatedClass); | ||
| hasTransformedResource = true; | ||
| } | ||
| } | ||
| } | ||
| NodeList nodeList = node.getChildNodes(); | ||
| for (int i = 0; i < nodeList.getLength(); i++) { | ||
| Node currentNode = nodeList.item(i); | ||
| walkThroughNodes(currentNode, context); | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public boolean hasTransformedResource() { | ||
| return hasTransformedResource; | ||
| } | ||
|
|
||
| @Override | ||
| public void modifyOutputStream(ZipOutputStream os, boolean preserveFileTimestamps) { | ||
| ZipEntry entry = new ZipEntry(resource); | ||
| entry.setTime(TransformerContext.getEntryTimestamp(preserveFileTimestamps, entry.getTime())); | ||
|
|
||
| try { | ||
| // Write the content back to the XML file | ||
| TransformerFactory transformerFactory = TransformerFactory.newInstance(); | ||
| DOMSource source = new DOMSource(doc); | ||
|
|
||
| // Result stream will be a ByteArrayOutputStream | ||
| ByteArrayOutputStream baos = new ByteArrayOutputStream(); | ||
| StreamResult result = new StreamResult(baos); | ||
| // Do the transformation and serialization | ||
| transformerFactory.newTransformer().transform(source, result); | ||
| os.putNextEntry(entry); | ||
| IOUtils.write(baos.toByteArray(), os); | ||
| os.closeEntry(); | ||
| } catch (IOException e) { | ||
| throw new RuntimeException(e); | ||
| } catch (TransformerException e) { | ||
| throw new RuntimeException(e); | ||
| } finally { | ||
| hasTransformedResource = false; | ||
| doc = null; | ||
| } | ||
| } | ||
|
|
||
| @Override | ||
| public String getName() { | ||
| return getClass().getSimpleName(); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.