blob: cea15c03b111edfe829afe9a9e6521709dc5cb75 (
plain) (
blame)
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
|
/*
* =====================================================================================
*
* Filename: db.cpp
*
* Description: Connection to a postgres database
*
* Version: 1.0
* Created: 04/06/2023 11:40:39 AM
* Revision: none
* Compiler: gcc
*
* Author: Cara Salter (muirrum), cara@devcara.com
* Organization: Worcester Polytechnic Institute
*
* =====================================================================================
*/
#include <fmt/core.h>
#include <mutex>
#include <stdlib.h>
#include <stdio.h>
#include <iostream>
#include <142bot/db.hpp>
#include <pqxx/pqxx>
#include <fmt/format.h>
#include <cstdarg>
#include <sentry.h>
using namespace std;
namespace db {
std::mutex db_mutex;
std::string _error;
/**
* Connects to a postgres database, returns false if error
**/
pqxx::connection connect(const std::string &host, const std::string &user, const std::string &pass, const std::string &db, int port) {
std::lock_guard<std::mutex> db_lock(db_mutex);
std::string cn_s = "postgresql://";
if (!user.empty()) {
cn_s = cn_s + user;
}
if (!pass.empty() && !user.empty()) {
cn_s = cn_s + ":" + pass;
}
if ((!user.empty() || !pass.empty())) {
cn_s = cn_s + "@";
}
if (!host.empty()) {
cn_s = cn_s + "localhost";
}
if (port != 0 && !host.empty()) {
cn_s = cn_s + ":" + std::to_string(port);
}
if (!db.empty()) {
cn_s = cn_s + "/" + db;
}
sentry_value_t crumb = sentry_value_new_breadcrumb("default", "Started Database Connection");
sentry_value_set_by_key(crumb, "level", sentry_value_new_string("db"));
sentry_value_set_by_key(crumb, "data", sentry_value_new_string(cn_s.c_str()));
sentry_add_breadcrumb(crumb);
try {
pqxx::connection c{cn_s};
return c;
} catch (std::exception const &e) {
_error = e.what();
throw e;
}
}
const std::string& error() {
return _error;
}
}
|
