Skip to content

Commit a5b57af

Browse files
kuhekuhe
committed
feat(cbor): add protocol resolution priority system
1 parent 5865b65 commit a5b57af

File tree

4 files changed

+204
-6
lines changed

4 files changed

+204
-6
lines changed

‎smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/DirectedTypeScriptCodegen.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
import java.nio.file.Paths;
1919
import java.util.ArrayList;
2020
import java.util.Collection;
21-
import java.util.HashMap;
21+
import java.util.LinkedHashMap;
2222
import java.util.List;
2323
import java.util.Map;
2424
import java.util.Set;
@@ -136,7 +136,9 @@ private ProtocolGenerator resolveProtocolGenerator(
136136
TypeScriptSettings settings
137137
) {
138138
// Collect all of the supported protocol generators.
139-
Map<ShapeId, ProtocolGenerator> generators = new HashMap<>();
139+
// Preserve insertion order as default priority order.
140+
Map<ShapeId, ProtocolGenerator> generators = new LinkedHashMap<>();
141+
140142
for (TypeScriptIntegration integration : integrations) {
141143
for (ProtocolGenerator generator : integration.getProtocolGenerators()) {
142144
generators.put(generator.getProtocol(), generator);

‎smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/TypeScriptSettings.java

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515

1616
package software.amazon.smithy.typescript.codegen;
1717

18+
import java.util.ArrayList;
1819
import java.util.Arrays;
1920
import java.util.Collections;
2021
import java.util.List;
@@ -36,6 +37,7 @@
3637
import software.amazon.smithy.model.shapes.ShapeId;
3738
import software.amazon.smithy.model.traits.DefaultTrait;
3839
import software.amazon.smithy.model.traits.RequiredTrait;
40+
import software.amazon.smithy.typescript.codegen.protocols.ProtocolPriority;
3941
import software.amazon.smithy.utils.SmithyUnstableApi;
4042

4143
/**
@@ -450,8 +452,13 @@ public ShapeId resolveServiceProtocol(Model model, ServiceShape service, Set<Sha
450452
+ "generate in smithy-build.json to generate this service.");
451453
}
452454

453-
return resolvedProtocols.stream()
454-
.filter(supportedProtocols::contains)
455+
List<ShapeId> protocolPriority = ProtocolPriority.getProtocolPriority(service.toShapeId());
456+
List<ShapeId> protocolPriorityList = protocolPriority != null && !protocolPriority.isEmpty()
457+
? protocolPriority
458+
: new ArrayList<>(supportedProtocols);
459+
460+
return protocolPriorityList.stream()
461+
.filter(resolvedProtocols::contains)
455462
.findFirst()
456463
.orElseThrow(() -> new UnresolvableProtocolException(String.format(
457464
"The %s service supports the following unsupported protocols %s. The following protocol "

‎smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/protocols/ProtocolPriority.java

Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
/*
2+
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
3+
* SPDX-License-Identifier: Apache-2.0
4+
*/
5+
6+
package software.amazon.smithy.typescript.codegen.protocols;
7+
8+
import java.util.ArrayList;
9+
import java.util.HashMap;
10+
import java.util.List;
11+
import java.util.Map;
12+
import software.amazon.smithy.model.shapes.ShapeId;
13+
14+
15+
/**
16+
* Allows customization of protocol selection for specific services or a global default ordering.
17+
*/
18+
public final class ProtocolPriority {
19+
private static final Map<ShapeId, List<ShapeId>> SERVICE_PROTOCOL_PRIORITY_CUSTOMIZATIONS = new HashMap<>();
20+
private static List<ShapeId> customDefaultPriority = null;
21+
22+
private ProtocolPriority() {}
23+
24+
/**
25+
* @param serviceShapeId - service scope.
26+
* @param protocolPriorityOrder - priority order of protocols.
27+
*/
28+
public static void setProtocolPriority(ShapeId serviceShapeId, List<ShapeId> protocolPriorityOrder) {
29+
SERVICE_PROTOCOL_PRIORITY_CUSTOMIZATIONS.put(serviceShapeId, protocolPriorityOrder);
30+
}
31+
32+
/**
33+
* @param defaultProtocolPriorityOrder - use for all services that don't have a more specific priority order.
34+
*/
35+
public static void setCustomDefaultProtocolPriority(List<ShapeId> defaultProtocolPriorityOrder) {
36+
customDefaultPriority = new ArrayList<>(defaultProtocolPriorityOrder);
37+
}
38+
39+
/**
40+
* @param serviceShapeId - service scope.
41+
* @return priority order of protocols or null if no override exists.
42+
*/
43+
public static List<ShapeId> getProtocolPriority(ShapeId serviceShapeId) {
44+
return SERVICE_PROTOCOL_PRIORITY_CUSTOMIZATIONS.getOrDefault(
45+
serviceShapeId,
46+
customDefaultPriority != null ? new ArrayList<>(customDefaultPriority) : null
47+
);
48+
}
49+
50+
/**
51+
* @param serviceShapeId - to unset.
52+
*/
53+
public static void deleteProtocolPriority(ShapeId serviceShapeId) {
54+
SERVICE_PROTOCOL_PRIORITY_CUSTOMIZATIONS.remove(serviceShapeId);
55+
}
56+
57+
/**
58+
* Unset the custom default priority order.
59+
*/
60+
public static void deleteCustomDefaultProtocolPriority() {
61+
customDefaultPriority = null;
62+
}
63+
}

‎smithy-typescript-codegen/src/test/java/software/amazon/smithy/typescript/codegen/TypeScriptSettingsTest.java

Lines changed: 128 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,31 @@
11
package software.amazon.smithy.typescript.codegen;
22

3+
import java.util.LinkedHashSet;
4+
import java.util.List;
5+
import java.util.stream.Stream;
36
import org.junit.jupiter.api.Test;
7+
import org.junit.jupiter.api.extension.ExtendWith;
48
import org.junit.jupiter.params.ParameterizedTest;
59
import org.junit.jupiter.params.provider.Arguments;
610
import org.junit.jupiter.params.provider.MethodSource;
11+
import org.mockito.Mock;
12+
import org.mockito.junit.jupiter.MockitoExtension;
713
import software.amazon.smithy.model.Model;
14+
import software.amazon.smithy.model.knowledge.ServiceIndex;
815
import software.amazon.smithy.model.node.Node;
916
import software.amazon.smithy.model.node.ObjectNode;
17+
import software.amazon.smithy.model.shapes.ServiceShape;
1018
import software.amazon.smithy.model.shapes.ShapeId;
11-
12-
import java.util.stream.Stream;
19+
import software.amazon.smithy.typescript.codegen.protocols.ProtocolPriority;
20+
import software.amazon.smithy.utils.MapUtils;
1321

1422
import static org.hamcrest.MatcherAssert.assertThat;
1523
import static org.hamcrest.Matchers.equalTo;
1624
import static org.junit.jupiter.api.Assertions.assertEquals;
25+
import static org.mockito.ArgumentMatchers.any;
26+
import static org.mockito.Mockito.when;
1727

28+
@ExtendWith(MockitoExtension.class)
1829
public class TypeScriptSettingsTest {
1930

2031
@Test
@@ -87,6 +98,121 @@ private static Stream<Arguments> providePackageDescriptionTestCases() {
8798
);
8899
}
89100

101+
@Test
102+
public void resolveServiceProtocol(@Mock Model model,
103+
@Mock ServiceShape service,
104+
@Mock ServiceIndex serviceIndex) {
105+
TypeScriptSettings subject = new TypeScriptSettings();
106+
107+
// note: these are mock protocol names.
108+
ShapeId rpcv2Cbor = ShapeId.from("namespace#rpcv2Cbor");
109+
ShapeId json1_0 = ShapeId.from("namespace#json1_0");
110+
ShapeId json1_1 = ShapeId.from("namespace#json1_1");
111+
ShapeId restJson1 = ShapeId.from("namespace#restJson1");
112+
ShapeId restXml = ShapeId.from("namespace#restXml");
113+
ShapeId query = ShapeId.from("namespace#query");
114+
ShapeId serviceQuery = ShapeId.from("namespace#serviceQuery");
115+
116+
when(model.getKnowledge(any(), any())).thenReturn(serviceIndex);
117+
ShapeId serviceShapeId = ShapeId.from("namespace#Service");
118+
when(service.toShapeId()).thenReturn(serviceShapeId);
119+
120+
LinkedHashSet<ShapeId> protocolShapeIds = new LinkedHashSet<>(
121+
List.of(
122+
json1_0, json1_1, restJson1, rpcv2Cbor, restXml, query, serviceQuery
123+
)
124+
);
125+
126+
{
127+
// spec case 1.
128+
when(serviceIndex.getProtocols(service)).thenReturn(MapUtils.of(
129+
rpcv2Cbor, null,
130+
json1_0, null
131+
));
132+
ShapeId protocol = subject.resolveServiceProtocol(model, service, protocolShapeIds);
133+
// Note: JS customization JSON higher default priority than CBOR.
134+
assertEquals(json1_0, protocol);
135+
}
136+
137+
{
138+
// spec case 2.
139+
when(serviceIndex.getProtocols(service)).thenReturn(MapUtils.of(
140+
rpcv2Cbor, null
141+
));
142+
ShapeId protocol = subject.resolveServiceProtocol(model, service, protocolShapeIds);
143+
assertEquals(rpcv2Cbor, protocol);
144+
}
145+
146+
{
147+
// spec case 3.
148+
when(serviceIndex.getProtocols(service)).thenReturn(MapUtils.of(
149+
rpcv2Cbor, null,
150+
json1_0, null,
151+
query, null
152+
));
153+
ShapeId protocol = subject.resolveServiceProtocol(model, service, protocolShapeIds);
154+
// Note: JS customization JSON higher default priority than CBOR.
155+
assertEquals(json1_0, protocol);
156+
}
157+
158+
{
159+
// spec case 4.
160+
when(serviceIndex.getProtocols(service)).thenReturn(MapUtils.of(
161+
json1_0, null,
162+
query, null
163+
));
164+
ShapeId protocol = subject.resolveServiceProtocol(model, service, protocolShapeIds);
165+
assertEquals(json1_0, protocol);
166+
}
167+
168+
{
169+
// spec case 5.
170+
when(serviceIndex.getProtocols(service)).thenReturn(MapUtils.of(
171+
query, null
172+
));
173+
ShapeId protocol = subject.resolveServiceProtocol(model, service, protocolShapeIds);
174+
assertEquals(query, protocol);
175+
}
176+
177+
{
178+
// service override, non-spec
179+
when(serviceIndex.getProtocols(service)).thenReturn(MapUtils.of(
180+
json1_0, null,
181+
json1_1, null,
182+
restJson1, null,
183+
rpcv2Cbor, null,
184+
restXml, null,
185+
query, null,
186+
serviceQuery, null
187+
));
188+
ProtocolPriority.setProtocolPriority(serviceShapeId, List.of(
189+
serviceQuery, rpcv2Cbor, json1_1, restJson1, restXml, query
190+
));
191+
ShapeId protocol = subject.resolveServiceProtocol(model, service, protocolShapeIds);
192+
ProtocolPriority.deleteProtocolPriority(serviceShapeId);
193+
assertEquals(serviceQuery, protocol);
194+
}
195+
196+
{
197+
// global default override
198+
when(serviceIndex.getProtocols(service)).thenReturn(MapUtils.of(
199+
json1_0, null,
200+
json1_1, null,
201+
restJson1, null,
202+
rpcv2Cbor, null,
203+
restXml, null,
204+
query, null,
205+
serviceQuery, null
206+
));
207+
ProtocolPriority.setCustomDefaultProtocolPriority(List.of(
208+
rpcv2Cbor, json1_1, restJson1, restXml, query
209+
));
210+
ShapeId protocol = subject.resolveServiceProtocol(model, service, protocolShapeIds);
211+
ProtocolPriority.deleteCustomDefaultProtocolPriority();
212+
assertEquals(rpcv2Cbor, protocol);
213+
}
214+
}
215+
90216
@Test
91217
public void resolvesSupportProtocols() {
92218
// TODO

0 commit comments

Comments
 (0)