@@ -114,19 +114,14 @@ def rbs_files(options = {})
114
114
Enumerator . new do |y |
115
115
prefix = options . fetch ( :prefix , '' )
116
116
codegenerated_plugins = codegen_plugins ( prefix )
117
- client_class = Views ::RBS ::ClientClass . new (
118
- service_name : @service . name ,
119
- codegenerated_plugins : codegenerated_plugins ,
120
- aws_sdk_core_lib_path : @aws_sdk_core_lib_path ,
121
- legacy_endpoints : @service . legacy_endpoints? ,
122
- signature_version : @service . signature_version ,
123
- api : @service . api ,
124
- waiters : @service . waiters ,
125
- protocol : @service . protocol ,
126
- add_plugins : @service . add_plugins ,
127
- remove_plugins : @service . remove_plugins ,
128
- )
117
+ client_class = client_class_rbs ( codegenerated_plugins )
129
118
y . yield ( "#{ prefix } /client.rbs" , client_class . render )
119
+ if @service . protocol_settings [ 'h2' ]
120
+ y . yield (
121
+ "#{ prefix } /async_client.rbs" ,
122
+ async_client_class_rbs ( codegenerated_plugins ) . render
123
+ )
124
+ end
130
125
y . yield ( "#{ prefix } /errors.rbs" , Views ::RBS ::ErrorsModule . new (
131
126
service : @service
132
127
) . render )
@@ -135,7 +130,7 @@ def rbs_files(options = {})
135
130
client_class : client_class ,
136
131
api : @service . api ,
137
132
resources : @service . resources ,
138
- paginators : @service . paginators ,
133
+ paginators : @service . paginators
139
134
) . render )
140
135
y . yield ( "#{ prefix } /waiters.rbs" , Views ::RBS ::WaitersModule . new (
141
136
service_name : @service . name ,
@@ -206,6 +201,22 @@ def client_class(codegenerated_plugins)
206
201
) . render
207
202
end
208
203
204
+ def client_class_rbs ( codegenerated_plugins )
205
+ Views ::RBS ::ClientClass . new (
206
+ service_name : @service . name ,
207
+ codegenerated_plugins : codegenerated_plugins ,
208
+ aws_sdk_core_lib_path : @aws_sdk_core_lib_path ,
209
+ legacy_endpoints : @service . legacy_endpoints? ,
210
+ signature_version : @service . signature_version ,
211
+ api : @service . api ,
212
+ waiters : @service . waiters ,
213
+ protocol : @service . protocol ,
214
+ add_plugins : @service . add_plugins ,
215
+ remove_plugins : @service . remove_plugins ,
216
+ protocol_settings : @service . protocol_settings
217
+ )
218
+ end
219
+
209
220
def async_client_class ( codegenerated_plugins )
210
221
Views ::AsyncClientClass . new (
211
222
service_identifier : @service . identifier ,
@@ -226,6 +237,22 @@ def async_client_class(codegenerated_plugins)
226
237
) . render
227
238
end
228
239
240
+ def async_client_class_rbs ( codegenerated_plugins )
241
+ Views ::RBS ::AsyncClientClass . new (
242
+ service_name : @service . name ,
243
+ codegenerated_plugins : codegenerated_plugins ,
244
+ aws_sdk_core_lib_path : @aws_sdk_core_lib_path ,
245
+ legacy_endpoints : @service . legacy_endpoints? ,
246
+ signature_version : @service . signature_version ,
247
+ api : @service . api ,
248
+ protocol : @service . protocol ,
249
+ add_plugins : @service . add_plugins ,
250
+ remove_plugins : @service . remove_plugins ,
251
+ protocol_settings : @service . protocol_settings ,
252
+ async_client : true
253
+ )
254
+ end
255
+
229
256
def errors_module
230
257
Views ::ErrorsModule . new ( service : @service ) . render
231
258
end
0 commit comments