Merge pull request #38 from RelationalAI/ag-memory-usage

Expose memory usage metrics from object_store_ffi

Author: andrebsguedes

Project.toml

@@ -1,6 +1,6 @@
 name = "RustyObjectStore"
 uuid = "1b5eed3d-1f46-4baa-87f3-a4a892b23610"
-version = "0.6.1"
+version = "0.6.5"
 
 [deps]
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
@@ -16,7 +16,7 @@ ReTestItems = "1"
 Sockets = "1"
 Test = "1"
 julia = "1.8"
-object_store_ffi_jll = "0.6.0"
+object_store_ffi_jll = "0.6.5"
 
 [extras]
 CloudBase = "85eb1798-d7c4-4918-bb13-c944d38e27ed"

src/RustyObjectStore.jl

@@ -5,6 +5,7 @@ export StaticConfig, ClientOptions, Config, AzureConfig, AWSConfig
 export status_code, is_connection, is_timeout, is_early_eof, is_unknown, is_parse_url
 export get_object_stream, ReadStream, finish!
 export put_object_stream, WriteStream, cancel!, shutdown!
+export current_metrics
 
 using Base.Libc.Libdl: dlext
 using Base: @kwdef, @lock
@@ -1325,4 +1326,12 @@ function _unsafe_write(stream::WriteStream, input::Ptr{UInt8}, nbytes::Int; flus
     return Int(response.length)
 end
 
+struct Metrics
+    live_bytes::Int64
+end
+
+function current_metrics()
+    return @ccall rust_lib.current_metrics()::Metrics
+end
+
 end # module