Refactor.
Some checks failed
Gitea Actions Demo / Explore-Gitea-Actions (push) Has been cancelled

This commit is contained in:
2026-03-25 03:41:32 -04:00
parent 02caf62f29
commit 316ae06654

View File

@@ -39,15 +39,15 @@ mod support_tests {
static TIMEOUT: Duration = Duration::from_secs(10);
struct PreBuildClient {
pub struct MTTClient {
queue: Queue,
msg_id: MessageID,
rx: Receiver<Message>,
sender_id: SenderID,
session_id: Uuid,
}
impl PreBuildClient {
fn new(mut queue: Queue) -> Self {
impl MTTClient {
fn new(mut queue: Queue, sess_id: Option<String>, lang: Option<Language>) -> Self {
let sess_name = Session::doc_names()[0].clone();
let (tx, rx) = channel();
let sender_id = queue.add_sender(tx);
@@ -64,42 +64,6 @@ impl PreBuildClient {
Include::Just(Action::Error),
),
];
for path in paths.iter().cloned() {
let reg_msg = Register::new(sender_id.clone(), RegMsg::AddRoute(path));
queue.send(Message::with_id(msg_id.clone(), reg_msg));
let result = rx.recv().unwrap();
}
Self {
queue: queue,
msg_id: msg_id,
rx: rx,
sender_id: sender_id,
}
}
fn send<MA>(&self, action: MA) -> Records
where
MA: Into<MsgAction>,
{
self.queue
.send(Message::with_id(self.msg_id.clone(), action));
let result = self.rx.recv().unwrap();
match result.get_action() {
MsgAction::Records(data) => data.clone(),
_ => unreachable!("session requests should always return data"),
}
}
}
pub struct MTTClient {
queue: Queue,
rx: Receiver<Message>,
sender_id: SenderID,
session_id: Uuid,
}
impl MTTClient {
fn internal_new(prebuild: PreBuildClient, lang: Option<Language>) -> Self {
let mut add = Addition::new(Session::doc_names()[0].clone());
match lang {
Some(language) => {
@@ -108,55 +72,56 @@ impl MTTClient {
}
None => {}
}
let result = prebuild.send(add);
let rec = result.iter().last().unwrap();
let session_id = match rec.get(Session::id_field_names()[0].clone()).unwrap() {
Field::Uuid(data) => data.clone(),
_ => unreachable!("should always be uuid"),
for path in paths.iter().cloned() {
let reg_msg = Register::new(sender_id.clone(), RegMsg::AddRoute(path));
queue.send(Message::with_id(msg_id.clone(), reg_msg));
let result = rx.recv().unwrap();
}
match sess_id {
Some(id) => {
let sess_id = match Uuid::try_from(id.as_str()) {
Ok(data) => {
let mut qry = Query::new(Session::doc_names()[0].clone());
let mut calc = Calculation::new(Operand::Equal);
calc.add_value(CalcValue::Existing(FieldType::Uuid))
.unwrap();
calc.add_value(data.clone()).unwrap();
qry.add(Session::id_field_names()[0].clone(), calc);
queue.send(Message::with_id(msg_id.clone(), qry));
}
Err(_) => queue.send(Message::with_id(msg_id.clone(), add.clone())),
};
}
None => queue.send(Message::with_id(msg_id.clone(), add.clone())),
};
let result = rx.recv().unwrap();
let session_id = match result.get_action() {
MsgAction::Records(result) => {
let mut holder = result.clone();
if holder.len() == 0 {
queue.send(Message::with_id(msg_id.clone(), add));
let new_sess = rx.recv().unwrap();
holder = match new_sess.get_action() {
MsgAction::Records(new_holder) => new_holder.clone(),
_ => unreachable!("should only receive session records"),
}
}
let rec = holder.iter().last().unwrap();
match rec.get(Session::id_field_names()[0].clone()).unwrap() {
Field::Uuid(data) => data.clone(),
_ => unreachable!("should always be uuid"),
}
}
_ => unreachable!("should only receive session records"),
};
Self {
queue: prebuild.queue,
rx: prebuild.rx,
sender_id: prebuild.sender_id,
queue: queue,
rx: rx,
sender_id: sender_id,
session_id: session_id,
}
}
fn new(mut queue: Queue, lang: Option<Language>) -> Self {
let prebuild = PreBuildClient::new(queue);
Self::internal_new(prebuild, lang)
}
fn with_session(queue: Queue, id: String, lang: Option<Language>) -> Self {
let prebuild = PreBuildClient::new(queue.clone());
let sess_id = match Uuid::try_from(id.as_str()) {
Ok(data) => data,
Err(_) => return Self::internal_new(prebuild, lang),
};
let mut qry = Query::new(Session::doc_names()[0].clone());
let mut calc = Calculation::new(Operand::Equal);
calc.add_value(CalcValue::Existing(FieldType::Uuid))
.unwrap();
calc.add_value(sess_id).unwrap();
qry.add(Session::id_field_names()[0].clone(), calc);
let result = prebuild.send(qry);
if result.len() == 0 {
Self::internal_new(prebuild, lang)
} else {
let rec = result.iter().last().unwrap();
let session_id = match rec.get(Session::id_field_names()[0].clone()).unwrap() {
Field::Uuid(data) => data.clone(),
_ => unreachable!("should always be uuid"),
};
Self {
queue: queue,
rx: prebuild.rx,
sender_id: prebuild.sender_id.clone(),
session_id: session_id,
}
}
}
pub fn session_id(&self) -> String {
self.session_id.to_string()
}
@@ -188,15 +153,15 @@ impl MoreThanText {
}
pub fn client(&self) -> MTTClient {
MTTClient::new(self.queue.clone(), None)
MTTClient::new(self.queue.clone(), None, None)
}
pub fn client_with_language(&self, lang: Language) -> MTTClient {
MTTClient::new(self.queue.clone(), Some(lang))
MTTClient::new(self.queue.clone(), None, Some(lang))
}
pub fn client_with_session(&self, id: String, lang: Option<Language>) -> MTTClient {
MTTClient::with_session(self.queue.clone(), id, lang)
MTTClient::new(self.queue.clone(), Some(id), lang)
}
fn new_session(lang: Option<Language>) -> ClientAction {