sentry - Connect to Postgres for Application

Author: elpiel

sentry/src/db.rs

@@ -1,14 +1,28 @@
 use redis::aio::MultiplexedConnection;
 use redis::RedisError;
 
+use bb8::Pool;
+use bb8_postgres::{
+    tokio_postgres::{Error as PostgresError, NoTls},
+    PostgresConnectionManager,
+};
 use lazy_static::lazy_static;
 
 lazy_static! {
     static ref REDIS_URL: String =
         std::env::var("REDIS_URL").unwrap_or_else(|_| String::from("redis://127.0.0.1:6379"));
+    static ref POSTGRES_URL: String = std::env::var("POSTGRES_URL")
+        .unwrap_or_else(|_| String::from("postgresql://postgres:postgres@localhost:5432"));
 }
 
 pub async fn redis_connection() -> Result<MultiplexedConnection, RedisError> {
     let client = redis::Client::open(REDIS_URL.as_str()).expect("Wrong redis connection string");
     client.get_multiplexed_tokio_connection().await
 }
+
+pub async fn postgres_connection() -> Result<Pool<PostgresConnectionManager<NoTls>>, PostgresError>
+{
+    let pg_mgr = PostgresConnectionManager::new_from_stringlike(POSTGRES_URL.as_str(), NoTls)?;
+
+    Pool::builder().build(pg_mgr).await
+}

sentry/src/lib.rs

@@ -3,6 +3,8 @@
 
 use crate::middleware::auth;
 use crate::middleware::cors::{cors, Cors};
+use bb8::Pool;
+use bb8_postgres::{tokio_postgres::NoTls, PostgresConnectionManager};
 use hyper::service::{make_service_fn, service_fn};
 use hyper::{Body, Error, Method, Request, Response, Server, StatusCode};
 use primitives::adapter::Adapter;
@@ -41,6 +43,7 @@ pub struct Application<A: Adapter> {
     adapter: A,
     logger: Logger,
     redis: MultiplexedConnection,
+    _postgres: Pool<PostgresConnectionManager<NoTls>>,
     _clustered: bool,
     port: u16,
     config: Config,
@@ -52,6 +55,7 @@ impl<A: Adapter + 'static> Application<A> {
         config: Config,
         logger: Logger,
         redis: MultiplexedConnection,
+        postgres: Pool<PostgresConnectionManager<NoTls>>,
         clustered: bool,
         port: u16,
     ) -> Self {
@@ -60,6 +64,7 @@ impl<A: Adapter + 'static> Application<A> {
             config,
             logger,
             redis,
+            _postgres: postgres,
             _clustered: clustered,
             port,
         }

sentry/src/main.rs

@@ -9,7 +9,7 @@ use primitives::config::configuration;
 use primitives::util::logging::{Async, PrefixedCompactFormat, TermDecorator};
 use primitives::util::tests::prep_db::{AUTH, IDS};
 use primitives::ValidatorId;
-use sentry::db::redis_connection;
+use sentry::db::{postgres_connection, redis_connection};
 use sentry::Application;
 use slog::{o, Drain, Logger};
 use std::convert::TryFrom;
@@ -98,15 +98,16 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     let logger = logger();
     let redis = redis_connection().await?;
+    let postgres = postgres_connection().await?;
 
     match adapter {
         AdapterTypes::EthereumAdapter(adapter) => {
-            Application::new(*adapter, config, logger, redis, clustered, port)
+            Application::new(*adapter, config, logger, redis, postgres, clustered, port)
                 .run()
                 .await
         }
         AdapterTypes::DummyAdapter(adapter) => {
-            Application::new(*adapter, config, logger, redis, clustered, port)
+            Application::new(*adapter, config, logger, redis, postgres, clustered, port)
                 .run()
                 .await
         }