normalize arkvx

src/main.rs

@@ -86,28 +86,20 @@ async fn main() {
         .layer(CompressionLayer::new().zstd(true))
         .route("/api.json", get(serve_api))
         .layer(CompressionLayer::new().zstd(true))
-        .api_route(
-            "/arkvc_holdings",
-            get_with(routes::arkvc_holdings, |mut o| {
-                o = o.id("ARKVC Holdings");
-                description_date(o)
-            }),
-        )
-        .layer(
-            rate_limit_global(500)
-                .layer(rate_limit_ip())
-                .layer(cors())
-                .layer(CompressionLayer::new().zstd(true)),
-        )
         .api_route(
             "/ark_holdings",
             get_with(routes::ark_holdings, |mut o| {
-                o = o.id("ARK* ETF Holdings");
+                // test description
+                o = o.id("ARK* ETF Holdings").description(r"
+                | date       ┆ ticker     ┆ cusip      ┆ company      ┆ market_value ┆ shares  ┆ share_price ┆ weight │
+                │ ---        ┆ ---        ┆ ---        ┆ ---          ┆ ---          ┆ ---     ┆ ---         ┆ ---    │
+                │ date       ┆ str        ┆ str        ┆ str          ┆ i64          ┆ f64     ┆ f64         ┆ f64    │
+                ");
                 description_date(o)
             }),
         )
         .layer(
-            rate_limit_global(200)
+            rate_limit_global(1000)
                 .layer(rate_limit_ip())
                 .layer(cors())
                 .layer(CompressionLayer::new().zstd(true)),

src/routes.rs

@@ -3,16 +3,6 @@ use axum::extract::Query;
 
 mod polars_utils;
 
-pub async fn arkvc_holdings(date_range: Query<polars_utils::DateRange>) -> impl IntoApiResponse {
-    let df = polars_utils::get_parquet("ARKVC".to_owned()).await.unwrap();
-
-    let filter_df = polars_utils::filter_date_range(df, date_range)
-        .await
-        .unwrap();
-
-    axum::Json(polars_utils::to_json(filter_df).await.unwrap())
-}
-
 pub async fn ark_holdings(
     ticker: Query<polars_utils::Ticker>,
     date_range: Query<polars_utils::DateRange>,

src/routes/polars_utils.rs

@@ -56,13 +56,19 @@ pub async fn filter_date_range(
 }
 
 #[derive(Serialize, Deserialize, JsonSchema, strum_macros::Display)]
-pub enum TypeTicker {
+pub enum Ticker {
+    ARKVX,
     ARKF,
     ARKG,
     ARKK,
     ARKQ,
     ARKW,
     ARKX,
+    // ARKA,
+    // ARKZ,
+    ARKC,
+    ARKD,
+    ARKY,
 }
 
 #[derive(Serialize, Deserialize, JsonSchema)]