icydb_core/db/query/session/
delete.rs1use crate::{
2 db::{
3 DbSession,
4 query::{
5 Query, QueryError,
6 expr::{FilterExpr, SortExpr},
7 plan::{ExecutablePlan, ExplainPlan},
8 predicate::Predicate,
9 },
10 response::Response,
11 },
12 traits::{CanisterKind, EntityKind, EntityValue, SingletonEntity},
13};
14
15pub struct SessionDeleteQuery<'a, C, E>
24where
25 C: CanisterKind,
26 E: EntityKind<Canister = C>,
27{
28 session: &'a DbSession<C>,
29 query: Query<E>,
30}
31
32impl<'a, C, E> SessionDeleteQuery<'a, C, E>
33where
34 C: CanisterKind,
35 E: EntityKind<Canister = C>,
36{
37 pub(crate) const fn new(session: &'a DbSession<C>, query: Query<E>) -> Self {
38 Self { session, query }
39 }
40
41 #[must_use]
46 pub const fn query(&self) -> &Query<E> {
47 &self.query
48 }
49
50 #[must_use]
55 pub fn by_key(mut self, key: E::Id) -> Self {
56 self.query = self.query.by_key(key);
57 self
58 }
59
60 #[must_use]
61 pub fn many<I>(mut self, keys: I) -> Self
62 where
63 I: IntoIterator<Item = E::Id>,
64 {
65 self.query = self.query.by_keys(keys);
66 self
67 }
68
69 #[must_use]
70 pub fn filter(mut self, predicate: Predicate) -> Self {
71 self.query = self.query.filter(predicate);
72 self
73 }
74
75 pub fn filter_expr(mut self, expr: FilterExpr) -> Result<Self, QueryError> {
76 self.query = self.query.filter_expr(expr)?;
77 Ok(self)
78 }
79
80 pub fn sort_expr(mut self, expr: SortExpr) -> Result<Self, QueryError> {
81 self.query = self.query.sort_expr(expr)?;
82 Ok(self)
83 }
84
85 #[must_use]
86 pub fn order_by(mut self, field: impl AsRef<str>) -> Self {
87 self.query = self.query.order_by(field);
88 self
89 }
90
91 #[must_use]
92 pub fn order_by_desc(mut self, field: impl AsRef<str>) -> Self {
93 self.query = self.query.order_by_desc(field);
94 self
95 }
96
97 #[must_use]
98 pub fn limit(mut self, limit: u32) -> Self {
99 self.query = self.query.limit(limit);
100 self
101 }
102
103 pub fn explain(&self) -> Result<ExplainPlan, QueryError> {
108 self.query.explain()
109 }
110
111 pub fn plan(&self) -> Result<ExecutablePlan<E>, QueryError> {
112 self.query.plan()
113 }
114
115 pub fn execute(&self) -> Result<Response<E>, QueryError>
123 where
124 E: EntityValue,
125 {
126 self.session.execute_query(self.query())
127 }
128
129 pub fn is_empty(&self) -> Result<bool, QueryError>
131 where
132 E: EntityValue,
133 {
134 Ok(self.execute()?.is_empty())
135 }
136
137 pub fn count(&self) -> Result<u32, QueryError>
139 where
140 E: EntityValue,
141 {
142 Ok(self.execute()?.count())
143 }
144
145 pub fn require_one(&self) -> Result<(), QueryError>
147 where
148 E: EntityValue,
149 {
150 self.execute()?.require_one().map_err(QueryError::Response)
151 }
152
153 pub fn require_some(&self) -> Result<(), QueryError>
155 where
156 E: EntityValue,
157 {
158 self.execute()?.require_some().map_err(QueryError::Response)
159 }
160}
161
162impl<C, E> SessionDeleteQuery<'_, C, E>
163where
164 C: CanisterKind,
165 E: EntityKind<Canister = C> + SingletonEntity,
166{
167 #[must_use]
169 pub fn only(mut self, id: E::Id) -> Self {
170 self.query = self.query.only(id);
171 self
172 }
173}