import org.dmg.pmml.Extension; //导入依赖的package包/类
@Test
public void unmarshal() throws Exception {
	PMML pmml;

	System.setProperty("javax.xml.accessExternalDTD", "file");

	try(InputStream is = ResourceUtil.getStream(XXEAttackTest.class);){
		pmml = JAXBUtil.unmarshalPMML(new StreamSource(is));
	} finally {
		System.clearProperty("javax.xml.accessExternalDTD");
	}

	List<Extension> extensions = pmml.getExtensions();
	assertEquals(1, extensions.size());

	Extension extension = extensions.get(0);
	assertEquals(Arrays.asList("lol"), extension.getContent());
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
/**
 * Quite manually parse our fake model representation in PMML.
 */
private static ALSModelDescription read(InputStream in) throws JAXBException, SAXException {

  PMML pmml = IOUtil.unmarshal(in);
  List<Extension> extensions = pmml.getExtensions();
  Preconditions.checkNotNull(extensions);
  Preconditions.checkArgument(!extensions.isEmpty());

  ALSModelDescription model = new ALSModelDescription();

  for (Extension extension : extensions) {
    String name  = extension.getName();
    String value = extension.getValue();
    Preconditions.checkNotNull(name);
    Preconditions.checkNotNull(value);
    model.getPathByKey().put(name, value);
  }

  return model;
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
/**
 * @param pmml PMML model to add extension to, with a single {@code String} content and no value.
 *  The content is encoded as if they were being added to a PMML {@link Array} and are
 *  space-separated with PMML quoting rules
 * @param key extension key
 * @param content list of values to add as a {@code String}
 */
public static void addExtensionContent(PMML pmml, String key, Collection<?> content) {
  if (content.isEmpty()) {
    return;
  }
  String joined = TextUtils.joinPMMLDelimited(content);
  pmml.addExtensions(new Extension().setName(key).addContent(joined));
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
static
private Extension createExtension(String content){
	Extension extension = new Extension()
		.addContent(content);

	return extension;
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
/**
 * Retrieves the {@link com.feedzai.fos.impl.weka.utils.pmml.PMMLConversionCommons.Algorithm} represented
 * as a PMML extension named {@link com.feedzai.fos.impl.weka.utils.pmml.PMMLConversionCommons#ALGORITHM_EXTENSION_ELEMENT}.
 *
 * @param pmml The {@link PMML}.
 * @return The {@link com.feedzai.fos.impl.weka.utils.pmml.PMMLConversionCommons.Algorithm} in the PMML.
 * @throws PMMLConversionException If it fails to retrieve the algorithm from the given PPML.
 */
private static Algorithm getAlgorithm(PMML pmml) throws PMMLConversionException {
    for (Extension extension : pmml.getExtensions()) {
        if (ALGORITHM_EXTENSION_ELEMENT.equals(extension.getName())) {
            return Algorithm.valueOf(extension.getValue());
        }
    }

    throw new PMMLConversionException("Couldn't find '" + ALGORITHM_EXTENSION_ELEMENT + "' extension element in PMML.");
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
static
private List<BayesInput> parseBayesInputs(NaiveBayesModel naiveBayesModel){
	BayesInputs bayesInputs = naiveBayesModel.getBayesInputs();

	if(!bayesInputs.hasExtensions()){
		return bayesInputs.getBayesInputs();
	}

	List<BayesInput> result = new ArrayList<>(bayesInputs.getBayesInputs());

	// The support for continuous fields using the TargetValueStats element was officially introduced in PMML schema version 4.2.
	// However, it is possible to encounter this feature in older PMML schema version documents (most notably, produced by R's "pmml" package),
	// where the offending BayesInput element is surrounded by an Extension element:
	// <BayesInputs>
	//   <BayesInput>
	//     <PairCounts/>
	//   </BayesInput>
	//   <Extension>
	//     <BayesInput>
	//       <TargetValueStats/>
	//     </BayesInput>
	//   </Extension>
	// </BayesInputs>
	List<Extension> extensions = bayesInputs.getExtensions();
	for(Extension extension : extensions){
		List<?> objects = extension.getContent();

		for(Object object : objects){

			if(object instanceof BayesInput){
				BayesInput bayesInput = (BayesInput)object;

				result.add(bayesInput);
			}
		}
	}

	return result;
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
/**
 * Quite manually write our fake model representation in PMML.
 */
private static void write(OutputStream out, ALSModelDescription model) throws JAXBException {
  PMML pmml = new PMML(null, null, "4.1");
  for (Map.Entry<String,String> entry : model.getPathByKey().entrySet()) {
    Extension extension = new Extension();
    extension.setName(entry.getKey());
    extension.setValue(entry.getValue());
    pmml.getExtensions().add(extension);
  }
  IOUtil.marshal(pmml, out);
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
public static String getExtensionValue(PMML pmml, String name) {
  return pmml.getExtensions().stream().filter(extension -> name.equals(extension.getName())).findFirst().
      map(Extension::getValue).orElse(null);
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
static
private List<?> getDeepContent(Object object){
	Extension extension = (Extension)object;

	return extension.getContent();
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
static
private void checkExtension(String name, String value, Extension extension){
	assertEquals(name, extension.getName());
	assertEquals(value, extension.getValue());
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
/**
 * Adds a {@link org.dmg.pmml.Extension} element to the {@link org.dmg.pmml.PMML} with the
 * algorithm used to build the classifier.
 * <p/>
 * The name of the extension is {@link com.feedzai.fos.impl.weka.utils.pmml.PMMLConversionCommons#ALGORITHM_EXTENSION_ELEMENT}.
 *
 * @param pmml      The {@link PMML} to which add the algorithm.
 * @param algorithm The algorithm used to build the classifier.
 */
private static void addAlgorithm(PMML pmml, Algorithm algorithm) {
    Extension extension = new Extension();
    extension.withName(ALGORITHM_EXTENSION_ELEMENT);
    extension.withValue(algorithm.name());

    pmml.withExtensions(extension);
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
static
public List<?> getContent(HasExtensions<?> hasExtensions){
	List<Extension> extensions = hasExtensions.getExtensions();

	assertEquals(1, extensions.size());

	Extension extension = extensions.get(0);

	return extension.getContent();
}
 

import org.dmg.pmml.Extension; //导入依赖的package包/类
/**
 * @param pmml PMML model to add extension to, with no content. It may possibly duplicate
 *  existing extensions.
 * @param key extension key
 * @param value extension value
 */
public static void addExtension(PMML pmml, String key, Object value) {
  pmml.addExtensions(new Extension().setName(key).setValue(value.toString()));
}