cn.dustlight.flow.zeebe.services.ZeebeMessageService Maven / Gradle / Ivy

Go to download
package cn.dustlight.flow.zeebe.services;

import io.camunda.zeebe.client.ZeebeClient;
import cn.dustlight.flow.core.flow.message.MessageService;
import reactor.core.publisher.Mono;

public class ZeebeMessageService implements MessageService {

    private ZeebeClient zeebeClient;

    public ZeebeMessageService(ZeebeClient zeebeClient) {
        this.zeebeClient = zeebeClient;
    }

    @Override
    public Mono publishMessage(String clientId, String messageName, String key) {
        return Mono.create(sink ->
                sink.onRequest(unused ->
                        zeebeClient.newPublishMessageCommand()
                                .messageName(String.format("c%s-%s", clientId, messageName))
                                .correlationKey(key)
                                .send()
                                .whenComplete(((publishMessageResponse, throwable) -> {
                                    if (throwable != null)
                                        sink.error(throwable);
                                    else
                                        sink.success();
                                }))));
    }


}