Struct ProduceService 

pub struct ProduceService;

A Service using Storage as Context taking ProduceRequest returning ProduceResponse.

use bytes::Bytes;
use rama::{Context, Layer as _, Service as _, layer::MapStateLayer};
use tansu_sans_io::{
    CreateTopicsRequest, ErrorCode, ProduceRequest,
    create_topics_request::CreatableTopic,
    produce_request::{PartitionProduceData, TopicProduceData},
    record::{Record, deflated::Frame, inflated},
};
use tansu_storage::{CreateTopicsService, Error, ProduceService, StorageContainer};
use url::Url;

const CLUSTER_ID: &str = "tansu";
const NODE_ID: i32 = 111;
const HOST: &str = "localhost";
const PORT: i32 = 9092;

let storage = StorageContainer::builder()
    .cluster_id(CLUSTER_ID)
    .node_id(NODE_ID)
    .advertised_listener(Url::parse(&format!("tcp://{HOST}:{PORT}"))?)
    .storage(Url::parse("memory://tansu/")?)
    .build()
    .await?;

let create_topic = {
    let storage = storage.clone();
    MapStateLayer::new(|_| storage).into_layer(CreateTopicsService)
};

let name = "abcba";

let response = create_topic
    .serve(
        Context::default(),
        CreateTopicsRequest::default()
            .topics(Some(vec![
                CreatableTopic::default()
                    .name(name.into())
                    .num_partitions(5)
                    .replication_factor(3)
                    .assignments(Some([].into()))
                    .configs(Some([].into())),
            ]))
            .validate_only(Some(false)),
    )
    .await?;

let topics = response.topics.unwrap_or_default();
assert_eq!(1, topics.len());
assert_eq!(ErrorCode::None, ErrorCode::try_from(topics[0].error_code)?);

let produce = {
    let storage = storage.clone();
    MapStateLayer::new(|_| storage).into_layer(ProduceService)
};

let partition = 0;

let response = produce
    .serve(
        Context::default(),
        ProduceRequest::default().topic_data(Some(
            [TopicProduceData::default()
                .name(name.into())
                .partition_data(Some(
                    [PartitionProduceData::default()
                        .index(partition)
                        .records(Some(Frame {
                            batches: vec![
                                inflated::Batch::builder()
                                    .record(
                                        Record::builder().value(
                                            Bytes::from_static(
                                                b"Lorem ipsum dolor sit amet",
                                            )
                                            .into(),
                                        ),
                                    )
                                    .build()
                                    .and_then(TryInto::try_into)?,
                            ],
                        }))]
                    .into(),
                ))]
            .into(),
        )),
    )
    .await?;

let topics = response.responses.as_deref().unwrap_or_default();
assert_eq!(1, topics.len());
let partitions = topics[0].partition_responses.as_deref().unwrap_or_default();
assert_eq!(1, partitions.len());
assert_eq!(
    ErrorCode::None,
    ErrorCode::try_from(partitions[0].error_code)?
);

Trait Implementations

impl ApiKey for ProduceService

const KEY: i16 = ProduceRequest::KEY

impl Clone for ProduceService

fn clone(&self) -> ProduceService

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ProduceService

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ProduceService

fn default() -> ProduceService

Returns the “default value” for a type.

impl Hash for ProduceService

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl Ord for ProduceService

fn cmp(&self, other: &ProduceService) -> Ordering

This method returns an Ordering between self and other.

fn max(self, other: Self) -> Self

where Self: Sized,

Compares and returns the maximum of two values.

fn min(self, other: Self) -> Self

where Self: Sized,

Compares and returns the minimum of two values.

fn clamp(self, min: Self, max: Self) -> Self

where Self: Sized,

Restrict a value to a certain interval.

impl PartialEq for ProduceService

fn eq(&self, other: &ProduceService) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl PartialOrd for ProduceService

fn partial_cmp(&self, other: &ProduceService) -> Option<Ordering>

This method returns an ordering between self and other values if one exists.

fn lt(&self, other: &Rhs) -> bool

Tests less than (for self and other) and is used by the < operator.

fn le(&self, other: &Rhs) -> bool

Tests less than or equal to (for self and other) and is used by the <= operator.

fn gt(&self, other: &Rhs) -> bool

Tests greater than (for self and other) and is used by the > operator.

fn ge(&self, other: &Rhs) -> bool

Tests greater than or equal to (for self and other) and is used by the >= operator.

impl<G> Service<G, ProduceRequest> for ProduceService

where G: Storage,

type Response = ProduceResponse

The type of response returned by the service.

type Error = Error

The type of error returned by the service.

async fn serve( &self, ctx: Context<G>, req: ProduceRequest, ) -> Result<Self::Response, Self::Error>

Serve a response or error for the given request, using the given context.

fn boxed(self) -> BoxService<S, Request, Self::Response, Self::Error>

Box this service to allow for dynamic dispatch.

impl Copy for ProduceService

impl Eq for ProduceService

impl StructuralPartialEq for ProduceService