use reifydb_core::encoded::key::EncodedKey;
use serde::{Serialize, de::DeserializeOwned};
use super::FFIRawStatefulOperator;
use crate::{error::Result, operator::context::OperatorContext};
pub trait FFIWindowStateful: FFIRawStatefulOperator {
type State: Serialize + DeserializeOwned;
fn load_state(&self, ctx: &mut OperatorContext, window_key: &EncodedKey) -> Result<Option<Self::State>> {
ctx.state().get::<Self::State>(window_key)
}
fn save_state(&self, ctx: &mut OperatorContext, window_key: &EncodedKey, value: &Self::State) -> Result<()> {
ctx.state().set(window_key, value)
}
fn remove_window(&self, ctx: &mut OperatorContext, window_key: &EncodedKey) -> Result<()> {
ctx.state().remove(window_key)
}
}