swim.io.mqtt.MqttSettings Maven / Gradle / Ivy

Go to download
// Copyright 2015-2019 SWIM.AI inc.
//
// 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
//
//     http://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 swim.io.mqtt;

import swim.codec.Debug;
import swim.codec.Format;
import swim.codec.Output;
import swim.io.SocketSettings;
import swim.io.TcpSettings;
import swim.io.TlsSettings;
import swim.structure.Form;
import swim.structure.Item;
import swim.structure.Kind;
import swim.structure.Record;
import swim.structure.Value;
import swim.util.Murmur3;

public class MqttSettings implements Debug {
  protected final SocketSettings socketSettings;
  protected final int maxPayloadSize;

  public MqttSettings(SocketSettings socketSettings, int maxPayloadSize) {
    this.socketSettings = socketSettings;
    this.maxPayloadSize = maxPayloadSize;
  }

  public final SocketSettings socketSettings() {
    return this.socketSettings;
  }

  public MqttSettings socketSettings(SocketSettings socketSettings) {
    return copy(socketSettings, this.maxPayloadSize);
  }

  public final TlsSettings tlsSettings() {
    return this.socketSettings.tlsSettings();
  }

  public MqttSettings tlsSettings(TlsSettings tlsSettings) {
    return socketSettings(this.socketSettings.tlsSettings(tlsSettings));
  }

  public final TcpSettings tcpSettings() {
    return this.socketSettings.tcpSettings();
  }

  public MqttSettings tcpSettings(TcpSettings tcpSettings) {
    return socketSettings(this.socketSettings.tcpSettings(tcpSettings));
  }

  public final int maxPayloadSize() {
    return this.maxPayloadSize;
  }

  public MqttSettings maxPayloadSize(int maxPayloadSize) {
    return copy(this.socketSettings, maxPayloadSize);
  }

  protected MqttSettings copy(SocketSettings socketSettings, int maxPayloadSize) {
    return new MqttSettings(socketSettings, maxPayloadSize);
  }

  public Value toValue() {
    return form().mold(this).toValue();
  }

  @Override
  public boolean equals(Object other) {
    if (this == other) {
      return true;
    } else if (other instanceof MqttSettings) {
      final MqttSettings that = (MqttSettings) other;
      return this.socketSettings.equals(that.socketSettings)
          && this.maxPayloadSize == that.maxPayloadSize;
    }
    return false;
  }

  @Override
  public int hashCode() {
    if (hashSeed == 0) {
      hashSeed = Murmur3.seed(MqttSettings.class);
    }
    return Murmur3.mash(Murmur3.mix(Murmur3.mix(hashSeed,
        this.socketSettings.hashCode()), this.maxPayloadSize));
  }

  @Override
  public void debug(Output output) {
    output = output.write("MqttSettings").write('.').write("standard").write('(').write(')')
        .write('.').write("socketSettings").write('(').debug(this.socketSettings).write(')')
        .write('.').write("maxPayloadSize").write('(').debug(this.maxPayloadSize).write(')');
  }

  @Override
  public String toString() {
    return Format.debug(this);
  }

  private static int hashSeed;

  private static MqttSettings standard;

  private static Form form;

  public static MqttSettings standard() {
    if (standard == null) {
      int maxPayloadSize;
      try {
        maxPayloadSize = Integer.parseInt(System.getProperty("swim.mqtt.max.payload.size"));
      } catch (NumberFormatException error) {
        maxPayloadSize = 16 * 1024 * 1024;
      }

      standard = new MqttSettings(SocketSettings.standard(), maxPayloadSize);
    }
    return standard;
  }

  public static MqttSettings from(SocketSettings socketSettings) {
    return standard().socketSettings(socketSettings);
  }

  @Kind
  public static Form form() {
    if (form == null) {
      form = new MqttSettingsForm();
    }
    return form;
  }
}

final class MqttSettingsForm extends Form {
  @Override
  public MqttSettings unit() {
    return MqttSettings.standard();
  }

  @Override
  public Class type() {
    return MqttSettings.class;
  }

  @Override
  public Item mold(MqttSettings settings) {
    if (settings != null) {
      final MqttSettings standard = MqttSettings.standard();
      final Record mqtt = Record.create(2).attr("mqtt");
      if (settings.maxPayloadSize != standard.maxPayloadSize) {
        mqtt.slot("maxPayloadSize", settings.maxPayloadSize);
      }
      return Record.of(mqtt).concat(SocketSettings.form().mold(settings.socketSettings));
    } else {
      return Item.extant();
    }
  }

  @Override
  public MqttSettings cast(Item item) {
    final Value value = item.toValue();
    final MqttSettings standard = MqttSettings.standard();
    int maxPayloadSize = standard.maxPayloadSize;
    for (Item member : value) {
      if (member.getAttr("mqtt").isDefined()) {
        maxPayloadSize = member.get("maxPayloadSize").intValue(maxPayloadSize);
      }
    }
    final SocketSettings socketSettings = SocketSettings.form().cast(value);
    return new MqttSettings(socketSettings, maxPayloadSize);
  }
}