pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2026-01-04 03:25:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

add seed data generation (#13)

Browse Source 
* merge upstream

* add seed data generation
This commit is contained in:
KevinHuSh
2023-12-22 20:02:16 +08:00
committed by GitHub
parent 1eb186a25f
commit 8a65ad888e
8 changed files with 117 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/service/tag_info.rs
View File

@ -1,6 +1,6 @@
use chrono::{FixedOffset, Utc};
use sea_orm::{ActiveModelTrait, DbConn, DbErr, DeleteResult, EntityTrait, PaginatorTrait, QueryOrder, ColumnTrait, QueryFilter};
use sea_orm::ActiveValue::Set;
use sea_orm::ActiveValue::{Set, NotSet};
use crate::entity::tag_info;
use crate::entity::tag_info::Entity;
@ -51,7 +51,10 @@ impl Mutation {
regx: Set(form_data.regx.to_owned()),
color: Set(form_data.color.to_owned()),
icon: Set(form_data.icon.to_owned()),
folder_id: Set(form_data.folder_id.to_owned()),
folder_id: match form_data.folder_id {
0 => NotSet,
_ => Set(form_data.folder_id.to_owned())
},
created_at: Set(now()),
updated_at: Set(now()),
}

7
src/service/user_info.rs
View File

@ -23,8 +23,11 @@ impl Query {
.await
}
pub async fn find_user_infos(db: &DbConn) -> Result<Vec<user_info::Model>, DbErr> {
Entity::find().all(db).await
pub async fn find_user_infos(db: &DbConn, email:&String) -> Result<Option<user_info::Model>, DbErr> {
Entity::find()
.filter(user_info::Column::Email.eq(email))
.one(db)
.await
}
pub async fn find_user_infos_in_page(