Lines
92.46 %
Functions
100 %
Branches
use std::error::Error as StdError;
use actix_web::{
http::header::{self, HeaderValue},
web::{self, Bytes},
HttpMessage, HttpRequest, HttpResponse, Responder, ResponseError,
};
use chrono::{TimeZone, Utc};
use csv::WriterBuilder;
use log::error;
use serde_json;
use sylvia_iot_corelib::{err::ErrResp, role::Role, strings};
use super::{
super::{
super::{middleware::FullTokenInfo, ErrReq, State},
get_unit_inner,
},
request, response,
use crate::models::application_uldata::{
ApplicationUlData, ListOptions, ListQueryCond, SortCond, SortKey,
const LIST_LIMIT_DEFAULT: u64 = 100;
const LIST_CURSOR_MAX: u64 = 100;
const CSV_FIELDS: &'static str =
"dataId,proc,pub,unitCode,networkCode,networkAddr,unitId,deviceId,time,profile,data,extension\n";
/// `GET /{base}/api/v1/application-uldata/count`
pub async fn get_count(
req: HttpRequest,
query: web::Query<request::GetCountQuery>,
state: web::Data<State>,
) -> impl Responder {
const FN_NAME: &'static str = "get_count";
let mut query: request::GetCountQuery = (*query).clone();
if let Some(network) = query.network {
query.network = Some(network.to_lowercase());
}
if let Some(addr) = query.addr {
query.addr = Some(addr.to_lowercase());
let unit_cond = match get_unit_cond(FN_NAME, &req, query.unit.as_ref(), &state).await {
Err(e) => return e,
Ok(cond) => cond,
let cond = match get_list_cond(&query, &unit_cond).await {
Err(e) => return e.error_response(),
match state.model.application_uldata().count(&cond).await {
Err(e) => {
error!("[{}] count error: {}", FN_NAME, e);
ErrResp::ErrDb(Some(e.to_string())).error_response()
Ok(count) => HttpResponse::Ok().json(response::GetCount {
data: response::GetCountData { count },
}),
/// `GET /{base}/api/v1/application-uldata/list`
pub async fn get_list(
query: web::Query<request::GetListQuery>,
const FN_NAME: &'static str = "get_list";
let cond_query = request::GetCountQuery {
unit: query.unit.clone(),
device: query.device.clone(),
network: match query.network.as_ref() {
None => None,
Some(network) => Some(network.to_lowercase()),
addr: match query.addr.as_ref() {
Some(addr) => Some(addr.to_lowercase()),
profile: query.profile.clone(),
tfield: query.tfield.clone(),
tstart: query.tstart,
tend: query.tend,
Err(e) => return Ok(e),
let cond = match get_list_cond(&cond_query, &unit_cond).await {
Err(e) => return Err(e),
let sort_cond = match get_sort_cond(&query.sort) {
let opts = ListOptions {
cond: &cond,
offset: query.offset,
limit: match query.limit {
None => Some(LIST_LIMIT_DEFAULT),
Some(limit) => Some(limit),
sort: Some(sort_cond.as_slice()),
cursor_max: Some(LIST_CURSOR_MAX),
let (list, cursor) = match state.model.application_uldata().list(&opts, None).await {
error!("[{}] list error: {}", FN_NAME, e);
return Err(ErrResp::ErrDb(Some(e.to_string())));
Ok((list, cursor)) => match cursor {
None => match query.format.as_ref() {
Some(request::ListFormat::Array) => {
return Ok(HttpResponse::Ok().json(list_transform(&list)))
Some(request::ListFormat::Csv) => {
let bytes = match list_transform_bytes(&list, true, true, query.format.as_ref())
{
return Err(ErrResp::ErrUnknown(Some(format!(
"transform CSV error: {}",
e
))))
Ok(bytes) => bytes,
return Ok(HttpResponse::Ok()
.insert_header((header::CONTENT_TYPE, "text/csv"))
.insert_header((
header::CONTENT_DISPOSITION,
"attachment;filename=application-uldata.csv",
))
.body(bytes));
_ => {
return Ok(HttpResponse::Ok().json(response::GetList {
data: list_transform(&list),
}))
Some(_) => (list, cursor),
// TODO: detect client disconnect
let query_format = query.format.clone();
let stream = async_stream::stream! {
let query = query.0.clone();
network: query.network.clone(),
addr: query.addr.clone(),
Err(_) => return,
let mut list = list;
let mut cursor = cursor;
let mut is_first = true;
loop {
yield list_transform_bytes(&list, is_first, cursor.is_none(), query.format.as_ref());
is_first = false;
if cursor.is_none() {
break;
let (_list, _cursor) = match state.model.application_uldata().list(&opts, cursor).await {
Err(_) => break,
Ok((list, cursor)) => (list, cursor),
list = _list;
cursor = _cursor;
match query_format {
Some(request::ListFormat::Csv) => Ok(HttpResponse::Ok()
.streaming(stream)),
_ => Ok(HttpResponse::Ok().streaming(stream)),
async fn get_list_cond<'a>(
query: &'a request::GetCountQuery,
unit_cond: &'a Option<String>,
) -> Result<ListQueryCond<'a>, ErrResp> {
let mut cond = ListQueryCond {
unit_id: match unit_cond.as_ref() {
Some(unit_id) => Some(unit_id.as_str()),
..Default::default()
if let Some(device_id) = query.device.as_ref() {
if device_id.len() > 0 {
cond.device_id = Some(device_id.as_str());
if let Some(network_code) = query.network.as_ref() {
if network_code.len() > 0 {
cond.network_code = Some(network_code.as_str());
if let Some(network_addr) = query.addr.as_ref() {
if network_addr.len() > 0 {
cond.network_addr = Some(network_addr.as_str());
if let Some(profile) = query.profile.as_ref() {
if profile.len() > 0 {
cond.profile = Some(profile.as_str());
if let Some(start) = query.tstart.as_ref() {
match query.tfield.as_ref() {
None => return Err(ErrResp::ErrParam(Some("missing `tfield`".to_string()))),
Some(tfield) => match tfield.as_str() {
"proc" => cond.proc_gte = Some(Utc.timestamp_nanos(*start * 1000000)),
"pub" => cond.pub_gte = Some(Utc.timestamp_nanos(*start * 1000000)),
"time" => cond.time_gte = Some(Utc.timestamp_nanos(*start * 1000000)),
_ => return Err(ErrResp::ErrParam(Some("invalid `tfield`".to_string()))),
if let Some(end) = query.tend.as_ref() {
"proc" => cond.proc_lte = Some(Utc.timestamp_nanos(*end * 1000000)),
"pub" => cond.pub_lte = Some(Utc.timestamp_nanos(*end * 1000000)),
"time" => cond.time_lte = Some(Utc.timestamp_nanos(*end * 1000000)),
Ok(cond)
async fn get_unit_cond(
fn_name: &str,
req: &HttpRequest,
query_unit: Option<&String>,
state: &web::Data<State>,
) -> Result<Option<String>, HttpResponse> {
let token_info = match req.extensions_mut().get::<FullTokenInfo>() {
None => {
error!("[{}] token not found", fn_name);
return Err(
ErrResp::ErrUnknown(Some("token info not found".to_string())).error_response(),
);
Some(token_info) => token_info.clone(),
let broker_base = state.broker_base.as_str();
let client = state.client.clone();
match query_unit {
if !Role::is_role(&token_info.info.roles, Role::ADMIN)
&& !Role::is_role(&token_info.info.roles, Role::MANAGER)
return Err(ErrResp::ErrParam(Some("missing `unit`".to_string())).error_response());
Ok(None)
Some(unit_id) => match unit_id.len() {
0 => Ok(None),
let token = match HeaderValue::from_str(token_info.token.as_str()) {
error!("[{}] get token error: {}", fn_name, e);
return Err(ErrResp::ErrUnknown(Some(format!("get token error: {}", e)))
.error_response());
Ok(value) => value,
match get_unit_inner(fn_name, &client, broker_base, unit_id, &token).await {
error!("[{}] get unit error", fn_name);
return Err(e);
Ok(unit) => match unit {
return Err(ErrResp::Custom(
ErrReq::UNIT_NOT_EXIST.0,
ErrReq::UNIT_NOT_EXIST.1,
None,
)
.error_response())
Some(_) => Ok(Some(unit_id.clone())),
fn get_sort_cond(sort_args: &Option<String>) -> Result<Vec<SortCond>, ErrResp> {
match sort_args.as_ref() {
None => Ok(vec![SortCond {
key: SortKey::Proc,
asc: false,
}]),
Some(args) => {
let mut args = args.split(",");
let mut sort_cond = vec![];
while let Some(arg) = args.next() {
let mut cond = arg.split(":");
let key = match cond.next() {
None => return Err(ErrResp::ErrParam(Some("wrong sort argument".to_string()))),
Some(field) => match field {
"proc" => SortKey::Proc,
"pub" => SortKey::Pub,
"time" => SortKey::Time,
"network" => SortKey::NetworkCode,
"addr" => SortKey::NetworkAddr,
return Err(ErrResp::ErrParam(Some(format!(
"invalid sort key {}",
field
let asc = match cond.next() {
Some(asc) => match asc {
"asc" => true,
"desc" => false,
"invalid sort asc {}",
asc
if cond.next().is_some() {
return Err(ErrResp::ErrParam(Some(
"invalid sort condition".to_string(),
)));
sort_cond.push(SortCond { key, asc });
Ok(sort_cond)
fn list_transform(list: &Vec<ApplicationUlData>) -> Vec<response::GetListData> {
let mut ret = vec![];
for item in list.iter() {
ret.push(data_transform(&item));
ret
fn list_transform_bytes(
list: &Vec<ApplicationUlData>,
with_start: bool,
with_end: bool,
format: Option<&request::ListFormat>,
) -> Result<Bytes, Box<dyn StdError>> {
let mut build_str = match with_start {
false => "".to_string(),
true => match format {
Some(request::ListFormat::Array) => "[".to_string(),
let bom = String::from_utf8(vec![0xEF, 0xBB, 0xBF])?;
format!("{}{}", bom, CSV_FIELDS)
_ => "{\"data\":[".to_string(),
let mut is_first = with_start;
for item in list {
match format {
let mut writer = WriterBuilder::new().has_headers(false).from_writer(vec![]);
writer.serialize(data_transform_csv(item))?;
build_str += String::from_utf8(writer.into_inner()?)?.as_str();
if is_first {
} else {
build_str.push(',');
let json_str = match serde_json::to_string(&data_transform(item)) {
Err(e) => return Err(Box::new(e)),
Ok(str) => str,
build_str += json_str.as_str();
if with_end {
build_str += match format {
Some(request::ListFormat::Array) => "]",
Some(request::ListFormat::Csv) => "",
_ => "]}",
Ok(Bytes::copy_from_slice(build_str.as_str().as_bytes()))
fn data_transform(data: &ApplicationUlData) -> response::GetListData {
response::GetListData {
data_id: data.data_id.clone(),
proc: strings::time_str(&data.proc),
publish: strings::time_str(&data.publish),
unit_code: data.unit_code.clone(),
network_code: data.network_code.clone(),
network_addr: data.network_addr.clone(),
unit_id: data.unit_id.clone(),
device_id: data.device_id.clone(),
time: strings::time_str(&data.time),
profile: data.profile.clone(),
data: data.data.clone(),
extension: data.extension.clone(),
fn data_transform_csv(data: &ApplicationUlData) -> response::GetListCsvData {
response::GetListCsvData {
unit_code: match data.unit_code.as_ref() {
None => "".to_string(),
Some(unit_code) => unit_code.clone(),
extension: match data.extension.as_ref() {
Some(extension) => match serde_json::to_string(extension) {
Err(_) => "".to_string(),
Ok(extension) => extension,