go through upload, create kb, add doc to kb (#11)

* add field progress msg into docinfo; add file processing procedure

* go through upload, create kb, add doc to kb
This commit is contained in:
KevinHuSh
2023-12-19 19:28:01 +08:00
committed by GitHub
parent d94c6df4a4
commit 249b27c1e0
24 changed files with 537 additions and 85 deletions

View File

@@ -2,6 +2,7 @@ use chrono::Local;
use sea_orm::{ActiveModelTrait, ColumnTrait, DbConn, DbErr, DeleteResult, EntityTrait, PaginatorTrait, QueryFilter, QueryOrder};
use sea_orm::ActiveValue::Set;
use crate::entity::kb_info;
use crate::entity::kb2_doc;
use crate::entity::kb_info::Entity;
pub struct Query;
@@ -21,6 +22,13 @@ impl Query {
.all(db)
.await
}
pub async fn find_kb_infos_by_name(db: &DbConn, name: String) -> Result<Vec<kb_info::Model>, DbErr> {
Entity::find()
.filter(kb_info::Column::KbName.eq(name))
.all(db)
.await
}
pub async fn find_kb_infos_in_page(
db: &DbConn,
@@ -48,7 +56,7 @@ impl Mutation {
kb_info::ActiveModel {
kb_id: Default::default(),
uid: Set(form_data.uid.to_owned()),
kn_name: Set(form_data.kn_name.to_owned()),
kb_name: Set(form_data.kb_name.to_owned()),
icon: Set(form_data.icon.to_owned()),
created_at: Set(Local::now().date_naive()),
updated_at: Set(Local::now().date_naive()),
@@ -57,6 +65,24 @@ impl Mutation {
.await
}
pub async fn add_docs(
db: &DbConn,
kb_id: i64,
doc_ids: Vec<i64>
)-> Result<(), DbErr> {
for did in doc_ids{
let _ = kb2_doc::ActiveModel {
id: Default::default(),
kb_id: Set(kb_id),
did: Set(did),
}
.save(db)
.await?;
}
Ok(())
}
pub async fn update_kb_info_by_id(
db: &DbConn,
id: i64,
@@ -71,7 +97,7 @@ impl Mutation {
kb_info::ActiveModel {
kb_id: kb_info.kb_id,
uid: kb_info.uid,
kn_name: Set(form_data.kn_name.to_owned()),
kb_name: Set(form_data.kb_name.to_owned()),
icon: Set(form_data.icon.to_owned()),
created_at: Default::default(),
updated_at: Set(Local::now().date_naive()),