io.ray.serve.dag.DeploymentNode 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.dag;

import io.ray.serve.deployment.Deployment;
import io.ray.serve.handle.DeploymentHandle;
import java.util.Map;

public class DeploymentNode extends DAGNode {

  private String appName;

  private Deployment deployment;

  private DeploymentHandle deploymentHandle;

  public DeploymentNode(
      Deployment deployment,
      String appName,
      Object[] deploymentInitArgs,
      Map rayActorOptions,
      Map otherArgsToResolve) {
    super(deploymentInitArgs, rayActorOptions, otherArgsToResolve);
    this.appName = appName;
    this.deployment = deployment;
    this.deploymentHandle = new DeploymentHandle(deployment.getName(), appName);
  }

  @Override
  public DAGNode copyImpl(
      Object[] newArgs, Map newOptions, Map newOtherArgsToResolve) {
    return new DeploymentNode(deployment, appName, newArgs, newOptions, newOtherArgsToResolve);
  }

  public String getAppName() {
    return appName;
  }

  public Deployment getDeployment() {
    return deployment;
  }

  public DeploymentHandle getDeploymentHandle() {
    return deploymentHandle;
  }
}