-
-
Notifications
You must be signed in to change notification settings - Fork 26
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add transactions-by-account service in rust
- Loading branch information
1 parent
412f504
commit 92e57bb
Showing
6 changed files
with
154 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,17 +1,39 @@ | ||
FROM mxfactorial/go-base:v1 as builder | ||
FROM rust:latest as builder | ||
|
||
COPY . . | ||
WORKDIR /app | ||
|
||
WORKDIR /app/services/transactions-by-account | ||
COPY . ./ | ||
|
||
RUN go build -o transactions-by-account ./cmd | ||
RUN rustup target add x86_64-unknown-linux-musl | ||
RUN apt update && \ | ||
apt install -y musl-tools perl make | ||
RUN update-ca-certificates | ||
|
||
FROM golang:alpine | ||
ENV USER=transactions-by-account | ||
ENV UID=10007 | ||
|
||
WORKDIR /app | ||
RUN adduser \ | ||
--disabled-password \ | ||
--gecos "" \ | ||
--home "/nonexistent" \ | ||
--shell "/sbin/nologin" \ | ||
--no-create-home \ | ||
--uid "${UID}" \ | ||
"${USER}" | ||
|
||
RUN USER=root cargo build \ | ||
--manifest-path=services/transactions-by-account/Cargo.toml \ | ||
--target x86_64-unknown-linux-musl \ | ||
--release | ||
|
||
FROM alpine | ||
|
||
COPY --from=builder /etc/passwd /etc/passwd | ||
COPY --from=builder /etc/group /etc/group | ||
COPY --from=builder /app/target/x86_64-unknown-linux-musl/release/transactions-by-account /usr/local/bin | ||
|
||
COPY --from=builder /app/services/transactions-by-account/transactions-by-account . | ||
EXPOSE 10007 | ||
|
||
EXPOSE 10008 | ||
USER transactions-by-account:transactions-by-account | ||
|
||
CMD ["/app/transactions-by-account"] | ||
CMD [ "/usr/local/bin/transactions-by-account" ] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,18 @@ | ||
[package] | ||
name = "transactions-by-account" | ||
version = "0.1.0" | ||
edition = "2021" | ||
rust-version.workspace = true | ||
|
||
[dependencies] | ||
axum = "0.7.4" | ||
tracing = "0.1.40" | ||
tracing-subscriber = { version = "0.3.18", features = ["env-filter"] } | ||
tokio = { version = "1.35.1", features = ["macros", "rt-multi-thread"] } | ||
shutdown = { path = "../../crates/shutdown" } | ||
pg = { path = "../../crates/pg" } | ||
service = { path = "../../crates/service" } | ||
types = { path = "../../crates/types" } | ||
|
||
[target.x86_64-unknown-linux-musl.dependencies] | ||
openssl = { version = "0.10", features = ["vendored"] } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,88 @@ | ||
use axum::{ | ||
extract::{Json, State}, | ||
http::StatusCode, | ||
routing::{get, post}, | ||
Router, | ||
}; | ||
use pg::postgres::{ConnectionPool, DB}; | ||
use service::Service; | ||
use shutdown::shutdown_signal; | ||
use std::{env, net::ToSocketAddrs}; | ||
use tokio::net::TcpListener; | ||
use types::request_response::{IntraTransactions, QueryByAccount}; | ||
|
||
// used by lambda to test for service availability | ||
const READINESS_CHECK_PATH: &str = "READINESS_CHECK_PATH"; | ||
|
||
async fn handle_event( | ||
State(pool): State<ConnectionPool>, | ||
event: Json<QueryByAccount>, | ||
) -> Result<axum::Json<IntraTransactions>, StatusCode> { | ||
let client_request = event.0; | ||
|
||
let svc = Service::new(pool.get_conn().await); | ||
|
||
let account = client_request.account_name; | ||
|
||
let record_limit = env::var("RETURN_RECORD_LIMIT") | ||
.unwrap_or_else(|_| panic!("RETURN_RECORD_LIMIT variable assignment")) | ||
.parse::<i64>() | ||
.unwrap(); | ||
|
||
let transactions = svc | ||
.get_last_n_transactions(account.clone(), record_limit) | ||
.await | ||
.map_err(|e| { | ||
tracing::error!("error: {}", e); | ||
StatusCode::INTERNAL_SERVER_ERROR | ||
})?; | ||
|
||
if transactions.is_empty() { | ||
tracing::error!("transaction requests not found"); | ||
} | ||
|
||
let response = IntraTransactions::new(account, transactions); | ||
|
||
Ok(axum::Json(response)) | ||
} | ||
|
||
#[tokio::main] | ||
async fn main() { | ||
tracing_subscriber::fmt::init(); | ||
|
||
let readiness_check_path = env::var(READINESS_CHECK_PATH) | ||
.unwrap_or_else(|_| panic!("{READINESS_CHECK_PATH} variable assignment")); | ||
|
||
let conn_uri = DB::create_conn_uri_from_env_vars(); | ||
|
||
let pool = DB::new_pool(&conn_uri).await; | ||
|
||
let app = Router::new() | ||
.route("/", post(handle_event)) | ||
.route( | ||
readiness_check_path.as_str(), | ||
get(|| async { StatusCode::OK }), | ||
) | ||
.with_state(pool); | ||
|
||
let hostname_or_ip = env::var("HOSTNAME_OR_IP").unwrap_or("0.0.0.0".to_string()); | ||
|
||
let port = env::var("TRANSACTIONS_BY_ACCOUNT_PORT").unwrap_or("10006".to_string()); | ||
|
||
let serve_addr = format!("{hostname_or_ip}:{port}"); | ||
|
||
let mut addrs_iter = serve_addr.to_socket_addrs().unwrap_or( | ||
format!("{hostname_or_ip}:{port}") | ||
.to_socket_addrs() | ||
.unwrap(), | ||
); | ||
|
||
let addr = addrs_iter.next().unwrap(); | ||
|
||
tracing::info!("listening on {}", addr); | ||
|
||
axum::serve(TcpListener::bind(addr).await.unwrap(), app) | ||
.with_graceful_shutdown(shutdown_signal()) | ||
.await | ||
.unwrap(); | ||
} |