com.thing2x.smqd.SessionStore.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of smqd-core_2.12 Show documentation
smqd-core
There is a newer version: 0.5.0
// Copyright 2018 UANGEL
//
// 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 com.thing2x.smqd

import com.thing2x.smqd.QoS.QoS
import com.thing2x.smqd.SessionStore.{ClientData, InitialData, SessionStoreToken, SubscriptionData}
import com.typesafe.scalalogging.StrictLogging

import scala.concurrent.Future

// 2018. 7. 2. - Created by Kwon, Yeong Eon

object SessionStore {

  trait SessionStoreToken {
    def clientId: ClientId
    def cleanSession: Boolean
  }

  case class ClientData(clientId: ClientId, subscriptions: Seq[SubscriptionData], pendingMessageSize: Long)
  case class SubscriptionData(filterPath: FilterPath, qos: QoS)
  case class MessageData(topicPath: TopicPath, qos: QoS, msgId: Int, msg: Any, var lastTryTime: Long, var acked: Boolean = false)
  case class InitialData(token: SessionStoreToken, subscriptions: Seq[SubscriptionData])
}

class SessionStore(smqdInstance: Smqd, delegate: SessionStoreDelegate) extends StrictLogging {

  def createSession(clientId: ClientId, cleanSession: Boolean): Future[InitialData] =
    delegate.createSession(clientId, cleanSession)

  def flushSession(token: SessionStoreToken): Future[SmqResult] =
    delegate.flushSession(token)

  def saveSubscription(token: SessionStoreToken, filterPath: FilterPath, qos: QoS): Future[SmqResult] =
    delegate.saveSubscription(token, filterPath, qos)

  def deleteSubscription(token: SessionStoreToken, filterPath: FilterPath): Future[SmqResult] =
    delegate.deleteSubscription(token, filterPath)

  def loadSubscriptions(token: SessionStoreToken): Future[Seq[SubscriptionData]] =
    delegate.loadSubscriptions(token)

  def storeBeforeDelivery(token: SessionStoreToken, topicPath: TopicPath, qos: QoS, isRetain: Boolean, msgId: Int, msg: Any): Future[SmqResult] =
    delegate.storeBeforeDelivery(token, topicPath, qos, isRetain, msgId, msg)

  // QoS = 1
  def deleteAfterDeliveryAck(token: SessionStoreToken, msgId: Int): Future[SmqResult] =
    delegate.deleteAfterDeliveryAck(token, msgId)

  // QoS = 2
  def updateAfterDeliveryAck(token: SessionStoreToken, msgId: Int): Future[SmqResult] =
    delegate.updateAfterDeliveryAck(token, msgId)

  // QoS = 2
  def deleteAfterDeliveryComplete(token: SessionStoreToken, msgId: Int): Future[SmqResult] =
    delegate.deleteAfterDeliveryAck(token, msgId)

  def snapshot(search: Option[String]): Future[Seq[ClientData]] =
    delegate.snapshot(search)

  def setSessionState(clientId: ClientId, connected: Boolean): Future[SmqResult] = {
    delegate.setSessionState(clientId, connected)
  }
}