feat: update

Author: eastspire

frameworks/Rust/hyperlane/src/main.rs

@@ -1,3 +1,5 @@
+#[allow(warnings)]
+#[allow(dead_code)]
 pub(crate) mod r#const;
 pub(crate) mod db;
 pub(crate) mod lazy;
@@ -20,6 +22,7 @@ pub(crate) use hyperlane::{
 pub(crate) use lazy::*;
 pub(crate) use rand::{Rng, SeedableRng, rng, rngs::SmallRng};
 pub(crate) use request_middleware::*;
+#[allow(unused_imports)]
 pub(crate) use route::*;
 pub(crate) use server::*;
 pub(crate) use sqlx::{

frameworks/Rust/hyperlane/src/request_middleware.rs

@@ -1,13 +1,32 @@
 use crate::*;
 
 pub async fn request(ctx: Context) {
-    let _ = ctx
-        .set_response_header(CONNECTION, CONNECTION_KEEP_ALIVE)
-        .await
-        .set_response_header(CONTENT_TYPE, APPLICATION_JSON)
+    ctx.set_response_header(CONNECTION, CONNECTION_KEEP_ALIVE)
         .await
         .set_response_header(SERVER, HYPERLANE)
         .await
         .set_response_header(DATE, gmt())
         .await;
+    #[cfg(feature = "plaintext")]
+    {
+        set_response_header(CONTENT_TYPE, TEXT_PLAIN).await;
+    }
+    #[cfg(feature = "fortunes")]
+    {
+        ctx = ctx
+            .set_response_header(CONTENT_TYPE, content_type_charset(TEXT_HTML, UTF8))
+            .await;
+    }
+    #[cfg(any(
+        feature = "json",
+        feature = "db",
+        feature = "query",
+        feature = "update",
+        feature = "cached_query"
+    ))]
+    {
+        ctx = ctx
+            .set_response_header(CONTENT_TYPE, APPLICATION_JSON)
+            .await;
+    }
 }

frameworks/Rust/hyperlane/src/route.rs

@@ -10,11 +10,7 @@ pub async fn json(ctx: Context) {
 }
 
 pub async fn plaintext(ctx: Context) {
-    let _ = ctx
-        .set_response_header(CONTENT_TYPE, TEXT_PLAIN)
-        .await
-        .send_response(200, RESPONSEDATA_BIN)
-        .await;
+    let _ = ctx.send_response(200, RESPONSEDATA_BIN).await;
 }
 
 pub async fn db(ctx: Context) {
@@ -56,11 +52,7 @@ pub async fn fortunes(ctx: Context) {
     ));
     fortunes_list.sort_by(|it, next| it.message.cmp(&next.message));
     let res: String = FortunesTemplate::new(fortunes_list).to_string();
-    let _ = ctx
-        .set_response_header(CONTENT_TYPE, content_type_charset(TEXT_HTML, UTF8))
-        .await
-        .send_response(200, res)
-        .await;
+    let _ = ctx.send_response(200, res).await;
 }
 
 pub async fn updates(ctx: Context) {

frameworks/Rust/hyperlane/src/server.rs

@@ -24,17 +24,30 @@ async fn init_server() {
     server.http_line_buffer_size(256).await;
     server.websocket_buffer_size(256).await;
     server.request_middleware(request).await;
+    #[cfg(feature = "plaintext")]
     server.route("/plaintext", plaintext).await;
+    #[cfg(feature = "json")]
     server.route("/json", json).await;
+    #[cfg(feature = "cached_query")]
     server.route("/cached-quer", cached_queries).await;
+    #[cfg(feature = "db")]
     server.route("/db", db).await;
+    #[cfg(feature = "query")]
     server.route("/query", queries).await;
+    #[cfg(feature = "fortunes")]
     server.route("/fortunes", fortunes).await;
+    #[cfg(feature = "update")]
     server.route("/upda", updates).await;
     server.listen().await.unwrap();
 }
 
 async fn init() {
+    #[cfg(any(
+        feature = "db",
+        feature = "query",
+        feature = "update",
+        feature = "fortunes"
+    ))]
     init_db().await;
     init_server().await;
 }