the-algorithm/product-mixer/component-library/src/main/scala/com/twitter/product_mixer/component_library/side_effect/UserSessionStoreUpdateSideEffect.scala
twitter-team ef4c5eb65e Twitter Recommendation Algorithm
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
2023-03-31 17:36:31 -05:00

36 lines
1.1 KiB
Scala

package com.twitter.product_mixer.component_library.side_effect
import com.twitter.product_mixer.core.functional_component.side_effect.PipelineResultSideEffect
import com.twitter.product_mixer.core.model.marshalling.HasMarshalling
import com.twitter.product_mixer.core.pipeline.PipelineQuery
import com.twitter.stitch.Stitch
import com.twitter.user_session_store.ReadWriteUserSessionStore
import com.twitter.user_session_store.WriteRequest
/**
* A [[PipelineResultSideEffect]] that writes to a [[ReadWriteUserSessionStore]]
*/
trait UserSessionStoreUpdateSideEffect[
Request <: WriteRequest,
Query <: PipelineQuery,
ResponseType <: HasMarshalling]
extends PipelineResultSideEffect[Query, ResponseType] {
/**
* Build the write request from the query
* @param query PipelineQuery
* @return WriteRequest
*/
def buildWriteRequest(query: Query): Option[Request]
val userSessionStore: ReadWriteUserSessionStore
final override def apply(
inputs: PipelineResultSideEffect.Inputs[Query, ResponseType]
): Stitch[Unit] = {
buildWriteRequest(inputs.query)
.map(userSessionStore.write)
.getOrElse(Stitch.Unit)
}
}