silentbalanceyh/vertx-zero

package io.vertx.mod.plugin.rpc.client;

import io.grpc.ManagedChannel;
import io.vertx.core.Future;
import io.vertx.core.json.JsonObject;
import io.vertx.mod.ipc.eon.IpcRequest;
import io.vertx.mod.ipc.service.VertxUnityServiceGrpc;
import io.vertx.mod.plugin.rpc.RpcRepdor;
import io.vertx.up.atom.agent.IpcData;
import io.vertx.up.uca.micro.ipc.DataEncap;

/**
 * Used by rpc client.
 */
public class UnityStub implements RpcStub {

    private final transient ManagedChannel channel;

    public UnityStub(final ManagedChannel channel) {
        this.channel = channel;
    }

    @Override
    public Future<JsonObject> traffic(final IpcData data) {

        final VertxUnityServiceGrpc.UnityServiceVertxStub stub
            = VertxUnityServiceGrpc.newVertxStub(this.channel);
        // Request
        final IpcRequest request = DataEncap.in(data);
        return stub.unityCall(request)
            .compose(RpcRepdor.create(this.getClass())::replyJson);
        // Call and return to future
/*        final Promise<JsonObject> handler = Promise.promise();
        stub.unityCall(request, response ->
            // Reply
            RpcRepdor.create(this.getClass()).replyJson(handler, response));
        return handler.future();*/
    }
}