/src/grpc-swift/Sources/GRPC/EventLoopFuture+RecoverFromUncleanShutdown.swift
Line | Count | Source |
1 | | /* |
2 | | * Copyright 2021, gRPC Authors All rights reserved. |
3 | | * |
4 | | * Licensed under the Apache License, Version 2.0 (the "License"); |
5 | | * you may not use this file except in compliance with the License. |
6 | | * You may obtain a copy of the License at |
7 | | * |
8 | | * http://www.apache.org/licenses/LICENSE-2.0 |
9 | | * |
10 | | * Unless required by applicable law or agreed to in writing, software |
11 | | * distributed under the License is distributed on an "AS IS" BASIS, |
12 | | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
13 | | * See the License for the specific language governing permissions and |
14 | | * limitations under the License. |
15 | | */ |
16 | | import NIOCore |
17 | | |
18 | | extension EventLoopFuture where Value == Void { |
19 | 0 | internal func recoveringFromUncleanShutdown() -> EventLoopFuture<Void> { |
20 | 0 | // We can ignore unclean shutdown since gRPC is self-terminated and therefore not prone to |
21 | 0 | // truncation attacks. |
22 | 0 | return self.flatMapErrorThrowing { error in |
23 | 0 | if error.isNIOSSLUncleanShutdown { |
24 | 0 | return () |
25 | 0 | } else { |
26 | 0 | throw error |
27 | 0 | } |
28 | 0 | } |
29 | 0 | } |
30 | | } |