Compare commits

...

8 Commits

10 changed files with 635 additions and 400 deletions

View File

@ -1,8 +1,6 @@
[registries.lugit]
index = "sparse+http://lugit.local/api/packages/lucodear/cargo/"
index = "sparse+https://git.lucode.dev/api/packages/lucas/cargo/"
[registries.lugit-sa]
index = "sparse+http://lugit.local/api/packages/singleaction/cargo/"
[registry]
global-credential-providers = ["cargo:token"]

1004
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@ -1,6 +1,6 @@
[package]
name = "rustler-core"
version = "0.3.10"
version = "0.4.1"
edition = "2021"
description = "🐎 » rustler-core market data extractor core functionality"
authors = ["Lucas Colombo <lucas@lucode.ar>"]
@ -45,7 +45,7 @@ redis = { version = "0.29.2", features = ["tokio-comp"] }
serde = { version = "1.0.219", features = ["derive"] }
serde_json = "1.0.140"
uuid = { version = "1.16.0", features = ["v4", "fast-rng"] }
lool = { version = "^0.3.2", registry = "lugit", features = [
lool = { version = "^0.9.0", registry = "lugit", features = [
"cli.stylize",
"logger",
"sched.tokio",

View File

@ -16,6 +16,7 @@ impl MigrationTrait for Migration {
.col(ColumnDef::new(Ticker::Symbol).string().not_null())
.col(ColumnDef::new(Ticker::QuoteSymbol).string().null())
.col(ColumnDef::new(Ticker::MarketId).string().not_null())
.col(ColumnDef::new(Ticker::Active).boolean().not_null().default(true))
.foreign_key(
ForeignKey::create()
.name("fk_ticker_market_id")
@ -51,4 +52,6 @@ enum Ticker {
QuoteSymbol,
/// Market ID
MarketId,
/// Active status of the ticker. This defines if quotes are generated for this ticker or not.
Active,
}

View File

@ -35,21 +35,20 @@ pub mod db {
},
};
const RUSTLER_DATABASE: &str = "RUSTLER_DATABASE";
const DATABASE_URL: &str = "DATABASE_URL";
fn get_default_conn_str() -> String {
let conn_str = s!("sqlite://rustler.db?mode=rwc");
info!(
"No `{}` env var found, using default: {}",
RUSTLER_DATABASE.italic(),
DATABASE_URL.italic(),
conn_str.green()
);
conn_str
}
pub async fn get_connection() -> Result<DatabaseConnection> {
let db_conn_str =
std::env::var(RUSTLER_DATABASE).unwrap_or_else(|_| get_default_conn_str());
let db_conn_str = std::env::var(DATABASE_URL).unwrap_or_else(|_| get_default_conn_str());
let mut conn_opts = ConnectOptions::new(db_conn_str.to_owned());
conn_opts.sqlx_logging(false);

View File

@ -11,6 +11,7 @@ pub struct Model {
pub symbol: String,
pub quote_symbol: Option<String>,
pub market_id: String,
pub active: bool,
}
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]

View File

@ -24,6 +24,13 @@ impl Service {
Ok(tickers)
}
/// 🐎 » get all active
pub async fn get_all_active(&self) -> Result<Vec<TickerModel>, DbErr> {
let tickers =
Ticker::find().filter(ticker::Column::Active.eq(true)).all(&self.conn).await?;
Ok(tickers)
}
/// 🐎 » retrieves a ticker from the database, given its id
pub async fn get(&self, id: String) -> Result<Option<TickerModel>, DbErr> {
let ticker = Ticker::find_by_id(id).one(&self.conn).await?;

View File

@ -23,6 +23,7 @@ message Ticker {
string symbol = 2;
optional string quote_symbol = 3;
string market_id = 4;
bool active = 5;
}
message Tickers {

View File

@ -22,6 +22,7 @@ impl Ticker {
symbol: self.symbol,
quote_symbol: self.quote_symbol,
market_id: self.market_id,
active: self.active,
}
}
@ -32,6 +33,7 @@ impl Ticker {
symbol: model.symbol,
quote_symbol: model.quote_symbol,
market_id: model.market_id,
active: model.active,
}
}
}

View File

@ -1,7 +1,7 @@
allow-branch = ["master"]
sign-commit = true
sign-tag = true
registry = "lugit-sa"
registry = "lugit"
pre-release-commit-message = "release: 🔖 v{{version}}"
tag-message = "release: 🔖 v{{version}}"
tag-prefix = ""