forked from openjdk/jdk
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
JimLaskey
committed
Mar 11, 2024
1 parent
5d8510b
commit fe0100c
Showing
4 changed files
with
229 additions
and
40 deletions.
There are no files selected for viewing
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
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
174 changes: 174 additions & 0 deletions
174
src/java.base/share/classes/jdk/internal/template/StringTemplateMetaData.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,174 @@ | ||
| /* | ||
| * Copyright (c) 2023, 2024, Oracle and/or its affiliates. All rights reserved. | ||
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. | ||
| * | ||
| * This code is free software; you can redistribute it and/or modify it | ||
| * under the terms of the GNU General Public License version 2 only, as | ||
| * published by the Free Software Foundation. Oracle designates this | ||
| * particular file as subject to the "Classpath" exception as provided | ||
| * by Oracle in the LICENSE file that accompanied this code. | ||
| * | ||
| * This code is distributed in the hope that it will be useful, but WITHOUT | ||
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | ||
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | ||
| * version 2 for more details (a copy is included in the LICENSE file that | ||
| * accompanied this code). | ||
| * | ||
| * You should have received a copy of the GNU General Public License version | ||
| * 2 along with this work; if not, write to the Free Software Foundation, | ||
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | ||
| * | ||
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA | ||
| * or visit www.oracle.com if you need additional information or have any | ||
| * questions. | ||
| */ | ||
|
|
||
| package jdk.internal.template; | ||
|
|
||
| import jdk.internal.access.JavaTemplateAccess; | ||
| import jdk.internal.access.SharedSecrets; | ||
| import jdk.internal.vm.annotation.Stable; | ||
|
|
||
| import java.lang.invoke.MethodHandle; | ||
| import java.lang.invoke.StringConcatException; | ||
| import java.lang.invoke.StringConcatFactory; | ||
| import java.util.ArrayList; | ||
| import java.util.List; | ||
| import java.util.function.Supplier; | ||
|
|
||
| /** | ||
| * This class provides a model for associating metadata by an owning class with a literal | ||
| * StringTemplate. | ||
| */ | ||
| public class StringTemplateMetaData { | ||
| private static final JavaTemplateAccess JTA = SharedSecrets.getJavaTemplateAccess(); | ||
|
|
||
| /** | ||
| * Per {@link StringTemplate} {@link MethodHandle}. | ||
| */ | ||
| @Stable | ||
| private MethodHandle methodHandle; | ||
|
|
||
| /** | ||
| * Public constructor. | ||
| * | ||
| * @param st associated {@link StringTemplate} | ||
| */ | ||
| public StringTemplateMetaData(StringTemplate st) { | ||
| this.methodHandle = null; | ||
| } | ||
|
|
||
| /** | ||
| * {@return the method handle} | ||
| */ | ||
| public MethodHandle methodHandle() { | ||
| return methodHandle; | ||
| } | ||
|
|
||
| /** | ||
| * Set the method handle. | ||
| * | ||
| * @param methodHandle the method handle to set | ||
| */ | ||
| public void setMethodHandle(MethodHandle methodHandle) { | ||
| this.methodHandle = methodHandle; | ||
| } | ||
|
|
||
| /** | ||
| * Construct the {@link MethodHandle}. Override if special handling is required. | ||
| * @param st associated {@link StringTemplate} | ||
| * @return | ||
| */ | ||
| public MethodHandle createMethodHandle(StringTemplate st) { | ||
| List<String> fragments = filterFragments(st.fragments()); | ||
| List<Class<?>> ptypes = JTA.getTypes(st); | ||
| MethodHandle[] filters = createValueFilters(ptypes); | ||
| List<Class<?>> fTypes = new ArrayList<>(); | ||
| for (int i = 0; i < filters.length; i++) { | ||
| if (filters[i] == null) { | ||
| fTypes.add(ptypes.get(i)); | ||
| } else { | ||
| fTypes.add(filters[i].type().returnType()); | ||
| } | ||
| } | ||
| MethodHandle mh = createConcat(fragments, fTypes); | ||
| mh = JTA.bindTo(st, mh); | ||
| return mh; | ||
| } | ||
|
|
||
| /** | ||
| * Filter the associated {@link StringTemplate StringTemplate's} fragments. Override | ||
| * if special handling is required. | ||
| * | ||
| * @param fragments associated {@link StringTemplate StringTemplate's} fragments | ||
| * | ||
| * @return filtered fragments | ||
| */ | ||
| public List<String> filterFragments(List<String> fragments) { | ||
| return fragments; | ||
| } | ||
|
|
||
| /** | ||
| * Create value filters for associated {@link StringTemplate StringTemplate's} values. | ||
| * Override if special handling is required. | ||
| * | ||
| * @param ptypes value types | ||
| * | ||
| * @return array of {@link MethodHandle} containing filters for each value or null if | ||
| * a value needs no filtering. | ||
| */ | ||
| public MethodHandle[] createValueFilters(List<Class<?>> ptypes) { | ||
| return new MethodHandle[ptypes.size()]; | ||
| } | ||
|
|
||
| /** | ||
| * Create the concat {@link MethodHandle}. | ||
| * | ||
| * @param fragments concat fragments | ||
| * @param ptypes concat value types | ||
| * | ||
| * @return {@link MethodHandle} that performs the basic concat. | ||
| */ | ||
| public MethodHandle createConcat(List<String> fragments, List<Class<?>> ptypes) { | ||
| MethodHandle mh; | ||
| try { | ||
| mh = StringConcatFactory.makeConcatWithTemplate(fragments, ptypes); | ||
| } catch (StringConcatException e) { | ||
| throw new InternalError(e); | ||
| } | ||
| return mh; | ||
| } | ||
|
|
||
| public String invoke(StringTemplate st) { | ||
| if (methodHandle != null) { | ||
| try { | ||
| return (String)methodHandle.invokeExact(st); | ||
| } catch (Throwable e) { | ||
| throw new InternalError(e); | ||
| } | ||
| } | ||
| return null; | ||
| } | ||
|
|
||
| /** | ||
| * Method to set up the metadata. Returns null if the {@link StringTemplate} has a | ||
| * different owner. | ||
| * | ||
| * @param st associated {@link StringTemplate} | ||
| * @param owner metadata owner | ||
| * @param supplier supplier to generate instance of metadata | ||
| * | ||
| * @return instance of the metadata class. | ||
| */ | ||
| public static <M extends StringTemplateMetaData> M getMetaData(StringTemplate st, Object owner, Supplier<M> supplier) { | ||
| if (JTA.isLiteral(st)) { | ||
| M metaData = JTA.getMetaData(st, owner, () -> { | ||
| M md = supplier.get(); | ||
| md.setMethodHandle(md.createMethodHandle(st)); | ||
| return md; | ||
| }); | ||
| return metaData; | ||
| } | ||
| return null; | ||
| } | ||
| } |