Java STGroupFile类代码示例

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
static String getSpecialCaseSetupBody(SchemaColumn schemaColumn) {
    STGroup group = new STGroupFile("eclectic/orc/template/methodSpecialCaseSetup.stg");
    // Special case setup is needed for lists to adjust list child vector size.
    List<SchemaColumn> listSchemaTypes = new ArrayList<>();
    Stack<SchemaColumn> structTypes = new Stack<>();
    structTypes.push(schemaColumn);
    while (!structTypes.isEmpty()) {
        for (SchemaColumn child : structTypes.pop().getComplexType().getStructChildren()) {
            if (child.getTypeInfo().isTypeStruct()) {
                structTypes.push(child);
            } else if (child.getTypeInfo().isTypeList()) {
                listSchemaTypes.add(child);
            }
        }
    }
    ST st = group.getInstanceOf("methodSpecialCaseSetup");
    st.add("list", listSchemaTypes);
    String s = st.render();
    logger.trace(s);
    return s;
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
@VisibleForTesting
String generateContent(URL staticFile, Api api) throws IOException {
    final STGroupFile stGroup = createSTGroup(staticFile);
    final String fileName = new File(staticFile.getPath()).getName();

    final ST st = stGroup.getInstanceOf("main");
    st.add("vendorName", vendorName);
    if (fileName.equals("ResourceClassMap.php.stg")) {
        st.add("package", TypeGenModel.TYPES);
    }
    if (fileName.equals("Config.php.stg")) {
        final String apiUri = api.getBaseUri().getTemplate();
        final String authUri = api.getSecuritySchemes().stream()
                .filter(securityScheme -> securityScheme.getSettings() instanceof OAuth20Settings)
                .map(securityScheme -> ((OAuth20Settings)securityScheme.getSettings()).getAccessTokenUri())
                .findFirst().orElse("");
        st.add("apiUri", apiUri);
        st.add("authUri", authUri);
    }
    return st.render();
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
private static STGroupFile getTemplateGroupFile(String templatePath) throws MalformedURLException {

        if (templatePath == null) {
             return null;
        }

        if (!templatePath.contains(".jar!")) {
            return new STGroupFile(templatePath);
        }

        String protocol = "jar:";
        if (templatePath.startsWith(protocol)) {
            protocol = "";
        }

        URL url = new URL(protocol + templatePath);
        return  new STGroupFile(url, "US-ASCII", '%', '%');
    }
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
private Content render(Renderable renderable, Path rootDir, String templateName) {
	String templateFile = rootDir.resolve(templateName+".stg").toAbsolutePath().toFile().toString();
	STGroupFile groupFile = new STGroupFile(templateFile,'$','$');
	initModelAdapter(groupFile);
	initRenderer(groupFile, renderable.context());
	
	ST template = Preconditions.checkNotNull(groupFile.getInstanceOf("page"),"page template is null: %s -> %s",templateName,groupFile);
	
	try {
		setContext(template, renderable);
		
		return Text.builder()
				.mimeType("text/html")
				.text(template.render())
				.build();
	} catch (RuntimeException rx) {
		throw new RuntimeException("template "+templateName, rx);
	}
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
private void initModelAdapter(STGroupFile groupFile) {
	TypesafeModelAdapter.of(MapLike.class, (m,name) -> m.get(name).orElseNull())
		.register(groupFile);
	
	TypesafeModelAdapter.of(PropertyTree.class, (t,name) -> t.find(Object.class, name).orElseNull())
		.register(groupFile);
	
	TypesafeModelAdapter.of(Optional.class, (o,name) -> {
			if (name.equals("is")) {
				return o.isPresent();
			}
			if (name.equals("get")) {
				return o.get();
			}
			return null;
		})
		.register(groupFile);
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
@Test
public void test2() {
 // Load the file
    final STGroup stGroup = new STGroupFile(CODEGEN_DIR + "file1.stg");

    // Pick the correct template
    final ST templateExample = stGroup.getInstanceOf("templateExample");

    // Pass on values to use when rendering
    templateExample.add("param", "Hello World");

    Integer k = 99;
    Object[] ar = new Object[] { "Ter", "Parr", k };
    templateExample.addAggr("items.{ firstName ,lastName, id }", ar); // add() uses varargs
    templateExample.addAggr("items.{firstName, lastName ,id}", ar);
    
    // Render
    final String render = templateExample.render();

    // Print
    System.out.println(render);
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
public static void test2() throws IOException { // test rig
     String templates =
"t1(q1=\"Some\\nText\") ::= <<\n" +
"<q1>\n" +
">>\n" +
"\n" +
"t2(p1) ::= <<\n" +
"<p1>\n" +
">>\n" +
"\n" +
"main() ::= <<\n" +
"START-<t1()>-END\n" +
"\n" +
"START-<t2(p1=\"Some\\nText\")>-END\n" +
">>\n";

     String tmpdir = System.getProperty("java.io.tmpdir");
     writeFile(tmpdir, "t.stg", templates);
     STGroup group = new STGroupFile(tmpdir+"/"+"t.stg");
     ST st = group.getInstanceOf("main");
     STViz viz = st.inspect();
 }
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
private void generateFeature(AutoIndentWriter writer, Repository repository, FlowType flow,
    ActorType actor) {
  final STGroupFile stFeatureGroup = new STGroupFile("templates/feature.stg");

  ST st = stFeatureGroup.getInstanceOf("feature");
  st.add("repository", repository);
  st.add("flow", flow);
  st.add("actor", actor);
  st.write(writer, errorListener);
  
  repository.getMessages().getMessage().stream()
  .filter(m -> flow.getName().equals(m.getFlow())).forEach(message -> {
    Responses responses = message.getResponses();
    if (responses != null)
      responses.getResponse().forEach(response -> generateFeatureScenario(writer, stFeatureGroup, repository, actor, message, response));
  });
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
private void generateFeatureScenario(AutoIndentWriter writer, STGroupFile stFeatureGroup, Repository repository, ActorType actor,
    MessageType message, ResponseType response) {
  ST st = stFeatureGroup.getInstanceOf("scenario");
  st.add("actor", actor);
  st.add("message", message);
  st.add("response", response);
  st.write(writer, errorListener);
  List<Object> messageElements = message.getStructure().getComponentOrComponentRefOrGroup();
  generateFeatureMessageElements(writer, stFeatureGroup, messageElements);
  List<Object> actions = response.getMessageRefOrAssignOrTrigger();
  for (Object action : actions) {
    if (action instanceof MessageRefType) {
      MessageRefType messageRef = (MessageRefType) action;
      ST stMessage = stFeatureGroup.getInstanceOf("messageRef");
      stMessage.add("actor", actor);
      stMessage.add("messageRef", messageRef);
      stMessage.write(writer, errorListener);

      MessageType responseMessage = findMessage(messageRef.getName(), messageRef.getScenario());
      List<Object> responseMessageElements =
          responseMessage.getStructure().getComponentOrComponentRefOrGroup();
      generateFeatureMessageElements(writer, stFeatureGroup, responseMessageElements);
    }
  }
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
protected void generateCodeForFoldersInIndex(STGroup targetGroup) {
	File targetFolder = getOutputDir(testTemplates+"");
	STGroup index = new STGroupFile(testTemplates+"/Index.stg");
	index.load(); // make sure the index group is loaded since we call rawGetDictionary
	Map<String, Object> folders = index.rawGetDictionary("TestFolders");
	if (folders != null) {
		for (String key : folders.keySet()) {
			final String testDir = testTemplates + "/" + key;
			STGroup testIndex = new STGroupFile(testDir + "/Index.stg");
			testIndex.load();
			Map<String, Object> templateNames = testIndex.rawGetDictionary("TestTemplates");
			if ( templateNames != null && !templateNames.isEmpty() ) {
				final ArrayList<String> sortedTemplateNames = new ArrayList<String>(templateNames.keySet());
				Collections.sort(sortedTemplateNames);
				generateTestFile(testIndex, targetGroup, testDir, sortedTemplateNames, targetFolder);
			}
		}
	}
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
public void run() throws IOException {

    // preload the domains
    for (ParameterDomain pd : domains) {
      pd.load();
    }

    // get the template file
    STGroup templateGroup = new STGroupFile(templateFilename, '$', '$');

    // get the printer
    Printer printer = new SysoutPrinter(templateGroup, contentTemplateName);
    if (filenameTemplateName != null) {
      printer =
          new FilePrinter(templateGroup, contentTemplateName, filenameTemplateName, outputDirname);
    }

    Generator generator = new CartesienGenerator(domains, printer);
    if (generationStrategy.equals("linear")) {
      generator = new LinearGenerator(domains, printer);
    }
    generator.generate();
  }
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
/**
 * Initialize the Duzzt embedded DSL generator.
 * 
 * @param reporter the reporter used to report errors
 * @throws DuzztInitializationException if a fatal error occurs during initialization
 */
public void init(Reporter reporter) throws DuzztInitializationException {
	URL url = GenerateEDSLProcessor.class.getResource(ST_RESOURCE_NAME);
	
	this.sourceGenGroup = new STGroupFile(
			url,
			ST_ENCODING,
			ST_DELIM_START_CHAR,
			ST_DELIM_STOP_CHAR);
	
	sourceGenGroup.setListener(new ReporterDiagnosticListener(reporter));
	
	sourceGenGroup.load();
	if(!sourceGenGroup.isDefined(ST_MAIN_TEMPLATE_NAME)) {
		sourceGenGroup = null;
		throw new DuzztInitializationException("Could not find main template '"
				+ ST_MAIN_TEMPLATE_NAME + "' in template group file " + url.toString());
	}
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
@Override
public void execute() throws MojoExecutionException{
	try {
		STGroup template = new STGroupFile(TEMPLATES_THRIFT_POJO);
		JavaDocBuilder docBuilder = getBuilder();

		getLog().info(
				String.format("Executing ThriftPojoGenerator. Sources [ %s ]; outputDirectory [ %s ]; postfix [ %s ]; inteface [ %s ]", sources,
						outputDirectory, outputPostfix, interfaceName));

		generate(template, docBuilder);

		project.addCompileSourceRoot(outputDirectory.getAbsolutePath());
	} catch (Exception e) {
		getLog().error("General error", e);
           throw new MojoExecutionException("Error generatings Pojos", e);
	}
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
@Override
public void render(PetriNet pn, Writer writer) throws RenderException, IOException {
	verifyNet(pn);

	STGroup group = new STGroupFile("uniol/apt/io/renderer/impl/LoLAPN.stg");
	ST pnTemplate = group.getInstanceOf("pn");
	pnTemplate.add("name", pn.getName());

	// Handle places
	pnTemplate.add("places", pn.getPlaces());

	// Handle the initial marking
	for (Place p : pn.getPlaces()) {
		Token val = pn.getInitialMarking().getToken(p);
		if (val.getValue() != 0) {
			pnTemplate.addAggr("marking.{place, weight}", p, val.getValue());
		}
	}

	// Handle transitions (and arcs)
	pnTemplate.add("transitions", pn.getTransitions());

	pnTemplate.write(new AutoIndentWriter(writer), new ThrowingErrorListener());
}
 

import org.stringtemplate.v4.STGroupFile; //导入依赖的package包/类
@Override
public void render(TransitionSystem ts, Writer writer) throws RenderException, IOException {
	STGroup group = new STGroupFile("uniol/apt/io/renderer/impl/AptLTS.stg");
	ST ltsTemplate = group.getInstanceOf("lts");

	ltsTemplate.add("name", ts.getName());
	if (ts.hasExtension("description")) {
		ltsTemplate.add("description", ts.getExtension("description"));
	}
	ltsTemplate.add("extensions", ts.getWriteToFileExtensions());
	ltsTemplate.add("states", decorateStates(ts));
	ltsTemplate.add("arcs", ts.getEdges());
	ltsTemplate.add("events", ts.getAlphabetEvents());

	ltsTemplate.write(new AutoIndentWriter(writer), new ThrowingErrorListener());
}