mirror of
https://github.com/mCaptcha/mCaptcha.git
synced 2026-02-12 10:35:39 +00:00
Deploying to gh-pages from @ b30bc67bd4 🚀
This commit is contained in:
@@ -943,6 +943,21 @@
|
||||
<span id="938">938</span>
|
||||
<span id="939">939</span>
|
||||
<span id="940">940</span>
|
||||
<span id="941">941</span>
|
||||
<span id="942">942</span>
|
||||
<span id="943">943</span>
|
||||
<span id="944">944</span>
|
||||
<span id="945">945</span>
|
||||
<span id="946">946</span>
|
||||
<span id="947">947</span>
|
||||
<span id="948">948</span>
|
||||
<span id="949">949</span>
|
||||
<span id="950">950</span>
|
||||
<span id="951">951</span>
|
||||
<span id="952">952</span>
|
||||
<span id="953">953</span>
|
||||
<span id="954">954</span>
|
||||
<span id="955">955</span>
|
||||
</pre><pre class="rust"><code><span class="comment">/*
|
||||
* Copyright (C) 2022 Aravinth Manivannan <realaravinth@batsense.net>
|
||||
*
|
||||
@@ -959,10 +974,13 @@
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/</span>
|
||||
<span class="kw">use</span> <span class="ident">std::str::FromStr</span>;
|
||||
|
||||
<span class="kw">use</span> <span class="ident">db_core::dev</span>::<span class="kw-2">*</span>;
|
||||
|
||||
<span class="kw">use</span> <span class="ident">sqlx::postgres::PgPoolOptions</span>;
|
||||
<span class="kw">use</span> <span class="ident">sqlx::types::time::OffsetDateTime</span>;
|
||||
<span class="kw">use</span> <span class="ident">sqlx::ConnectOptions</span>;
|
||||
<span class="kw">use</span> <span class="ident">sqlx::PgPool</span>;
|
||||
|
||||
<span class="kw">pub</span> <span class="kw">mod</span> <span class="ident">errors</span>;
|
||||
@@ -987,6 +1005,7 @@
|
||||
|
||||
<span class="kw">pub</span> <span class="kw">struct</span> <span class="ident">Fresh</span> {
|
||||
<span class="kw">pub</span> <span class="ident">pool_options</span>: <span class="ident">PgPoolOptions</span>,
|
||||
<span class="kw">pub</span> <span class="ident">disable_logging</span>: <span class="ident">bool</span>,
|
||||
<span class="kw">pub</span> <span class="ident">url</span>: <span class="ident">String</span>,
|
||||
}
|
||||
|
||||
@@ -1008,11 +1027,22 @@
|
||||
<span class="kw">type</span> <span class="ident">Pool</span> <span class="op">=</span> <span class="ident">Database</span>;
|
||||
<span class="kw">async</span> <span class="kw">fn</span> <span class="ident">connect</span>(<span class="self">self</span>) -> <span class="ident">DBResult</span><span class="op"><</span><span class="ident"><span class="self">Self</span>::Pool</span><span class="op">></span> {
|
||||
<span class="kw">let</span> <span class="ident">pool</span> <span class="op">=</span> <span class="kw">match</span> <span class="self">self</span> {
|
||||
<span class="ident"><span class="self">Self</span>::Fresh</span>(<span class="ident">fresh</span>) => <span class="ident">fresh</span>
|
||||
.<span class="ident">pool_options</span>
|
||||
.<span class="ident">connect</span>(<span class="kw-2">&</span><span class="ident">fresh</span>.<span class="ident">url</span>)
|
||||
.<span class="kw">await</span>
|
||||
.<span class="ident">map_err</span>(<span class="op">|</span><span class="ident">e</span><span class="op">|</span> <span class="ident">DBError::DBError</span>(<span class="ident">Box::new</span>(<span class="ident">e</span>)))<span class="question-mark">?</span>,
|
||||
<span class="ident"><span class="self">Self</span>::Fresh</span>(<span class="ident">fresh</span>) => {
|
||||
<span class="kw">let</span> <span class="kw-2">mut</span> <span class="ident">connect_options</span> <span class="op">=</span>
|
||||
<span class="ident">sqlx::postgres::PgConnectOptions::from_str</span>(<span class="kw-2">&</span><span class="ident">fresh</span>.<span class="ident">url</span>).<span class="ident">unwrap</span>();
|
||||
<span class="kw">if</span> <span class="ident">fresh</span>.<span class="ident">disable_logging</span> {
|
||||
<span class="ident">connect_options</span>.<span class="ident">disable_statement_logging</span>();
|
||||
}
|
||||
<span class="ident">sqlx::postgres::PgConnectOptions::from_str</span>(<span class="kw-2">&</span><span class="ident">fresh</span>.<span class="ident">url</span>)
|
||||
.<span class="ident">unwrap</span>()
|
||||
.<span class="ident">disable_statement_logging</span>();
|
||||
<span class="ident">fresh</span>
|
||||
.<span class="ident">pool_options</span>
|
||||
.<span class="ident">connect_with</span>(<span class="ident">connect_options</span>)
|
||||
.<span class="kw">await</span>
|
||||
.<span class="ident">map_err</span>(<span class="op">|</span><span class="ident">e</span><span class="op">|</span> <span class="ident">DBError::DBError</span>(<span class="ident">Box::new</span>(<span class="ident">e</span>)))<span class="question-mark">?</span>
|
||||
}
|
||||
|
||||
<span class="ident"><span class="self">Self</span>::Existing</span>(<span class="ident">conn</span>) => <span class="ident">conn</span>.<span class="number">0</span>,
|
||||
};
|
||||
<span class="prelude-val">Ok</span>(<span class="ident">Database</span> { <span class="ident">pool</span> })
|
||||
|
||||
Reference in New Issue
Block a user