1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use crate::MsgPackError;
use actix_web::http::header::ContentType;
use actix_web::web::Bytes;
use actix_web::{HttpResponse, HttpResponseBuilder};
use mime::APPLICATION_MSGPACK;
use serde::Serialize;

pub trait MsgPackResponseBuilder {
	/// Responder with compact representation
	fn msgpack<T: Serialize>(&mut self, value: T) -> HttpResponse;
	/// Responder with field names
	fn msgpack_named<T: Serialize>(&mut self, value: T) -> HttpResponse;
}

impl MsgPackResponseBuilder for HttpResponseBuilder {
	fn msgpack<T: Serialize>(&mut self, value: T) -> HttpResponse {
		match rmp_serde::to_vec(&value) {
			Ok(body) => {
				self.insert_header(ContentType(APPLICATION_MSGPACK));
				self.body(Bytes::from(body))
			},
			Err(err) => HttpResponse::from_error(MsgPackError::Serialize(err)),
		}
	}

	fn msgpack_named<T: Serialize>(&mut self, value: T) -> HttpResponse {
		match rmp_serde::to_vec_named(&value) {
			Ok(body) => {
				self.insert_header(ContentType(APPLICATION_MSGPACK));
				self.body(Bytes::from(body))
			},
			Err(err) => HttpResponse::from_error(MsgPackError::Serialize(err)),
		}
	}
}