com.navercorp.pinpoint.grpc.server.flowcontrol.DefaultServerCallWrapper Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of pinpoint-grpc Show documentation

There is a newer version: 3.0.0

package com.navercorp.pinpoint.grpc.server.flowcontrol;

import io.grpc.Metadata;
import io.grpc.ServerCall;
import io.grpc.Status;

import java.util.Objects;

public class DefaultServerCallWrapper implements ServerCallWrapper {
    private final ServerCall serverCall;
    private final String agentId;
    private final String applicationName;

    public DefaultServerCallWrapper(ServerCall serverCall, String applicationName, String agentId) {
        this.serverCall = Objects.requireNonNull(serverCall, "serverCall");
        this.applicationName = Objects.requireNonNull(applicationName, "applicationName");
        this.agentId = Objects.requireNonNull(agentId, "agentId");
    }

    @Override
    public void request(int numMessages) {
        this.serverCall.request(numMessages);
    }

    @Override
    public String getAgentId() {
        return agentId;
    }

    @Override
    public String getApplicationName() {
        return applicationName;
    }

    @Override
    public void cancel(Status status, Metadata trailers) {
        this.serverCall.close(status, new Metadata());
    }

    @Override
    public String toString() {
        return "DefaultServerCallWrapper{" +
                "serverCall=" + serverCall +
                ", agentId='" + agentId + '\'' +
                ", applicationName='" + applicationName + '\'' +
                '}';
    }
}