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
|
use std::fmt;
use sqlx::{sqlite::Sqlite, SqliteConnection, Transaction};
use crate::id::Id as BaseId;
pub trait Provider {
fn channels(&mut self) -> Channels;
}
impl<'c> Provider for Transaction<'c, Sqlite> {
fn channels(&mut self) -> Channels {
Channels(self)
}
}
pub struct Channels<'t>(&'t mut SqliteConnection);
#[derive(Debug, Eq, PartialEq, serde::Serialize)]
pub struct Channel {
pub id: Id,
pub name: String,
}
impl<'c> Channels<'c> {
/// Create a new channel.
pub async fn create(&mut self, name: &str) -> Result<Channel, sqlx::Error> {
let id = Id::generate();
let channel = sqlx::query_as!(
Channel,
r#"
insert
into channel (id, name)
values ($1, $2)
returning id as "id: Id", name
"#,
id,
name,
)
.fetch_one(&mut *self.0)
.await?;
Ok(channel)
}
pub async fn by_id(&mut self, channel: &Id) -> Result<Channel, sqlx::Error> {
let channel = sqlx::query_as!(
Channel,
r#"
select id as "id: Id", name
from channel
where id = $1
"#,
channel,
)
.fetch_one(&mut *self.0)
.await?;
Ok(channel)
}
pub async fn all(&mut self) -> Result<Vec<Channel>, sqlx::Error> {
let channels = sqlx::query_as!(
Channel,
r#"
select
channel.id as "id: Id",
channel.name
from channel
order by channel.name
"#,
)
.fetch_all(&mut *self.0)
.await?;
Ok(channels)
}
}
/// Stable identifier for a [Channel]. Prefixed with `C`.
#[derive(Clone, Debug, Eq, Hash, PartialEq, sqlx::Type, serde::Deserialize, serde::Serialize)]
#[sqlx(transparent)]
#[serde(transparent)]
pub struct Id(BaseId);
impl From<BaseId> for Id {
fn from(id: BaseId) -> Self {
Self(id)
}
}
impl Id {
pub fn generate() -> Self {
BaseId::generate("C")
}
}
impl fmt::Display for Id {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.0.fmt(f)
}
}
|
