mostro_core/
user.rs

1use chrono::Utc;
2use serde::{Deserialize, Serialize};
3#[cfg(feature = "sqlx")]
4use sqlx::FromRow;
5
6#[derive(Debug, Default, Deserialize, Serialize, Clone)]
7
8pub struct UserInfo {
9    /// User's rating
10    pub rating: f64,
11    /// User's total reviews
12    pub reviews: i64,
13    /// User's operating days
14    pub operating_days: u64,
15}
16
17/// Database representation of an user
18#[cfg_attr(feature = "sqlx", derive(FromRow))]
19#[derive(Debug, Default, Deserialize, Serialize, Clone, PartialEq)]
20pub struct User {
21    pub pubkey: String,
22    pub is_admin: i64,
23    pub is_solver: i64,
24    pub is_banned: i64,
25    pub category: i64,
26    /// We have to be sure that when a user creates a new order (or takes an order),
27    /// the trade_index is greater than the one we have in database
28    pub last_trade_index: i64,
29    pub total_reviews: i64,
30    pub total_rating: f64,
31    pub last_rating: i64,
32    pub max_rating: i64,
33    pub min_rating: i64,
34    pub created_at: i64,
35}
36
37impl User {
38    pub fn new(
39        pubkey: String,
40        is_admin: i64,
41        is_solver: i64,
42        is_banned: i64,
43        category: i64,
44        trade_index: i64,
45    ) -> Self {
46        Self {
47            pubkey,
48            is_admin,
49            is_solver,
50            is_banned,
51            category,
52            last_trade_index: trade_index,
53            total_reviews: 0,
54            total_rating: 0.0,
55            last_rating: 0,
56            max_rating: 0,
57            min_rating: 0,
58            created_at: Utc::now().timestamp(),
59        }
60    }
61
62    /// Update user rating
63    pub fn update_rating(&mut self, rating: u8) {
64        // Update user reputation
65        // increment first
66        self.total_reviews += 1;
67        let old_rating = self.total_rating;
68        // recompute new rating
69        if self.total_reviews <= 1 {
70            self.total_rating = rating.into();
71            self.max_rating = rating.into();
72            self.min_rating = rating.into();
73        } else {
74            self.total_rating =
75                old_rating + ((self.last_rating as f64) - old_rating) / (self.total_reviews as f64);
76            if self.max_rating < rating.into() {
77                self.max_rating = rating.into();
78            }
79            if self.min_rating > rating.into() {
80                self.min_rating = rating.into();
81            }
82        }
83        // Store last rating
84        self.last_rating = rating.into();
85    }
86}
mostro_core/user.rs

mostro_core/
user.rs
