Skip to content

Commit 6ac9a28

Browse files
committed
Do not require the bundle model to have a single service
1 parent 4cc20a7 commit 6ac9a28

1 file changed

Lines changed: 1 addition & 9 deletions

File tree

  • model-bundle/model-bundle-api/src/main/java/software/amazon/smithy/modelbundle/api

model-bundle/model-bundle-api/src/main/java/software/amazon/smithy/modelbundle/api/ModelBundles.java

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -53,17 +53,9 @@ private static Model getModel(SmithyBundle bundle) {
5353
}
5454
var b = model.toBuilder();
5555

56-
var serviceShapes = model.getServiceShapes();
57-
58-
if (serviceShapes.size() != 1) {
59-
throw new IllegalStateException("Expected exactly one service shape but got "
60-
+ serviceShapes.stream().map(ServiceShape::getId).toList());
61-
}
62-
63-
var service = serviceShapes.iterator().next();
56+
var serviceShape = model.expectShape(ShapeId.from(bundle.getServiceName()), ServiceShape.class);
6457

6558
// mix in the generic arg members
66-
var serviceShape = service.asServiceShape().get();
6759
var serviceBuilder = serviceShape.toBuilder();
6860
for (var opId : serviceShape.getAllOperations()) {
6961
var op = model.expectShape(opId, OperationShape.class);

0 commit comments

Comments
 (0)