Browse Source

0.1.8

master 0.1.8
Ryan 1 year ago
parent
commit
cababc43db
  1. 6
      RELEASE_NOTE.md
  2. 23
      src/aws/s3.rs
  3. 11
      src/database.rs

6
RELEASE_NOTE.md

@ -32,4 +32,8 @@ first deploy
0.1.7
======
- added s3 bucket constant
- fix some constant name(underline -> dash)
- fix some constant name(underline -> dash)
0.1.8
======
- added campaign database information

23
src/aws/s3.rs

@ -58,27 +58,4 @@ pub async fn read_file(bucket: &str, key: &str) -> Result<String, DeviceError> {
}
Ok(String::from_utf8_lossy(&data).into_owned())
}
#[cfg(test)]
mod tests {
use super::*;
use serial_test::serial;
use crate::constants;
fn setup() {
crate::env::set(constants::ENV_AWS_ACCESS_KEY, "AKIA32XY6L3KQA4QDIAP");
crate::env::set(constants::ENV_AWS_SECRET_KEY, "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");
crate::env::set(constants::ENV_AWS_REGION, "ap-northeast-2");
}
fn teardown() {}
#[tokio::test]
#[serial]
async fn test_read_file() {
setup();
assert!(read_file("apm-bigdata", "flower-device/test.csv").await.is_ok());
teardown();
}
}

11
src/database.rs

@ -2,6 +2,7 @@ use std::sync::Arc;
use deadpool_postgres::{Pool, Config, Runtime};
use tokio::sync::OnceCell;
use tokio_postgres::NoTls;
use crate::aws::s3::read_file;
use crate::config::get_config;
static CONNECTION_POOL: OnceCell<Arc<Pool>> = OnceCell::const_new();
@ -32,11 +33,11 @@ async fn initialize_campaign_connection_pool() -> Arc<Pool> {
let config = get_config().await;
let mut pg = Config::new();
pg.host = Some(config.database.host.clone());
pg.user = Some(config.database.user.clone());
pg.password = Some(config.database.password.clone());
pg.dbname = Some(config.database.name.clone());
pg.port = Some(config.database.port as u16);
pg.host = Some(config.campaign_database.host.clone());
pg.user = Some(config.campaign_database.user.clone());
pg.password = Some(config.campaign_database.password.clone());
pg.dbname = Some(config.campaign_database.name.clone());
pg.port = Some(config.campaign_database.port as u16);
let pool = pg.create_pool(Some(Runtime::Tokio1), NoTls).unwrap();

Loading…
Cancel
Save