pw_rpc/java/main/dev/pigweed/pw_rpc/Service.java

// Copyright 2021 The Pigweed 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
//
//     https://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 dev.pigweed.pw_rpc;

import com.google.common.collect.ImmutableMap;
import com.google.protobuf.MessageLite;
import java.util.Arrays;
import java.util.stream.Collectors;

/** Represents an RPC service: a collection of related methods. */
public class Service {
  private final String name;
  private final int id;
  private final ImmutableMap<Integer, Method> methods;

  public Service(String name, Method.Builder... methodBuilders) {
    this.name = name;
    this.id = Ids.calculate(name);
    Arrays.stream(methodBuilders).forEach(m -> m.setService(this));
    this.methods = ImmutableMap.copyOf(Arrays.stream(methodBuilders)
                                           .map(Method.Builder::build)
                                           .collect(Collectors.toMap(Method::id, m -> m)));
  }

  public String name() {
    return name;
  }

  public int id() {
    return id;
  }

  public ImmutableMap<Integer, Method> methods() {
    return methods;
  }

  public final Method method(String name) {
    return methods().get(Ids.calculate(name));
  }

  @Override
  public final String toString() {
    return name();
  }

  public static Method.Builder unaryMethod(
      String name, Class<? extends MessageLite> request, Class<? extends MessageLite> response) {
    return Method.builder()
        .setType(Method.Type.UNARY)
        .setName(name)
        .setRequest(request)
        .setResponse(response);
  }

  public static Method.Builder serverStreamingMethod(
      String name, Class<? extends MessageLite> request, Class<? extends MessageLite> response) {
    return Method.builder()
        .setType(Method.Type.SERVER_STREAMING)
        .setName(name)
        .setRequest(request)
        .setResponse(response);
  }

  public static Method.Builder clientStreamingMethod(
      String name, Class<? extends MessageLite> request, Class<? extends MessageLite> response) {
    return Method.builder()
        .setType(Method.Type.CLIENT_STREAMING)
        .setName(name)
        .setRequest(request)
        .setResponse(response);
  }

  public static Method.Builder bidirectionalStreamingMethod(
      String name, Class<? extends MessageLite> request, Class<? extends MessageLite> response) {
    return Method.builder()
        .setType(Method.Type.BIDIRECTIONAL_STREAMING)
        .setName(name)
        .setRequest(request)
        .setResponse(response);
  }
}