package com.fokantech.conf;
import lombok.extern.slf4j.Slf4j;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;
import java.io.*;
import java.util.List;
@Slf4j
public class UpdateLabel {
public static void updateValue(String path) {
File[] files = new File(path).listFiles();
for (int i = 0; i < files.length; i++) {
SAXReader saxReader = new SAXReader();
Document document = null;
try {
document = saxReader.read(files[i]);
Element annotation = document.getRootElement();
List<Element> objects = annotation.elements("object");
for (Element object : objects) {
String typeName = object.element("name").getStringValue();
if (typeName.equals("hat")){
object.element("name").setText("safetyhat");
log.debug("进行修正");
}else {
log.debug("不用修正");
}
System.out.println("=============================");
}
OutputFormat format = OutputFormat.createPrettyPrint();
format.setEncoding("UTF-8");
XMLWriter writer = new XMLWriter(new FileOutputStream(files[i]), format);
writer.write(document);
writer.close();
} catch (DocumentException e) {
e.printStackTrace();
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
log.debug("当前文件已检阅");
}
}
public static void main(String[] args) throws Exception {
updateValue("");
}
}
===============================================================================