95 lines
2.6 KiB
Rust
95 lines
2.6 KiB
Rust
use anyhow::Result;
|
|
use async_trait::async_trait;
|
|
use chrono::prelude::*;
|
|
use ipnetwork::IpNetwork;
|
|
use serde::{Deserialize, Serialize};
|
|
use sqlx::postgres::PgPool;
|
|
use uuid::Uuid;
|
|
|
|
use super::ListParams;
|
|
use super::Model;
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone)]
|
|
pub struct Owner {
|
|
pub id: Option<i32>,
|
|
pub name: String,
|
|
pub api_key: Uuid,
|
|
pub ip_address: Option<IpNetwork>,
|
|
pub mod_version: String,
|
|
pub created_at: Option<NaiveDateTime>,
|
|
pub updated_at: Option<NaiveDateTime>,
|
|
}
|
|
|
|
#[async_trait]
|
|
impl Model for Owner {
|
|
fn resource_name() -> &'static str {
|
|
"owner"
|
|
}
|
|
|
|
fn pk(&self) -> Option<i32> {
|
|
self.id
|
|
}
|
|
|
|
async fn get(db: &PgPool, id: i32) -> Result<Self> {
|
|
let timer = std::time::Instant::now();
|
|
let result = sqlx::query_as!(Self, "SELECT * FROM owners WHERE id = $1", id)
|
|
.fetch_one(db)
|
|
.await?;
|
|
let elapsed = timer.elapsed();
|
|
debug!("SELECT * FROM owners ... {:.3?}", elapsed);
|
|
Ok(result)
|
|
}
|
|
|
|
async fn save(self, db: &PgPool) -> Result<Self> {
|
|
let timer = std::time::Instant::now();
|
|
let result = sqlx::query_as!(
|
|
Self,
|
|
"INSERT INTO owners
|
|
(name, api_key, ip_address, mod_version, created_at, updated_at)
|
|
VALUES ($1, $2, $3, $4, now(), now())
|
|
RETURNING *",
|
|
self.name,
|
|
self.api_key,
|
|
self.ip_address,
|
|
self.mod_version,
|
|
)
|
|
.fetch_one(db)
|
|
.await?;
|
|
let elapsed = timer.elapsed();
|
|
debug!("INSERT INTO owners ... {:.3?}", elapsed);
|
|
Ok(result)
|
|
}
|
|
|
|
async fn list(db: &PgPool, list_params: ListParams) -> Result<Vec<Self>> {
|
|
let timer = std::time::Instant::now();
|
|
let result = if let Some(order_by) = list_params.get_order_by() {
|
|
sqlx::query_as!(
|
|
Self,
|
|
"SELECT * FROM owners
|
|
ORDER BY $1
|
|
LIMIT $2
|
|
OFFSET $3",
|
|
order_by,
|
|
list_params.limit.unwrap_or(10),
|
|
list_params.offset.unwrap_or(0),
|
|
)
|
|
.fetch_all(db)
|
|
.await?
|
|
} else {
|
|
sqlx::query_as!(
|
|
Self,
|
|
"SELECT * FROM owners
|
|
LIMIT $1
|
|
OFFSET $2",
|
|
list_params.limit.unwrap_or(10),
|
|
list_params.offset.unwrap_or(0),
|
|
)
|
|
.fetch_all(db)
|
|
.await?
|
|
};
|
|
let elapsed = timer.elapsed();
|
|
debug!("SELECT * FROM owners ... {:.3?}", elapsed);
|
|
Ok(result)
|
|
}
|
|
}
|