feat(sqlite): Trace sqlite host components

Signed-off-by: Caleb Schoepp <[email protected]>

Author: calebschoepp

Cargo.lock

@@ -12,4 +12,8 @@ anyhow = "1.0"
 rusqlite = { version = "0.29.0", features = [ "bundled" ] }
 rand = "0.8"
 once_cell = "1"
-tokio = "1"
+tokio = "1"
+tracing = { workspace = true }
+
+[lints]
+workspace = true

crates/sqlite-inproc/Cargo.toml

@@ -8,6 +8,7 @@ use async_trait::async_trait;
 use once_cell::sync::OnceCell;
 use spin_sqlite::Connection;
 use spin_world::v2::sqlite;
+use tracing::{instrument, Level};
 
 #[derive(Debug, Clone)]
 pub enum InProcDatabaseLocation {
@@ -46,6 +47,7 @@ impl InProcConnection {
 
 #[async_trait]
 impl Connection for InProcConnection {
+    #[instrument(name = "spin_sqlite_inproc.query", skip(self), err(level = Level::INFO), fields(otel.kind = "client", db.system = "sqlite", otel.name = query))]
     async fn query(
         &self,
         query: &str,
@@ -60,6 +62,7 @@ impl Connection for InProcConnection {
             .map_err(|e| sqlite::Error::Io(e.to_string()))?
     }
 
+    #[instrument(name = "spin_sqlite_inproc.execute_batch", skip(self), err(level = Level::INFO), fields(otel.kind = "client", db.system = "sqlite", db.statements = statements))]
     async fn execute_batch(&self, statements: &str) -> anyhow::Result<()> {
         let connection = self.db_connection()?;
         let statements = statements.to_owned();

crates/sqlite-inproc/src/lib.rs

@@ -15,3 +15,7 @@ spin-world = { path = "../world" }
 sqlparser = "0.34"
 libsql = { version = "0.3.2", features = ["remote"], default-features = false }
 tokio = { version = "1", features = ["full"] }
+tracing = { workspace = true }
+
+[lints]
+workspace = true

crates/sqlite-libsql/Cargo.toml

@@ -1,11 +1,13 @@
 use spin_world::v2::sqlite::{self, RowResult};
+use tracing::{instrument, Level};
 
 #[derive(Clone)]
 pub struct LibsqlClient {
     inner: libsql::Connection,
 }
 
 impl LibsqlClient {
+    #[instrument(name = "spin_sqlite_libsql.create_connection", skip(token), err(level = Level::INFO), fields(otel.kind = "client", db.system = "sqlite"))]
     pub async fn create(url: String, token: String) -> anyhow::Result<Self> {
         let db = libsql::Builder::new_remote(url, token).build().await?;
         let inner = db.connect()?;
@@ -15,6 +17,7 @@ impl LibsqlClient {
 
 #[async_trait::async_trait]
 impl spin_sqlite::Connection for LibsqlClient {
+    #[instrument(name = "spin_sqlite_libsql.query", skip(self), err(level = Level::INFO), fields(otel.kind = "client", db.system = "sqlite", otel.name = query))]
     async fn query(
         &self,
         query: &str,
@@ -34,6 +37,7 @@ impl spin_sqlite::Connection for LibsqlClient {
         })
     }
 
+    #[instrument(name = "spin_sqlite_libsql.execute_batch", skip(self), err(level = Level::INFO), fields(otel.kind = "client", db.system = "sqlite", db.statements = statements))]
     async fn execute_batch(&self, statements: &str) -> anyhow::Result<()> {
         self.inner.execute_batch(statements).await?;
 

crates/sqlite-libsql/src/lib.rs

@@ -12,3 +12,4 @@ spin-app = { path = "../app" }
 spin-world = { path = "../world" }
 table = { path = "../table" }
 tokio = "1"
+tracing = { workspace = true }