93 lines
3.5 KiB
Kotlin
93 lines
3.5 KiB
Kotlin
package org.session.libsession.messaging.jobs
|
|
|
|
import nl.komponents.kovenant.Promise
|
|
import nl.komponents.kovenant.deferred
|
|
import org.session.libsession.messaging.sending_receiving.MessageReceiver
|
|
import org.session.libsession.messaging.sending_receiving.handle
|
|
import org.session.libsession.messaging.utilities.Data
|
|
import org.session.libsignal.utilities.Log
|
|
|
|
class MessageReceiveJob(val data: ByteArray, val serverHash: String? = null, val openGroupMessageServerID: Long? = null, val openGroupID: String? = null) : Job {
|
|
override var delegate: JobDelegate? = null
|
|
override var id: String? = null
|
|
override var failureCount: Int = 0
|
|
|
|
override val maxFailureCount: Int = 10
|
|
companion object {
|
|
val TAG = MessageReceiveJob::class.simpleName
|
|
val KEY: String = "MessageReceiveJob"
|
|
|
|
// Keys used for database storage
|
|
private val DATA_KEY = "data"
|
|
private val SERVER_HASH_KEY = "serverHash"
|
|
private val OPEN_GROUP_MESSAGE_SERVER_ID_KEY = "openGroupMessageServerID"
|
|
private val OPEN_GROUP_ID_KEY = "open_group_id"
|
|
}
|
|
|
|
override fun execute() {
|
|
executeAsync().get()
|
|
}
|
|
|
|
fun executeAsync(): Promise<Unit, Exception> {
|
|
val deferred = deferred<Unit, Exception>()
|
|
try {
|
|
val isRetry: Boolean = failureCount != 0
|
|
val (message, proto) = MessageReceiver.parse(this.data, this.openGroupMessageServerID)
|
|
message.serverHash = serverHash
|
|
MessageReceiver.handle(message, proto, this.openGroupID)
|
|
this.handleSuccess()
|
|
deferred.resolve(Unit)
|
|
} catch (e: Exception) {
|
|
Log.e(TAG, "Couldn't receive message.", e)
|
|
if (e is MessageReceiver.Error && !e.isRetryable) {
|
|
Log.e("Loki", "Message receive job permanently failed.", e)
|
|
this.handlePermanentFailure(e)
|
|
} else {
|
|
Log.e("Loki", "Couldn't receive message.", e)
|
|
this.handleFailure(e)
|
|
}
|
|
deferred.resolve(Unit) // The promise is just used to keep track of when we're done
|
|
}
|
|
return deferred.promise
|
|
}
|
|
|
|
private fun handleSuccess() {
|
|
delegate?.handleJobSucceeded(this)
|
|
}
|
|
|
|
private fun handlePermanentFailure(e: Exception) {
|
|
delegate?.handleJobFailedPermanently(this, e)
|
|
}
|
|
|
|
private fun handleFailure(e: Exception) {
|
|
delegate?.handleJobFailed(this, e)
|
|
}
|
|
|
|
override fun serialize(): Data {
|
|
val builder = Data.Builder().putByteArray(DATA_KEY, data)
|
|
serverHash?.let { builder.putString(SERVER_HASH_KEY, it) }
|
|
openGroupMessageServerID?.let { builder.putLong(OPEN_GROUP_MESSAGE_SERVER_ID_KEY, it) }
|
|
openGroupID?.let { builder.putString(OPEN_GROUP_ID_KEY, it) }
|
|
return builder.build();
|
|
}
|
|
|
|
override fun getFactoryKey(): String {
|
|
return KEY
|
|
}
|
|
|
|
class Factory: Job.Factory<MessageReceiveJob> {
|
|
|
|
override fun create(data: Data): MessageReceiveJob {
|
|
val dataArray = data.getByteArray(DATA_KEY)
|
|
val serverHash = data.getStringOrDefault(SERVER_HASH_KEY, null)
|
|
val openGroupMessageServerID = data.getLongOrDefault(OPEN_GROUP_MESSAGE_SERVER_ID_KEY, -1).let { if (it == -1L) null else it }
|
|
val openGroupID = data.getStringOrDefault(OPEN_GROUP_ID_KEY, null)
|
|
return MessageReceiveJob(
|
|
dataArray,
|
|
serverHash,
|
|
openGroupMessageServerID,
|
|
openGroupID
|
|
)
|
|
}
|
|
}
|
|
} |