style: rearrange kuzu.rs into sections

Author: badmonster0

src/ops/storages/kuzu.rs

@@ -12,21 +12,36 @@ use crate::prelude::*;
 use crate::setup::{ResourceSetupStatus, SetupChangeType};
 use crate::{ops::sdk::*, setup::CombinedState};
 
+const SELF_CONTAINED_TAG_FIELD_NAME: &str = "__self_contained";
+
+////////////////////////////////////////////////////////////
+// Public Types
+////////////////////////////////////////////////////////////
+
 #[derive(Debug, Deserialize, Clone)]
 pub struct ConnectionSpec {
     /// The URL of the [Kuzu API server](https://kuzu.com/docs/api/server/overview),
     /// e.g. `http://localhost:8000`.
     api_server_url: String,
 }
 
-type KuzuGraphElement = GraphElementType<ConnectionSpec>;
-
 #[derive(Debug, Deserialize)]
 pub struct Spec {
     connection: spec::AuthEntryReference<ConnectionSpec>,
     mapping: GraphElementMapping,
 }
 
+#[derive(Debug, Deserialize)]
+pub struct Declaration {
+    connection: spec::AuthEntryReference<ConnectionSpec>,
+    #[serde(flatten)]
+    decl: GraphDeclaration,
+}
+
+////////////////////////////////////////////////////////////
+// Utils to deal with Kuzu
+////////////////////////////////////////////////////////////
+
 struct CypherBuilder {
     query: String,
     params: Vec<(String, serde_json::Value)>,
@@ -49,13 +64,6 @@ impl CypherBuilder {
     }
 }
 
-#[derive(Debug, Deserialize)]
-pub struct Declaration {
-    connection: spec::AuthEntryReference<ConnectionSpec>,
-    #[serde(flatten)]
-    decl: GraphDeclaration,
-}
-
 struct KuzuThinClient {
     reqwest_client: reqwest::Client,
     query_url: String,
@@ -94,8 +102,6 @@ impl KuzuThinClient {
     }
 }
 
-const SELF_CONTAINED_TAG_FIELD_NAME: &str = "__self_contained";
-
 fn kuzu_table_type(elem_type: &ElementType) -> &'static str {
     match elem_type {
         ElementType::Node(_) => "NODE",
@@ -152,6 +158,10 @@ fn value_type_to_kuzu(value_type: &ValueType) -> Result<String> {
     })
 }
 
+////////////////////////////////////////////////////////////
+// Setup
+////////////////////////////////////////////////////////////
+
 #[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq)]
 struct ReferencedNodeTable {
     table_name: String,
@@ -289,6 +299,12 @@ impl GraphElementDataSetupStatus {
     }
 }
 
+////////////////////////////////////////////////////////////
+// Factory implementation
+////////////////////////////////////////////////////////////
+
+type KuzuGraphElement = GraphElementType<ConnectionSpec>;
+
 pub struct Factory {
     reqwest_client: reqwest::Client,
 }