io.ray.serve.poll.LongPollRequest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ray-serve Show documentation

java for ray serve

The newest version!

package io.ray.serve.poll;

import java.io.Serializable;
import java.util.Map;

public class LongPollRequest implements Serializable {

  private static final long serialVersionUID = 6700308335001848617L;

  private Map keysToSnapshotIds;

  public LongPollRequest(Map keysToSnapshotIds) {
    this.keysToSnapshotIds = keysToSnapshotIds;
  }

  public Map getKeysToSnapshotIds() {
    return keysToSnapshotIds;
  }

  public io.ray.serve.generated.LongPollRequest toProtobuf() {
    io.ray.serve.generated.LongPollRequest.Builder builder =
        io.ray.serve.generated.LongPollRequest.newBuilder();
    if (keysToSnapshotIds != null) {
      keysToSnapshotIds.entrySet().stream()
          .forEach(
              entry -> builder.putKeysToSnapshotIds(entry.getKey().toString(), entry.getValue()));
    }
    return builder.build();
  }
}