io.github.davemeier82.homeautomation.hivemq.HiveMqMqttClient Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2021-2021 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.github.davemeier82.homeautomation.hivemq;

import com.hivemq.client.mqtt.datatypes.MqttQos;
import com.hivemq.client.mqtt.mqtt5.Mqtt5AsyncClient;
import com.hivemq.client.mqtt.mqtt5.Mqtt5ClientBuilder;
import com.hivemq.client.mqtt.mqtt5.message.auth.Mqtt5SimpleAuth;
import com.hivemq.client.mqtt.mqtt5.message.publish.Mqtt5Publish;
import io.github.davemeier82.homeautomation.core.event.EventPublisher;
import io.github.davemeier82.homeautomation.core.event.factory.EventFactory;
import io.github.davemeier82.homeautomation.core.mqtt.MqttClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.nio.ByteBuffer;
import java.util.Optional;
import java.util.concurrent.Executors;
import java.util.function.BiConsumer;

import static java.nio.charset.StandardCharsets.UTF_8;
import static java.util.UUID.randomUUID;

public class HiveMqMqttClient implements MqttClient {

  private static final Logger log = LoggerFactory.getLogger(HiveMqMqttClient.class);
  private final Mqtt5AsyncClient client;
  private final EventFactory eventFactory;
  private final EventPublisher eventPublisher;
  private final String subscriptionTopicPrefix;
  private boolean connected = false;

  public HiveMqMqttClient(
      EventFactory eventFactory,
      EventPublisher eventPublisher,
      String serverHost,
      int serverPort,
      String username,
      String password,
      String subscriptionTopicPrefix
  ) {
    this.eventFactory = eventFactory;
    this.eventPublisher = eventPublisher;
    this.subscriptionTopicPrefix = subscriptionTopicPrefix;

    Mqtt5ClientBuilder mqtt5ClientBuilder = com.hivemq.client.mqtt.MqttClient.builder()
                                                                             .useMqttVersion5()
                                                                             .identifier(randomUUID().toString())
                                                                             .executorConfig()
                                                                             .nettyExecutor(Executors.newVirtualThreadPerTaskExecutor())
                                                                             .applyExecutorConfig()
                                                                             .serverHost(serverHost)
                                                                             .serverPort(serverPort)
                                                                             .automaticReconnectWithDefaultConfig();

    if (username != null) {
      mqtt5ClientBuilder = mqtt5ClientBuilder.simpleAuth(
          Mqtt5SimpleAuth.builder()
                         .username(username)
                         .password(password.getBytes(UTF_8))
                         .build()
      );
    }

    client = mqtt5ClientBuilder.buildAsync();
  }

  @Override
  public boolean isConnected() {
    return connected;
  }

  private static void consume(BiConsumer> consumer, Mqtt5Publish publish) {
    try {
      consumer.accept(publish.getTopic().toString(), publish.getPayload());
    } catch (Throwable throwable) {
      // TODO: create ErrorEvent to allow sending push notifications
      log.error("error to consume message of topic: {}", publish.getTopic(), throwable);
    }
  }

  @Override
  public void connect() {
    client.connectWith()
          .willPublish().topic("homeautomation/will")
          .payload("homeautomation lost connection".getBytes())
          .applyWillPublish()
          .send()
          .whenComplete((connAck, throwable) -> {
            if (throwable != null) {
              log.error("failed to connect to server", throwable);
            } else {
              connected = true;
              eventPublisher.publishEvent(eventFactory.createMqttClientConnectedEvent(this));
            }
          });
  }

  @Override
  public void publish(String topic, String payload) {
    publish(topic, payload.getBytes(UTF_8));
  }

  @Override
  public void publish(String topic, byte[] payload) {
    client.publishWith()
          .topic(topic)
          .payload(payload)
          .qos(MqttQos.EXACTLY_ONCE)
          .send()
          .whenComplete((mqtt3Publish, throwable) -> {
            if (throwable != null) {
              log.error("failed to publish message to topic: {}", topic, throwable);
            }
          });
  }

  @Override
  public void subscribe(String topic, BiConsumer> consumer) {
    client.subscribeWith()
          .topicFilter(subscriptionTopicPrefix + topic)
          .callback(publish -> consume(consumer, publish))
          .send()
          .whenComplete((subAck, throwable) -> {
            if (throwable != null) {
              log.error("failed to subscribe to topic: {}", topic, throwable);
            } else {
              log.info("successfully subscribed to topic: {}", topic);
            }
          });
  }

  @Override
  public void disconnect() {
    connected = false;
    client.disconnect();
  }

}