com.github.charlemaznable.bunny.client.eventbus.BunnyEventBus Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bunny-client Show documentation
Bunny rabbits will skip hand-in-hand with baby lambs across sunny green meadows.
There is a newer version: 2023.0.4
package com.github.charlemaznable.bunny.client.eventbus;

import com.github.charlemaznable.bunny.client.config.BunnyClientConfig;
import com.github.charlemaznable.bunny.client.domain.BunnyBaseRequest;
import com.github.charlemaznable.bunny.client.domain.BunnyBaseResponse;
import com.github.charlemaznable.core.codec.NonsenseSignature;
import com.github.charlemaznable.core.codec.nonsense.NonsenseOptions;
import com.github.charlemaznable.core.codec.signature.SignatureOptions;
import com.google.inject.Inject;
import io.vertx.core.AsyncResult;
import io.vertx.core.Handler;
import io.vertx.core.Vertx;
import io.vertx.core.eventbus.EventBus;
import lombok.val;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.Nullable;

import static com.github.charlemaznable.bunny.client.domain.BunnyBaseResponse.RESP_CODE_KEY;
import static com.github.charlemaznable.bunny.client.domain.BunnyBaseResponse.RESP_CODE_OK;
import static com.github.charlemaznable.core.codec.Json.json;
import static com.github.charlemaznable.core.codec.Json.spec;
import static com.github.charlemaznable.core.codec.Json.unJson;
import static com.github.charlemaznable.core.lang.Condition.checkNotNull;
import static com.github.charlemaznable.core.lang.Condition.notNullThen;
import static com.github.charlemaznable.core.lang.Condition.nullThen;
import static com.github.charlemaznable.core.miner.MinerFactory.getMiner;
import static io.vertx.core.Future.failedFuture;
import static io.vertx.core.Future.succeededFuture;

@Component
public final class BunnyEventBus {

    private final EventBus eventBus;
    private final BunnyClientConfig bunnyClientConfig;
    private final NonsenseSignature nonsenseSignature;

    @Inject
    @Autowired
    public BunnyEventBus(Vertx vertx,
                         @Nullable BunnyClientConfig bunnyClientConfig,
                         @Nullable NonsenseOptions nonsenseOptions,
                         @Nullable SignatureOptions signatureOptions) {
        this.eventBus = checkNotNull(vertx).eventBus();
        this.bunnyClientConfig = nullThen(bunnyClientConfig,
                () -> getMiner(BunnyClientConfig.class));
        this.nonsenseSignature = new NonsenseSignature();
        notNullThen(nonsenseOptions, this.nonsenseSignature::nonsenseOptions);
        notNullThen(signatureOptions, this.nonsenseSignature::signatureOptions);
    }

    public  void request
            (BunnyBaseRequest request, Handler> handler) {
        val requestMap = nonsenseSignature.sign(request);
        eventBus.request(bunnyClientConfig.eventBusAddressPrefix() + request.getBunnyAddress(),
                json(requestMap), asyncResult -> {
                    if (asyncResult.failed()) {
                        handler.handle(failedFuture(asyncResult.cause()));
                        return;
                    }

                    val responseMap = unJson(asyncResult.result().body());
                    if (!RESP_CODE_OK.equals(responseMap.get(RESP_CODE_KEY))) {
                        // 失败响应, 不验证签名
                        handler.handle(succeededFuture(
                                spec(responseMap, request.responseClass())));
                        return;
                    }

                    if (!nonsenseSignature.verify(responseMap)) {
                        handler.handle(failedFuture(
                                new BunnyEventBusException("Response verify failed")));
                        return;
                    }
                    handler.handle(succeededFuture(
                            spec(responseMap, request.responseClass())));
                });
    }
}