fish-lib 0.2.3

A work-in-progress fishing game library containing the game/storage logic for a discord fishing game I'm working on.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

use crate::database::DatabaseInterface;
use crate::game::errors::repository::GameRepositoryError;
use crate::models::item::{Item, NewItem};
use crate::schema::fish_items;
use crate::traits::repository::Repository;
use chrono::Utc;
use diesel::prelude::*;
use std::sync::{Arc, RwLock};

pub trait ItemRepositoryInterface: Repository<Item> + Send + Sync {
    fn find_by_user(&self, user_id: i64) -> Result<Vec<Item>, GameRepositoryError>;
    fn find_by_type_and_user(
        &self,
        type_id: i32,
        user_id: i64,
    ) -> Result<Vec<Item>, GameRepositoryError>;
}

pub struct ItemRepository {
    db: Arc<RwLock<dyn DatabaseInterface>>,
}

impl ItemRepository {
    pub fn new(db: Arc<RwLock<dyn DatabaseInterface>>) -> Self {
        Self { db }
    }
}

impl ItemRepositoryInterface for ItemRepository {
    fn find_by_user(&self, user_id: i64) -> Result<Vec<Item>, GameRepositoryError> {
        let mut connection = self.get_connection()?;

        let results = fish_items::table
            .filter(fish_items::user_id.eq(user_id))
            .load::<Item>(&mut connection)?;

        Ok(results)
    }

    fn find_by_type_and_user(
        &self,
        type_id: i32,
        user_id: i64,
    ) -> Result<Vec<Item>, GameRepositoryError> {
        let mut connection = self.get_connection()?;

        let results = fish_items::table
            .filter(
                fish_items::type_id
                    .eq(type_id)
                    .and(fish_items::user_id.eq(user_id)),
            )
            .load::<Item>(&mut connection)?;

        Ok(results)
    }
}

impl Repository<Item> for ItemRepository {
    fn get_db(&self) -> Arc<RwLock<dyn DatabaseInterface>> {
        self.db.clone()
    }

    fn create(&self, new_entity: NewItem) -> Result<Item, GameRepositoryError> {
        let mut connection = self.get_connection()?;

        let new_result = diesel::insert_into(fish_items::table)
            .values(new_entity)
            .get_result::<Item>(&mut connection)?;

        Ok(new_result)
    }

    fn find(&self, id: i64) -> Result<Option<Item>, GameRepositoryError> {
        let mut connection = self.get_connection()?;
        let item = fish_items::table
            .find(id)
            .first::<Item>(&mut connection)
            .optional()?;
        Ok(item)
    }

    fn save(&self, mut entity: Item) -> Result<Item, GameRepositoryError> {
        let mut connection = self.get_connection()?;
        entity.updated_at = Utc::now();

        let updated_item = diesel::update(fish_items::table)
            .filter(fish_items::id.eq(entity.id))
            .set(entity)
            .get_result::<Item>(&mut connection)?;

        Ok(updated_item)
    }

    fn delete(&self, entity: Item) -> Result<bool, GameRepositoryError> {
        let mut connection = self.get_connection()?;

        let deleted_count = diesel::delete(fish_items::table)
            .filter(fish_items::id.eq(entity.id))
            .execute(&mut connection)?;

        Ok(deleted_count > 0)
    }
}