member_referrer is added

This commit is contained in:
병준 박 2022-08-19 07:06:00 +00:00
parent dde35b227c
commit 63048f6800
25 changed files with 1405 additions and 72 deletions

View File

@ -31,7 +31,7 @@ tokio = { version = "1", features = ["macros", "rt-multi-thread"] }
tokio-cron-scheduler = { version = "0" } tokio-cron-scheduler = { version = "0" }
uuid = { version = "0", features = ["serde", "v4", "v5"] } uuid = { version = "0", features = ["serde", "v4", "v5"] }
beteran-protobuf-rust = { git = "https://gitlab.loafle.net/bet/beteran-protobuf-rust.git", tag = "v0.1.64-snapshot" } beteran-protobuf-rust = { git = "https://gitlab.loafle.net/bet/beteran-protobuf-rust.git", tag = "v0.1.66-snapshot" }
beteran-common-rust = { git = "https://gitlab.loafle.net/bet/beteran-common-rust.git", tag = "v0.1.48-snapshot" } beteran-common-rust = { git = "https://gitlab.loafle.net/bet/beteran-common-rust.git", tag = "v0.1.50-snapshot" }
[build-dependencies] [build-dependencies]

View File

@ -12,8 +12,8 @@ CREATE TABLE IF NOT EXISTS members (
mobile_phone_number TEXT, mobile_phone_number TEXT,
state MEMBER_STATE DEFAULT 'normal', state MEMBER_STATE DEFAULT 'normal',
state_changed_at BIGINT, state_changed_at BIGINT,
referrer_member_id UUID, parent_member_id UUID,
referred_count BIGINT NOT NULL DEFAULT 0, child_member_count BIGINT NOT NULL DEFAULT 0,
last_signined_ip TEXT, last_signined_ip TEXT,
last_signined_at BIGINT, last_signined_at BIGINT,
created_at BIGINT NOT NULL DEFAULT (extract(epoch from now()) * 1000), created_at BIGINT NOT NULL DEFAULT (extract(epoch from now()) * 1000),
@ -30,8 +30,8 @@ CREATE TABLE IF NOT EXISTS members (
CONSTRAINT fk_members_member_level_id CONSTRAINT fk_members_member_level_id
FOREIGN KEY(member_level_id) FOREIGN KEY(member_level_id)
REFERENCES member_levels(id), REFERENCES member_levels(id),
CONSTRAINT fk_members_referrer_member_id CONSTRAINT fk_members_parent_member_id
FOREIGN KEY(referrer_member_id) FOREIGN KEY(parent_member_id)
REFERENCES members(id) REFERENCES members(id)
); );

View File

@ -0,0 +1,3 @@
DROP INDEX idx_member_referrers_member_id;
DROP TRIGGER tg_member_referrers_updated_at;
DROP TABLE member_referrers;

View File

@ -0,0 +1,22 @@
CREATE TABLE IF NOT EXISTS member_referrers (
id UUID DEFAULT uuid_generate_v4(),
member_id UUID NOT NULL,
can_use BOOLEAN NOT NULL DEFAULT TRUE,
memo TEXT,
created_at BIGINT NOT NULL DEFAULT (extract(epoch from now()) * 1000),
updated_at BIGINT NOT NULL DEFAULT (extract(epoch from now()) * 1000),
PRIMARY KEY (id),
CONSTRAINT fk_member_referrers_member_id
FOREIGN KEY(member_id)
REFERENCES members(id)
);
CREATE INDEX idx_member_referrers_member_id ON member_referrers (member_id);
-- trigger (updated_at)
CREATE TRIGGER tg_member_referrers_updated_at
BEFORE UPDATE
ON member_referrers
FOR EACH ROW
EXECUTE PROCEDURE update_updated_at_column();

View File

@ -17,8 +17,8 @@ SELECT
m.mobile_phone_number as m_mobile_phone_number, m.mobile_phone_number as m_mobile_phone_number,
m.state as m_state, m.state as m_state,
m.state_changed_at as m_state_changed_at, m.state_changed_at as m_state_changed_at,
m.referrer_member_id as m_referrer_member_id, m.parent_member_id as m_parent_member_id,
m.referred_count as m_referred_count, m.child_member_count as m_child_member_count,
m.last_signined_ip as m_last_signined_ip, m.last_signined_ip as m_last_signined_ip,
m.last_signined_at as m_last_signined_at, m.last_signined_at as m_last_signined_at,
m.created_at as m_created_at, m.created_at as m_created_at,
@ -60,8 +60,8 @@ SELECT
_m.mobile_phone_number as _m_mobile_phone_number, _m.mobile_phone_number as _m_mobile_phone_number,
_m.state as _m_state, _m.state as _m_state,
_m.state_changed_at as _m_state_changed_at, _m.state_changed_at as _m_state_changed_at,
_m.referrer_member_id as _m_referrer_member_id, _m.parent_member_id as _m_parent_member_id,
_m.referred_count as _m_referred_count, _m.child_member_count as _m_child_member_count,
_m.last_signined_ip as _m_last_signined_ip, _m.last_signined_ip as _m_last_signined_ip,
_m.last_signined_at as _m_last_signined_at, _m.last_signined_at as _m_last_signined_at,
_m.created_at as _m_created_at, _m.created_at as _m_created_at,
@ -95,7 +95,7 @@ FROM members as m
INNER JOIN member_levels ml INNER JOIN member_levels ml
ON ml.id = m.member_level_id ON ml.id = m.member_level_id
LEFT OUTER JOIN members _m LEFT OUTER JOIN members _m
ON _m.id = m.referrer_member_id ON _m.id = m.parent_member_id
LEFT OUTER JOIN member_bank_accounts mba LEFT OUTER JOIN member_bank_accounts mba
ON mba.member_id = m.id ON mba.member_id = m.id
LEFT OUTER JOIN banks _b LEFT OUTER JOIN banks _b

View File

@ -33,9 +33,9 @@ pub struct MemberModel {
/// ///
pub state_changed_at: Option<i64>, pub state_changed_at: Option<i64>,
/// ///
pub referrer_member: Option<_Member>, pub parent_member: Option<_Member>,
/// ///
pub referred_count: i64, pub child_member_count: i64,
/// ///
pub last_signined_ip: Option<String>, pub last_signined_ip: Option<String>,
/// ///
@ -50,9 +50,9 @@ pub struct MemberModel {
impl QueryableByName<diesel::pg::Pg> for MemberModel { impl QueryableByName<diesel::pg::Pg> for MemberModel {
fn build<R: diesel::row::NamedRow<diesel::pg::Pg>>(row: &R) -> diesel::deserialize::Result<Self> { fn build<R: diesel::row::NamedRow<diesel::pg::Pg>>(row: &R) -> diesel::deserialize::Result<Self> {
let referrer_member = match row let parent_member = match row
.get::<diesel::sql_types::Nullable<diesel::sql_types::Uuid>, Option<uuid::Uuid>>( .get::<diesel::sql_types::Nullable<diesel::sql_types::Uuid>, Option<uuid::Uuid>>(
"m_referrer_member_id", "m_parent_member_id",
)? { )? {
Some(_) => Some(_Member { Some(_) => Some(_Member {
id: row.get("_m_id")?, id: row.get("_m_id")?,
@ -65,8 +65,8 @@ impl QueryableByName<diesel::pg::Pg> for MemberModel {
mobile_phone_number: row.get("_m_mobile_phone_number")?, mobile_phone_number: row.get("_m_mobile_phone_number")?,
state: row.get("_m_state")?, state: row.get("_m_state")?,
state_changed_at: row.get("_m_state_changed_at")?, state_changed_at: row.get("_m_state_changed_at")?,
referrer_member_id: row.get("_m_referrer_member_id")?, parent_member_id: row.get("_m_parent_member_id")?,
referred_count: row.get("_m_referred_count")?, child_member_count: row.get("_m_child_member_count")?,
last_signined_ip: row.get("_m_last_signined_ip")?, last_signined_ip: row.get("_m_last_signined_ip")?,
last_signined_at: row.get("_m_last_signined_at")?, last_signined_at: row.get("_m_last_signined_at")?,
created_at: row.get("_m_created_at")?, created_at: row.get("_m_created_at")?,
@ -144,8 +144,8 @@ impl QueryableByName<diesel::pg::Pg> for MemberModel {
state: row.get("m_state")?, state: row.get("m_state")?,
state_changed_at: row.get("m_state_changed_at")?, state_changed_at: row.get("m_state_changed_at")?,
bank_account: member_bank_account, bank_account: member_bank_account,
referrer_member, parent_member,
referred_count: row.get("m_referred_count")?, child_member_count: row.get("m_child_member_count")?,
last_signined_ip: row.get("m_last_signined_ip")?, last_signined_ip: row.get("m_last_signined_ip")?,
last_signined_at: row.get("m_last_signined_at")?, last_signined_at: row.get("m_last_signined_at")?,
created_at: row.get("m_created_at")?, created_at: row.get("m_created_at")?,

View File

@ -0,0 +1,143 @@
//!
//!
use super::models;
use crate::repositories;
use diesel::{result::Error, sql_query, RunQueryDsl};
static MEMBER_QUERY: &str = "
SELECT
mr.id as mr_id,
mr.member_id as mr_member_id,
mr.can_use as mr_can_use,
mr.memo as mr_memo,
mr.created_at as mr_created_at,
mr.updated_at as mr_updated_at,
_m.id as _m_id,
_m.site_id as _m_site_id,
_m.member_class_id as _m_member_class_id,
_m.member_level_id as _m_member_level_id,
_m.username as _m_username,
_m.password as _m_password,
_m.nickname as _m_nickname,
_m.mobile_phone_number as _m_mobile_phone_number,
_m.state as _m_state,
_m.state_changed_at as _m_state_changed_at,
_m.parent_member_id as _m_parent_member_id,
_m.child_member_count as _m_child_member_count,
_m.last_signined_ip as _m_last_signined_ip,
_m.last_signined_at as _m_last_signined_at,
_m.created_at as _m_created_at,
_m.updated_at as _m_updated_at,
_m.deleted_at as _m_deleted_at
FROM member_referrers as m
INNER JOIN members _m
ON _m.id = mr.member_id
";
pub struct Composition {}
impl std::fmt::Debug for Composition {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
f.debug_struct("Composition of members").finish()
}
}
impl Default for Composition {
fn default() -> Self {
Self::new()
}
}
impl Composition {
///
pub fn new() -> Composition {
Composition {}
}
///
pub fn select(
&self,
conn: &diesel::PgConnection,
id: uuid::Uuid,
) -> Result<Option<models::MemberReferrerModel>, Error> {
let query = format!(
"
{}
WHERE
m.id = $1
",
MEMBER_QUERY
);
match sql_query(query)
.bind::<diesel::sql_types::Uuid, _>(id)
.get_result::<models::MemberReferrerModel>(conn)
{
Ok(m) => Ok(Some(m)),
Err(e) => match e {
diesel::result::Error::NotFound => Ok(None),
_ => Err(e),
},
}
}
///
pub fn select_all(
&self,
conn: &diesel::PgConnection,
find_all: &repositories::member_referrer::models::FindAll,
) -> Result<Vec<models::MemberReferrerModel>, Error> {
use std::fmt::Write;
let mut query = String::new();
write!(&mut query, "{}", MEMBER_QUERY)
.map_err(|e| diesel::result::Error::QueryBuilderError(e.to_string().into()))?;
let mut query_where = String::new();
if let Some(s) = &find_all.search {
if let Some(sp) = s.member_id {
if !query_where.is_empty() {
write!(&mut query_where, " AND ")
.map_err(|e| diesel::result::Error::QueryBuilderError(e.to_string().into()))?;
}
write!(&mut query_where, "mr.member_id = '{}'", sp)
.map_err(|e| diesel::result::Error::QueryBuilderError(e.to_string().into()))?;
}
if let Some(sp) = s.can_use {
if !query_where.is_empty() {
write!(&mut query_where, " AND ")
.map_err(|e| diesel::result::Error::QueryBuilderError(e.to_string().into()))?;
}
write!(&mut query_where, "mr.can_use = '{}'", sp)
.map_err(|e| diesel::result::Error::QueryBuilderError(e.to_string().into()))?;
}
if let Some(sp) = &s.memo_like {
if !query_where.is_empty() {
write!(&mut query_where, " AND ")
.map_err(|e| diesel::result::Error::QueryBuilderError(e.to_string().into()))?;
}
write!(&mut query_where, "mr.memo like '{}'", sp)
.map_err(|e| diesel::result::Error::QueryBuilderError(e.to_string().into()))?;
}
}
if !query_where.is_empty() {
write!(&mut query, " WHERE {}", query_where)
.map_err(|e| diesel::result::Error::QueryBuilderError(e.to_string().into()))?;
}
println!("query: {}", query);
match sql_query(query).get_results::<models::MemberReferrerModel>(conn) {
Ok(m) => Ok(m),
Err(e) => match e {
diesel::result::Error::NotFound => Ok(vec![]),
_ => Err(e),
},
}
}
}

View File

@ -0,0 +1,7 @@
//!
//!
///
pub mod composition;
///
pub mod models;

View File

@ -0,0 +1,54 @@
//!
//!
use crate::repositories::member::models::Member as _Member;
use diesel::deserialize::QueryableByName;
///
#[derive(Eq, Hash, PartialEq, Debug, Clone)]
pub struct MemberReferrerModel {
///
pub id: uuid::Uuid,
///
pub member: _Member,
///
pub can_use: bool,
///
pub memo: Option<String>,
///
pub created_at: i64,
///
pub updated_at: i64,
}
impl QueryableByName<diesel::pg::Pg> for MemberReferrerModel {
fn build<R: diesel::row::NamedRow<diesel::pg::Pg>>(row: &R) -> diesel::deserialize::Result<Self> {
let member = _Member {
id: row.get("_m_id")?,
site_id: row.get("_m_site_id")?,
member_class_id: row.get("_m_member_class_id")?,
member_level_id: row.get("_m_member_level_id")?,
username: row.get("_m_username")?,
password: row.get("_m_password")?,
nickname: row.get("_m_nickname")?,
mobile_phone_number: row.get("_m_mobile_phone_number")?,
state: row.get("_m_state")?,
state_changed_at: row.get("_m_state_changed_at")?,
parent_member_id: row.get("_m_parent_member_id")?,
child_member_count: row.get("_m_child_member_count")?,
last_signined_ip: row.get("_m_last_signined_ip")?,
last_signined_at: row.get("_m_last_signined_at")?,
created_at: row.get("_m_created_at")?,
updated_at: row.get("_m_updated_at")?,
deleted_at: row.get("_m_deleted_at")?,
};
Ok(MemberReferrerModel {
id: row.get("mr_id")?,
member,
can_use: row.get("mr_can_use")?,
memo: row.get("mr_memo")?,
created_at: row.get("mr_created_at")?,
updated_at: row.get("mr_updated_at")?,
})
}
}

View File

@ -1,3 +1,4 @@
pub mod identity; pub mod identity;
pub mod member; pub mod member;
pub mod member_referrer;
pub mod site; pub mod site;

View File

@ -26,9 +26,9 @@ pub struct Member {
/// ///
pub state_changed_at: Option<i64>, pub state_changed_at: Option<i64>,
/// ///
pub referrer_member_id: Option<uuid::Uuid>, pub parent_member_id: Option<uuid::Uuid>,
/// ///
pub referred_count: i64, pub child_member_count: i64,
/// ///
pub last_signined_ip: Option<String>, pub last_signined_ip: Option<String>,
/// ///
@ -52,7 +52,7 @@ pub struct NewMember {
/// ///
pub member_level_id: uuid::Uuid, pub member_level_id: uuid::Uuid,
/// ///
pub referrer_member_id: Option<uuid::Uuid>, pub parent_member_id: Option<uuid::Uuid>,
/// ///
pub username: String, pub username: String,
/// ///
@ -113,7 +113,7 @@ pub struct FindAllSearch {
/// ///
pub member_level_id: Option<uuid::Uuid>, pub member_level_id: Option<uuid::Uuid>,
/// ///
pub referrer_member_id: Option<uuid::Uuid>, pub parent_member_id: Option<uuid::Uuid>,
/// ///
pub username_like: Option<String>, pub username_like: Option<String>,
/// ///

View File

@ -291,7 +291,7 @@ impl Repository {
} }
/// ///
pub fn update_increase_referred_count( pub fn increase_child_member_count(
&self, &self,
conn: &diesel::PgConnection, conn: &diesel::PgConnection,
id: uuid::Uuid, id: uuid::Uuid,
@ -299,7 +299,21 @@ impl Repository {
use members::dsl; use members::dsl;
diesel::update(dsl::members.filter(dsl::id.eq(id))) diesel::update(dsl::members.filter(dsl::id.eq(id)))
.set(dsl::referred_count.eq(dsl::referred_count + 1)) .set(dsl::child_member_count.eq(dsl::child_member_count + 1))
.execute(conn)
.map(|c| c as u64)
}
///
pub fn decrease_child_member_count(
&self,
conn: &diesel::PgConnection,
id: uuid::Uuid,
) -> Result<u64, Error> {
use members::dsl;
diesel::update(dsl::members.filter(dsl::id.eq(id)))
.set(dsl::child_member_count.eq(dsl::child_member_count - 1))
.execute(conn) .execute(conn)
.map(|c| c as u64) .map(|c| c as u64)
} }

View File

@ -39,9 +39,9 @@ table! {
/// ///
state_changed_at -> Nullable<BigInt>, state_changed_at -> Nullable<BigInt>,
/// ///
referrer_member_id -> Nullable<Uuid>, parent_member_id -> Nullable<Uuid>,
/// ///
referred_count -> BigInt, child_member_count -> BigInt,
/// ///
last_signined_ip -> Nullable<Text>, last_signined_ip -> Nullable<Text>,
/// ///

View File

@ -0,0 +1,9 @@
//!
//!
///
pub mod models;
///
pub mod repository;
///
pub mod schema;

View File

@ -0,0 +1,60 @@
use super::schema::member_referrers;
use beteran_common_rust as bcr;
///
#[derive(Eq, Hash, Identifiable, Queryable, PartialEq, Debug, Clone)]
#[table_name = "member_referrers"]
pub struct MemberReferrer {
///
pub id: uuid::Uuid,
///
pub member_id: uuid::Uuid,
///
pub can_use: bool,
///
pub memo: Option<String>,
///
pub created_at: i64,
///
pub updated_at: i64,
}
///
#[derive(Insertable, Debug, Clone)]
#[table_name = "member_referrers"]
pub struct NewMemberReferrer {
///
pub member_id: uuid::Uuid,
}
///
#[derive(AsChangeset, Debug, Clone)]
#[table_name = "member_referrers"]
pub struct ModifyMemberReferrer {
///
pub can_use: bool,
///
pub memo: Option<String>,
}
///
#[derive(Debug, Clone)]
pub struct FindAllSearch {
///
pub member_id: Option<uuid::Uuid>,
///
pub can_use: Option<bool>,
///
pub memo_like: Option<String>,
}
///
#[derive(Debug, Clone)]
pub struct FindAll {
///
pub search: Option<FindAllSearch>,
///
pub pagination: Option<bcr::models::pagination::Pagination>,
///
pub sorts: Option<Vec<bcr::models::pagination::Sort>>,
}

View File

@ -0,0 +1,195 @@
//!
//!
use super::{models, schema::member_referrers};
use beteran_common_rust as bcr;
use diesel::prelude::*;
use diesel::result::Error;
///
pub struct Repository {}
impl std::fmt::Debug for Repository {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
f.debug_struct("Repository of member_referrers").finish()
}
}
impl Default for Repository {
fn default() -> Self {
Self::new()
}
}
impl Repository {
///
pub fn new() -> Repository {
Repository {}
}
///
pub fn insert(
&self,
conn: &diesel::PgConnection,
new_resource: &models::NewMemberReferrer,
) -> Result<models::MemberReferrer, Error> {
let inserted = diesel::insert_into(member_referrers::table)
.values(new_resource)
.get_result::<models::MemberReferrer>(conn)?;
Ok(inserted)
}
///
pub fn select(
&self,
conn: &diesel::PgConnection,
id: uuid::Uuid,
) -> Result<Option<models::MemberReferrer>, Error> {
match member_referrers::table
.find(id as uuid::Uuid)
.first::<models::MemberReferrer>(conn)
{
Ok(m) => Ok(Some(m)),
Err(e) => match e {
diesel::result::Error::NotFound => Ok(None),
_ => Err(e),
},
}
}
///
pub fn select_by_member_id(
&self,
conn: &diesel::PgConnection,
member_id: uuid::Uuid,
) -> Result<Option<models::MemberReferrer>, Error> {
use member_referrers::dsl;
match member_referrers::table
.filter(dsl::member_id.eq(member_id))
.first::<models::MemberReferrer>(conn)
{
Ok(m) => Ok(Some(m)),
Err(e) => match e {
diesel::result::Error::NotFound => Ok(None),
_ => Err(e),
},
}
}
///
pub fn select_all_count(
&self,
conn: &diesel::PgConnection,
find_all: &models::FindAll,
) -> Result<i64, Error> {
let mut q = member_referrers::table.into_boxed();
if let Some(s) = &find_all.search {
if let Some(sp) = s.member_id {
q = q.filter(member_referrers::dsl::member_id.eq(sp));
}
if let Some(sp) = s.can_use {
q = q.filter(member_referrers::dsl::can_use.eq(sp));
}
if let Some(sp) = &s.memo_like {
q = q.filter(member_referrers::dsl::memo.like(sp));
}
}
q.count().get_result(conn)
}
///
pub fn select_all(
&self,
conn: &diesel::PgConnection,
find_all: &models::FindAll,
) -> Result<Vec<models::MemberReferrer>, Error> {
let mut q = member_referrers::table.into_boxed();
if let Some(s) = &find_all.search {
if let Some(sp) = s.member_id {
q = q.filter(member_referrers::dsl::member_id.eq(sp));
}
if let Some(sp) = s.can_use {
q = q.filter(member_referrers::dsl::can_use.eq(sp));
}
if let Some(sp) = &s.memo_like {
q = q.filter(member_referrers::dsl::memo.like(sp));
}
}
if let Some(p) = &find_all.pagination {
let page = p.page.unwrap_or(1);
if let Some(page_size) = p.page_size {
q = q.offset(((page - 1) * page_size) as i64);
q = q.limit(page_size as i64);
}
}
if let Some(orderbys) = &find_all.sorts {
for s in orderbys {
match s {
bcr::models::pagination::Sort::ASC(property) => match property.as_str() {
"member_id" => {
q = q.order_by(member_referrers::member_id.asc());
}
"can_use" => {
q = q.order_by(member_referrers::can_use.asc());
}
"created_at" => {
q = q.order_by(member_referrers::created_at.asc());
}
"updated_at" => {
q = q.order_by(member_referrers::updated_at.asc());
}
_ => {}
},
bcr::models::pagination::Sort::DESC(property) => match property.as_str() {
"member_id" => {
q = q.order_by(member_referrers::member_id.desc());
}
"can_use" => {
q = q.order_by(member_referrers::can_use.desc());
}
"created_at" => {
q = q.order_by(member_referrers::created_at.desc());
}
"updated_at" => {
q = q.order_by(member_referrers::updated_at.desc());
}
_ => {}
},
};
}
}
q.load::<models::MemberReferrer>(conn)
}
///
pub fn update(
&self,
conn: &diesel::PgConnection,
id: uuid::Uuid,
modify: &models::ModifyMemberReferrer,
) -> Result<u64, Error> {
use member_referrers::dsl;
diesel::update(dsl::member_referrers.filter(dsl::id.eq(id)))
.set(modify)
.execute(conn)
.map(|c| c as u64)
}
///
pub fn delete(&self, conn: &diesel::PgConnection, id: uuid::Uuid) -> Result<u64, Error> {
use member_referrers::dsl;
diesel::delete(member_referrers::table.filter(dsl::id.eq(id)))
.execute(conn)
.map(|c| c as u64)
}
}

View File

@ -0,0 +1,20 @@
//!
//!
table! {
///
member_referrers(id) {
///
id -> Uuid,
///
member_id -> Uuid,
///
can_use -> Bool,
///
memo -> Nullable<Text>,
///
created_at -> BigInt,
///
updated_at -> BigInt,
}
}

View File

@ -6,6 +6,7 @@ pub mod member_bank_withdraw;
pub mod member_class; pub mod member_class;
pub mod member_level; pub mod member_level;
pub mod member_permission; pub mod member_permission;
pub mod member_referrer;
pub mod member_role; pub mod member_role;
pub mod member_session; pub mod member_session;
pub mod resource; pub mod resource;

View File

@ -35,8 +35,8 @@ impl From<&repositories::member::models::Member> for bpr::models::member::Member
site_id: d.site_id.to_string(), site_id: d.site_id.to_string(),
member_class_id: d.member_class_id.to_string(), member_class_id: d.member_class_id.to_string(),
member_level_id: d.member_level_id.to_string(), member_level_id: d.member_level_id.to_string(),
referrer_member_id: d.referrer_member_id.map(|d| d.to_string()), parent_member_id: d.parent_member_id.map(|d| d.to_string()),
referred_count: d.referred_count as u64, child_member_count: d.child_member_count as u64,
username: d.username.clone(), username: d.username.clone(),
nickname: d.nickname.clone(), nickname: d.nickname.clone(),
mobile_phone_number: d.mobile_phone_number.clone(), mobile_phone_number: d.mobile_phone_number.clone(),
@ -66,11 +66,11 @@ impl From<&compositions::member::models::MemberModel> for bpr::models::member::M
.bank_account .bank_account
.as_ref() .as_ref()
.map(|d| bpr::models::member_bank_account::MemberBankAccountModel::from(d)), .map(|d| bpr::models::member_bank_account::MemberBankAccountModel::from(d)),
referrer_member: d parent_member: d
.referrer_member .parent_member
.as_ref() .as_ref()
.map(bpr::models::member::Member::from), .map(bpr::models::member::Member::from),
referred_count: d.referred_count as u64, child_member_count: d.child_member_count as u64,
username: d.username.clone(), username: d.username.clone(),
nickname: d.nickname.clone(), nickname: d.nickname.clone(),
mobile_phone_number: d.mobile_phone_number.clone(), mobile_phone_number: d.mobile_phone_number.clone(),

View File

@ -200,7 +200,7 @@ impl Service<'_> {
}) })
})?; })?;
let referrer_member_id = match request.referrer_member_username { let parent_member_id = match request.referrer_member_username {
Some(referrer_member_username) => { Some(referrer_member_username) => {
match self match self
.member_repository .member_repository
@ -303,7 +303,7 @@ impl Service<'_> {
site_id, site_id,
member_class_id, member_class_id,
member_level_id, member_level_id,
referrer_member_id, parent_member_id,
username: request.username, username: request.username,
password: password_hash, password: password_hash,
nickname: request.nickname, nickname: request.nickname,
@ -504,18 +504,7 @@ impl Service<'_> {
} }
}; };
let ( let search = match request.search {
site_id,
member_class_id,
member_level_id,
referrer_member_id,
username_like,
nickname_like,
mobile_phone_number_like,
last_signined_ip,
state,
deleted_at,
) = match request.search {
Some(s) => { Some(s) => {
let site_id = match s.site_id { let site_id = match s.site_id {
Some(d) => match uuid::Uuid::from_str(d.as_str()) { Some(d) => match uuid::Uuid::from_str(d.as_str()) {
@ -577,16 +566,16 @@ impl Service<'_> {
}, },
None => None, None => None,
}; };
let referrer_member_id = match s.referrer_member_id { let parent_member_id = match s.parent_member_id {
Some(d) => match uuid::Uuid::from_str(d.as_str()) { Some(d) => match uuid::Uuid::from_str(d.as_str()) {
Ok(dd) => Some(dd), Ok(dd) => Some(dd),
Err(e) => { Err(e) => {
return Err(bcr::error::rpc::Error::InvalidParams( return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams { bcr::error::rpc::InvalidParams {
message: "invalid referrer_member_id param".to_string(), message: "invalid parent_member_id param".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail { detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(), location: "request".to_string(),
param: "referrer_member_id".to_string(), param: "parent_member_id".to_string(),
value: d.clone(), value: d.clone(),
error_type: bcr::error::rpc::InvalidParamsType::Required, error_type: bcr::error::rpc::InvalidParamsType::Required,
message: e.to_string(), message: e.to_string(),
@ -598,20 +587,20 @@ impl Service<'_> {
None => None, None => None,
}; };
( Some(repositories::member::models::FindAllSearch {
site_id, site_id,
member_class_id, member_class_id,
member_level_id, member_level_id,
referrer_member_id, parent_member_id,
s.username_like, username_like: s.username_like,
s.nickname_like, nickname_like: s.nickname_like,
s.mobile_phone_number_like, mobile_phone_number_like: s.mobile_phone_number_like,
s.last_signined_ip, last_signined_ip: s.last_signined_ip,
s.state.map(repositories::member::schema::MemberState::from), state: s.state.map(repositories::member::schema::MemberState::from),
s.deleted_at.map(|d| d > 0), deleted_at: s.deleted_at.map(|d| d > 0),
) })
} }
None => (None, None, None, None, None, None, None, None, None, None), None => None,
}; };
let conn = self.pool.get().map_err(|e| { let conn = self.pool.get().map_err(|e| {
@ -623,18 +612,6 @@ impl Service<'_> {
})?; })?;
let find_all = repositories::member::models::FindAll { let find_all = repositories::member::models::FindAll {
search: Some(repositories::member::models::FindAllSearch {
site_id,
member_class_id,
member_level_id,
referrer_member_id,
username_like,
nickname_like,
mobile_phone_number_like,
last_signined_ip,
state,
deleted_at,
}),
pagination: request pagination: request
.pagination .pagination
.as_ref() .as_ref()
@ -646,6 +623,7 @@ impl Service<'_> {
.map(beteran_common_rust::models::pagination::Sort::from) .map(beteran_common_rust::models::pagination::Sort::from)
.collect(), .collect(),
), ),
search,
}; };
let m = self let m = self

View File

@ -3,7 +3,6 @@
use std::str::FromStr; use std::str::FromStr;
use super::models;
use crate::compositions; use crate::compositions;
use crate::repositories; use crate::repositories;
use beteran_common_rust as bcr; use beteran_common_rust as bcr;

View File

@ -0,0 +1,2 @@
pub mod models;
pub mod service;

View File

@ -0,0 +1,33 @@
use crate::compositions;
use crate::repositories;
use beteran_protobuf_rust as bpr;
impl From<&repositories::member_referrer::models::MemberReferrer>
for bpr::models::member_referrer::MemberReferrer
{
fn from(d: &repositories::member_referrer::models::MemberReferrer) -> Self {
bpr::models::member_referrer::MemberReferrer {
id: d.id.to_string(),
member_id: d.member_id.to_string(),
can_use: d.can_use,
memo: d.memo.clone(),
created_at: d.created_at as u64,
updated_at: d.updated_at as u64,
}
}
}
impl From<&compositions::member_referrer::models::MemberReferrerModel>
for bpr::models::member_referrer::MemberReferrerModel
{
fn from(d: &compositions::member_referrer::models::MemberReferrerModel) -> Self {
bpr::models::member_referrer::MemberReferrerModel {
id: d.id.to_string(),
member: Some(bpr::models::member::Member::from(&d.member)),
can_use: d.can_use,
memo: d.memo.clone(),
created_at: d.created_at as u64,
updated_at: d.updated_at as u64,
}
}
}

View File

@ -0,0 +1,791 @@
//!
//!
use std::str::FromStr;
use crate::compositions;
use crate::repositories;
use beteran_common_rust as bcr;
use beteran_protobuf_rust as bpr;
use diesel::{
r2d2::{ConnectionManager, Pool},
PgConnection,
};
use prost::Message;
///
pub struct Service {
connection_broker: nats::asynk::Connection,
queue_broker: String,
pool: Pool<ConnectionManager<PgConnection>>,
site_composition: compositions::site::composition::Composition,
member_referrer_repository: repositories::member_referrer::repository::Repository,
member_referrer_composition: compositions::member_referrer::composition::Composition,
}
impl std::fmt::Debug for Service {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
f.debug_struct("Service of service.member.service.identity")
.finish()
}
}
impl Service {
///
pub fn new(
connection_broker: nats::asynk::Connection,
queue_broker: String,
pool: Pool<ConnectionManager<PgConnection>>,
) -> Service {
Service {
connection_broker,
queue_broker,
pool,
site_composition: compositions::site::composition::Composition::new(),
member_referrer_repository: repositories::member_referrer::repository::Repository::new(),
member_referrer_composition: compositions::member_referrer::composition::Composition::new(),
}
}
pub async fn subscribe(&self) -> std::result::Result<(), std::boxed::Box<dyn std::error::Error>> {
futures::try_join!(
self.create_member_referrer(),
self.list_member_referrers(),
self.get_member_referrer(),
self.update_member_referrer(),
self.delete_member_referrer(),
)
.map(|_| ())
}
fn check_site(
&self,
conn: &diesel::PgConnection,
url: Option<String>,
site_id: uuid::Uuid,
) -> Result<repositories::site::models::Site, bcr::error::rpc::Error> {
match self
.site_composition
.select_by_url(conn, url, site_id)
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})? {
Some(s) => Ok(s),
None => Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid site_url information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "client.site_url".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::None,
message: "".to_string(),
},
},
)),
}
}
async fn create_member_referrer(&self) -> Result<(), Box<dyn std::error::Error>> {
let s = self
.connection_broker
.queue_subscribe(
bpr::ss::member_referrer::SUBJECT_CREATE_MEMBER_REFERRER,
self.queue_broker.as_str(),
)
.await?;
while let Some(message) = s.next().await {
if let Err(e) = async {
let req =
bpr::ss::member_referrer::CreateMemberReferrerRequest::decode(message.data.as_slice())
.map_err(|e| {
bcr::error::rpc::Error::InvalidRequest(bcr::error::rpc::InvalidRequest {
message: format!("invalid request: {}", e),
})
})?;
let client = match req.client {
Some(c) => c,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid client information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "client".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let request = match req.request {
Some(r) => r,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid request information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "request".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let member_id = uuid::Uuid::from_str(request.member_id.as_str()).map_err(|e| {
bcr::error::rpc::Error::InvalidParams(bcr::error::rpc::InvalidParams {
message: "invalid member_id param".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "member_id".to_string(),
value: request.member_id.clone(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: e.to_string(),
},
})
})?;
let conn = self.pool.get().map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
let new_member_referrer =
repositories::member_referrer::models::NewMemberReferrer { member_id };
let s = self
.member_referrer_repository
.insert(&conn, &new_member_referrer)
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
let mr = self
.member_referrer_composition
.select(&conn, s.id)
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
message
.respond(
bpr::ss::member_referrer::CreateMemberReferrerResponse {
error: None,
result: Some(
bpr::ss::member_referrer::create_member_referrer_response::Result {
member_referrer: mr
.as_ref()
.map(bpr::models::member_referrer::MemberReferrerModel::from),
},
),
}
.encode_to_vec(),
)
.await
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
Ok::<(), bcr::error::rpc::Error>(())
}
.await
{
message
.respond(
bpr::ss::member_referrer::CreateMemberReferrerResponse {
error: Some(bpr::protobuf::rpc::Error::from(e)),
result: None,
}
.encode_to_vec(),
)
.await?;
}
}
Ok(())
}
async fn list_member_referrers(&self) -> Result<(), Box<dyn std::error::Error>> {
let s = self
.connection_broker
.queue_subscribe(
bpr::ss::member_referrer::SUBJECT_LIST_MEMBER_REFERRERS,
self.queue_broker.as_str(),
)
.await?;
while let Some(message) = s.next().await {
if let Err(e) = async {
let req =
bpr::ss::member_referrer::ListMemberReferrersRequest::decode(message.data.as_slice())
.map_err(|e| {
bcr::error::rpc::Error::InvalidRequest(bcr::error::rpc::InvalidRequest {
message: format!("invalid request: {}", e),
})
})?;
let client = match req.client {
Some(c) => c,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid client information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "client".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let request = match req.request {
Some(r) => r,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid request information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "request".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let search = match request.search {
Some(s) => {
let member_id = match s.member_id {
Some(d) => match uuid::Uuid::from_str(d.as_str()) {
Ok(dd) => Some(dd),
Err(e) => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid member_id param".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "member_id".to_string(),
value: d.clone(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: e.to_string(),
},
},
));
}
},
None => None,
};
Some(repositories::member_referrer::models::FindAllSearch {
member_id,
can_use: s.can_use,
memo_like: s.memo_like,
})
}
None => None,
};
let conn = self.pool.get().map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
let find_all = repositories::member_referrer::models::FindAll {
search,
pagination: request
.pagination
.as_ref()
.map(bcr::models::pagination::Pagination::from),
sorts: Some(
request
.sorts
.iter()
.map(beteran_common_rust::models::pagination::Sort::from)
.collect(),
),
};
let count = self
.member_referrer_repository
.select_all_count(&conn, &find_all)
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
let list = self
.member_referrer_composition
.select_all(&conn, &find_all)
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
message
.respond(
bpr::ss::member_referrer::ListMemberReferrersResponse {
error: None,
result: Some(
bpr::ss::member_referrer::list_member_referrers_response::Result {
member_referrers: list
.iter()
.map(|d| bpr::models::member_referrer::MemberReferrerModel::from(d))
.collect(),
},
),
}
.encode_to_vec(),
)
.await
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
Ok::<(), bcr::error::rpc::Error>(())
}
.await
{
message
.respond(
bpr::ss::member_referrer::ListMemberReferrersResponse {
error: Some(bpr::protobuf::rpc::Error::from(e)),
result: None,
}
.encode_to_vec(),
)
.await?;
}
}
Ok(())
}
async fn get_member_referrer(&self) -> Result<(), Box<dyn std::error::Error>> {
let s = self
.connection_broker
.queue_subscribe(
bpr::ss::member_referrer::SUBJECT_GET_MEMBER_REFERRER,
self.queue_broker.as_str(),
)
.await?;
while let Some(message) = s.next().await {
if let Err(e) = async {
let req =
bpr::ss::member_referrer::GetMemberReferrerRequest::decode(message.data.as_slice())
.map_err(|e| {
bcr::error::rpc::Error::InvalidRequest(bcr::error::rpc::InvalidRequest {
message: format!("invalid request: {}", e),
})
})?;
let client = match req.client {
Some(c) => c,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid client information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "client".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let request = match req.request {
Some(r) => r,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid request information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "request".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let id = uuid::Uuid::from_str(request.id.as_str()).map_err(|e| {
bcr::error::rpc::Error::InvalidParams(bcr::error::rpc::InvalidParams {
message: "invalid id param".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "id".to_string(),
value: request.id.clone(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: e.to_string(),
},
})
})?;
let conn = self.pool.get().map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
let s = self
.member_referrer_composition
.select(&conn, id)
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
message
.respond(
bpr::ss::member_referrer::GetMemberReferrerResponse {
error: None,
result: Some(
bpr::ss::member_referrer::get_member_referrer_response::Result {
member_referrer: s
.map(|d| bpr::models::member_referrer::MemberReferrerModel::from(&d)),
},
),
}
.encode_to_vec(),
)
.await
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
Ok::<(), bcr::error::rpc::Error>(())
}
.await
{
message
.respond(
bpr::ss::member_referrer::GetMemberReferrerResponse {
error: Some(bpr::protobuf::rpc::Error::from(e)),
result: None,
}
.encode_to_vec(),
)
.await?;
}
}
Ok(())
}
async fn update_member_referrer(&self) -> Result<(), Box<dyn std::error::Error>> {
let s = self
.connection_broker
.queue_subscribe(
bpr::ss::member_referrer::SUBJECT_UPDATE_MEMBER_REFERRER,
self.queue_broker.as_str(),
)
.await?;
while let Some(message) = s.next().await {
if let Err(e) = async {
let req =
bpr::ss::member_referrer::UpdateMemberReferrerRequest::decode(message.data.as_slice())
.map_err(|e| {
bcr::error::rpc::Error::InvalidRequest(bcr::error::rpc::InvalidRequest {
message: format!("invalid request: {}", e),
})
})?;
let client = match req.client {
Some(c) => c,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid client information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "client".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let request = match req.request {
Some(r) => r,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid request information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "request".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let id = uuid::Uuid::from_str(request.id.as_str()).map_err(|e| {
bcr::error::rpc::Error::InvalidParams(bcr::error::rpc::InvalidParams {
message: "invalid id param".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "id".to_string(),
value: request.id.clone(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: e.to_string(),
},
})
})?;
let conn = self.pool.get().map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
let modify_member_referrer = repositories::member_referrer::models::ModifyMemberReferrer {
can_use: request.can_use,
memo: request.memo,
};
let _affected = self
.member_referrer_repository
.update(&conn, id, &modify_member_referrer)
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
let s = self
.member_referrer_composition
.select(&conn, id)
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
message
.respond(
bpr::ss::member_referrer::UpdateMemberReferrerResponse {
error: None,
result: Some(
bpr::ss::member_referrer::update_member_referrer_response::Result {
member_referrer: s
.map(|d| bpr::models::member_referrer::MemberReferrerModel::from(&d)),
},
),
}
.encode_to_vec(),
)
.await
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
Ok::<(), bcr::error::rpc::Error>(())
}
.await
{
message
.respond(
bpr::ss::member_referrer::UpdateMemberReferrerResponse {
error: Some(bpr::protobuf::rpc::Error::from(e)),
result: None,
}
.encode_to_vec(),
)
.await?;
}
}
Ok(())
}
async fn delete_member_referrer(&self) -> Result<(), Box<dyn std::error::Error>> {
let s = self
.connection_broker
.queue_subscribe(
bpr::ss::member_referrer::SUBJECT_DELETE_MEMBER_REFERRER,
self.queue_broker.as_str(),
)
.await?;
while let Some(message) = s.next().await {
if let Err(e) = async {
let req =
bpr::ss::member_referrer::DeleteMemberReferrerRequest::decode(message.data.as_slice())
.map_err(|e| {
bcr::error::rpc::Error::InvalidRequest(bcr::error::rpc::InvalidRequest {
message: format!("invalid request: {}", e),
})
})?;
let client = match req.client {
Some(c) => c,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid client information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "client".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let request = match req.request {
Some(r) => r,
None => {
return Err(bcr::error::rpc::Error::InvalidParams(
bcr::error::rpc::InvalidParams {
message: "invalid request information".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "request".to_string(),
value: "".to_string(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: "".to_string(),
},
},
));
}
};
let id = uuid::Uuid::from_str(request.id.as_str()).map_err(|e| {
bcr::error::rpc::Error::InvalidParams(bcr::error::rpc::InvalidParams {
message: "invalid id param".to_string(),
detail: bcr::error::rpc::InvalidParamsDetail {
location: "request".to_string(),
param: "id".to_string(),
value: request.id.clone(),
error_type: bcr::error::rpc::InvalidParamsType::Required,
message: e.to_string(),
},
})
})?;
let conn = self.pool.get().map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
let _affected = self
.member_referrer_repository
.delete(&conn, id)
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
message
.respond(
bpr::ss::member_referrer::DeleteMemberReferrerResponse {
error: None,
result: Some(bpr::ss::member_referrer::delete_member_referrer_response::Result {}),
}
.encode_to_vec(),
)
.await
.map_err(|e| {
bcr::error::rpc::Error::Server(bcr::error::rpc::Server {
code: bpr::protobuf::rpc::Error::SERVER_00,
message: format!("server {}", e),
data: None,
})
})?;
Ok::<(), bcr::error::rpc::Error>(())
}
.await
{
message
.respond(
bpr::ss::member_referrer::DeleteMemberReferrerResponse {
error: Some(bpr::protobuf::rpc::Error::from(e)),
result: None,
}
.encode_to_vec(),
)
.await?;
}
}
Ok(())
}
}

View File

@ -6,4 +6,5 @@ pub mod member_bank_deposit;
pub mod member_bank_withdraw; pub mod member_bank_withdraw;
pub mod member_class; pub mod member_class;
pub mod member_level; pub mod member_level;
pub mod member_referrer;
pub mod site; pub mod site;