plane 0.5.5

Session backend orchestrator for ambitious browser-based apps.
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
105
106
107
108
109
110
111
112
113
114

use super::{connect::ConnectError, subscribe::emit_with_key};
use plane_common::{
    names::{BackendActionName, BackendName, Name},
    protocol::{BackendAction, BackendActionMessage},
    types::NodeId,
};
use sqlx::{PgPool, Postgres};

pub struct BackendActionDatabase {
    pool: PgPool,
}

impl BackendActionDatabase {
    pub fn new(pool: &PgPool) -> Self {
        // TODO: can't use a reference because the futures need to be Send.
        Self { pool: pool.clone() }
    }

    pub async fn ack_pending_action(
        &self,
        notification_id: &BackendActionName,
        drone_id: NodeId,
    ) -> anyhow::Result<()> {
        sqlx::query!(
            r#"
            update "backend_action"
            set acked_at = now()
            where "id" = $1
            and "drone_id" = $2
            "#,
            notification_id.to_string(),
            drone_id.as_i32(),
        )
        .execute(&self.pool)
        .await?;

        Ok(())
    }

    pub async fn pending_actions(
        &self,
        drone: NodeId,
        limit: i64,
    ) -> anyhow::Result<Vec<BackendActionMessage>> {
        let rows = sqlx::query!(
            r#"
            select "action"
            from "backend_action"
            where "drone_id" = $1
            and acked_at is null
            order by created_at asc
            limit $2
            "#,
            drone.as_i32(),
            limit,
        )
        .fetch_all(&self.pool)
        .await?;

        let mut actions = Vec::new();

        for row in rows {
            let action: BackendActionMessage = serde_json::from_value(row.action)?;
            actions.push(action);
        }

        Ok(actions)
    }

    pub async fn create_pending_action(
        &self,
        backend_id: &BackendName,
        drone_id: NodeId,
        action: &BackendAction,
    ) -> Result<(), ConnectError> {
        let mut txn = self.pool.begin().await?;
        create_pending_action(&mut txn, backend_id, drone_id, action).await?;
        txn.commit().await?;
        Ok(())
    }
}

pub async fn create_pending_action(
    txn: &mut sqlx::Transaction<'_, Postgres>,
    backend_id: &BackendName,
    drone_id: NodeId,
    action: &BackendAction,
) -> Result<(), ConnectError> {
    let action_id = BackendActionName::new_random();

    let backend_action = BackendActionMessage {
        action_id: action_id.clone(),
        backend_id: backend_id.clone(),
        drone_id,
        action: action.clone(),
    };

    emit_with_key(&mut *txn, &drone_id.to_string(), &backend_action).await?;

    sqlx::query!(
        r#"
        insert into "backend_action" ("id", "action", "drone_id", "backend_id")
        values ($1, $2, $3, $4)
        "#,
        action_id.to_string(),
        serde_json::to_value(&backend_action).expect("Backend action is always serializable"),
        drone_id.as_i32(),
        backend_id.to_string(),
    )
    .execute(&mut **txn)
    .await?;

    Ok(())
}