prosody 0.2.1 → 0.4.0

data/ext/prosody/src/handler/message.rs

@@ -100,9 +100,26 @@ impl Message {
     /// # Errors
     ///
     /// Returns an error if payload deserialization fails.
+    ///
+    /// The public RBS exposes this JSON value as `Message[Payload]#payload`.
+    /// That generic parameter is static-only: this native boundary continues
+    /// to deserialize JSON into ordinary Ruby objects without runtime schema
+    /// validation.
     fn payload(ruby: &Ruby, this: &Self) -> Result<Value, Error> {
         serialize(ruby, this.inner.payload())
     }
+
+    /// Clones the wrapped `ConsumerMessage` for a message-collection write.
+    ///
+    /// The wrapper holds the real `ConsumerMessage`, so a message write clones
+    /// the inner value directly — a cheap operation.
+    ///
+    /// # Returns
+    ///
+    /// An owned clone of the wrapped `ConsumerMessage`.
+    pub(crate) fn consumer_message(&self) -> ConsumerMessage<serde_json::Value> {
+        self.inner.clone()
+    }
 }
 
 impl From<ConsumerMessage<serde_json::Value>> for Message {

data/ext/prosody/src/handler/mod.rs

@@ -38,6 +38,7 @@ use tracing_opentelemetry::OpenTelemetrySpanExt;
 
 mod context;
 mod message;
+mod state;
 mod trigger;
 
 /// A handler that bridges between Kafka messages and Ruby message processing
@@ -117,7 +118,7 @@ impl FallibleHandler for RubyHandler {
         _demand_type: DemandType,
     ) -> Result<(), Self::Error>
     where
-        C: EventContext,
+        C: EventContext<Payload = Self::Payload>,
     {
         // Create a new span for the on_message operation as a child of the message's
         // span
@@ -207,7 +208,7 @@ impl FallibleHandler for RubyHandler {
         _demand_type: DemandType,
     ) -> Result<(), Self::Error>
     where
-        C: EventContext,
+        C: EventContext<Payload = Self::Payload>,
     {
         // Only process application timers; internal timers are handled by middleware
         if trigger.timer_type != TimerType::Application {
@@ -340,6 +341,7 @@ pub enum RubyHandlerError {
 pub fn init(ruby: &Ruby) -> Result<(), Error> {
     context::init(ruby)?;
     message::init(ruby)?;
+    state::init(ruby)?;
     trigger::init(ruby)?;
 
     Ok(())