/
ErrorExampleServer.java
64 lines (59 loc) · 2.73 KB
/
ErrorExampleServer.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
/*
* Copyright © 2021 Apple Inc. and the ServiceTalk project authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.servicetalk.examples.grpc.errors;
import io.servicetalk.concurrent.api.Single;
import io.servicetalk.grpc.api.GrpcServiceContext;
import io.servicetalk.grpc.api.GrpcStatusException;
import io.servicetalk.grpc.netty.GrpcServers;
import com.google.rpc.BadRequest;
import com.google.rpc.BadRequest.FieldViolation;
import com.google.rpc.Status;
import io.grpc.examples.errors.Greeter.GreeterService;
import io.grpc.examples.errors.HelloReply;
import io.grpc.examples.errors.HelloRequest;
import static com.google.protobuf.Any.pack;
import static io.servicetalk.concurrent.api.Single.failed;
import static io.servicetalk.concurrent.api.Single.succeeded;
import static io.servicetalk.grpc.api.GrpcStatusCode.INVALID_ARGUMENT;
/**
* Extends the async "Hello World" example to include support for application error propagation.
*/
public final class ErrorExampleServer {
public static void main(String... args) throws Exception {
GrpcServers.forPort(8080)
.listenAndAwait(new ErrorGreeterService())
.awaitShutdown();
}
private static final class ErrorGreeterService implements GreeterService {
@Override
public Single<HelloReply> sayHello(final GrpcServiceContext ctx, final HelloRequest request) {
if (request.getToken().isEmpty()) {
return failed(GrpcStatusException.of(Status.newBuilder()
.setCode(INVALID_ARGUMENT.value())
.setMessage("example message for invalid argument")
.addDetails(pack(BadRequest.newBuilder().addFieldViolations(
FieldViolation.newBuilder()
.setField(HelloRequest.getDescriptor().findFieldByNumber(2).getFullName())
.setDescription("missing required field!")
.build())
.build()))
.build()));
} else {
return succeeded(HelloReply.newBuilder().setMessage("hello " + request.getName()).build());
}
}
}
}