leetcode_datahouse/src/database.rs at main · Narayanbhat166/leetcode_datahouse · GitHub

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
use diesel::pg::PgConnection;
use diesel::prelude::*;

use crate::models;
use crate::schema::submission;

#[derive(Queryable, Debug)]
pub struct Submission {
    pub id: i64,
    pub memory: Option<f32>,
    pub memory_display: Option<String>,
    pub memory_percentile: Option<f32>,
    pub notes: Option<String>,
    pub runtime: Option<f32>,
    pub runtime_percentile: Option<f32>,
    pub status_code: Option<i32>,
    pub timestamp: Option<i32>,
    pub code_hash: Option<String>,
    pub username: Option<String>,
    pub question_id: Option<String>,
}

#[derive(Insertable)]
#[diesel(table_name= submission)]
pub struct NewSubmission {
    pub id: i64,
    pub memory: Option<f32>,
    pub memory_display: Option<String>,
    pub memory_percentile: Option<f32>,
    pub notes: Option<String>,
    pub runtime: Option<f32>,
    pub runtime_percentile: Option<f32>,
    pub status_code: Option<i32>,
    pub timestamp: Option<i32>,
    pub code_hash: Option<String>,
    pub username: Option<String>,
    pub question_id: Option<String>,
}

pub fn create_connection(database_url: &str) -> Result<PgConnection, ConnectionError> {
    PgConnection::establish(database_url)
}

pub fn insert_submission(
    new_submission: NewSubmission,
    conn: &mut PgConnection,
) -> Result<Submission, diesel::result::Error> {
    let query = diesel::insert_into(submission::dsl::submission).values(new_submission);
    query.get_result::<Submission>(conn)
}

impl From<models::ScrappedResponse> for NewSubmission {
    fn from(item: models::ScrappedResponse) -> Self {
        let submission_details =
            item.submission_data
                .data
                .submission_details
                .map(|submission_data| {
                    let code_hash = None;
                    // utils::generate_hash(&submission_data.code.clone().unwrap_or_default());
                    let username = submission_data.user.map(|user| user.username);

                    let submission_id = i64::try_from(item.submission_id).unwrap();

                    Self {
                        id: submission_id,
                        memory: submission_data.memory,
                        memory_display: submission_data.memory_display,
                        memory_percentile: submission_data.memory_percentile,
                        notes: submission_data.notes,
                        runtime: submission_data.runtime,
                        runtime_percentile: submission_data.runtime_percentile,
                        status_code: submission_data.status_code,
                        timestamp: submission_data.timestamp,
                        code_hash,
                        username,
                        question_id: submission_data
                            .question
                            .map(|question| question.question_id),
                    }
                });

        submission_details.unwrap()
    }
}