@@ -121,14 +121,13 @@ async fn connect(url: Url) -> anyhow::Result<async_nats::Client> {
121
121
Ok ( client)
122
122
}
123
123
124
+ type Bucket = Arc < RwLock < HashMap < String , Bytes > > > ;
125
+
124
126
#[ derive( Clone , Debug , Default ) ]
125
- pub struct Server ( Arc < RwLock < HashMap < Bytes , Arc < RwLock < HashMap < String , Bytes > > > > > > ) ;
127
+ struct Server ( Arc < RwLock < HashMap < Bytes , Bucket > > > ) ;
126
128
127
129
impl Server {
128
- async fn bucket (
129
- & self ,
130
- bucket : impl AsRef < [ u8 ] > ,
131
- ) -> Result < Arc < RwLock < HashMap < String , Bytes > > > > {
130
+ async fn bucket ( & self , bucket : impl AsRef < [ u8 ] > ) -> Result < Bucket > {
132
131
debug ! ( "looking up bucket" ) ;
133
132
let store = self . 0 . read ( ) . await ;
134
133
store
@@ -139,7 +138,7 @@ impl Server {
139
138
}
140
139
141
140
impl bindings:: exports:: wasi:: keyvalue:: store:: HandlerBucket < Option < HeaderMap > > for Server {
142
- #[ instrument( level = "trace" , ret) ]
141
+ #[ instrument( level = "trace" , skip ( _cx ) , ret) ]
143
142
async fn get (
144
143
& self ,
145
144
_cx : Option < HeaderMap > ,
@@ -151,7 +150,7 @@ impl bindings::exports::wasi::keyvalue::store::HandlerBucket<Option<HeaderMap>>
151
150
Ok ( Ok ( bucket. get ( & key) . cloned ( ) ) )
152
151
}
153
152
154
- #[ instrument( level = "trace" , ret) ]
153
+ #[ instrument( level = "trace" , skip ( _cx ) , ret) ]
155
154
async fn set (
156
155
& self ,
157
156
_cx : Option < HeaderMap > ,
@@ -165,7 +164,7 @@ impl bindings::exports::wasi::keyvalue::store::HandlerBucket<Option<HeaderMap>>
165
164
Ok ( Ok ( ( ) ) )
166
165
}
167
166
168
- #[ instrument( level = "trace" , ret) ]
167
+ #[ instrument( level = "trace" , skip ( _cx ) , ret) ]
169
168
async fn delete (
170
169
& self ,
171
170
_cx : Option < HeaderMap > ,
@@ -178,7 +177,7 @@ impl bindings::exports::wasi::keyvalue::store::HandlerBucket<Option<HeaderMap>>
178
177
Ok ( Ok ( ( ) ) )
179
178
}
180
179
181
- #[ instrument( level = "trace" , ret) ]
180
+ #[ instrument( level = "trace" , skip ( _cx ) , ret) ]
182
181
async fn exists (
183
182
& self ,
184
183
_cx : Option < HeaderMap > ,
@@ -190,7 +189,7 @@ impl bindings::exports::wasi::keyvalue::store::HandlerBucket<Option<HeaderMap>>
190
189
Ok ( Ok ( bucket. contains_key ( & key) ) )
191
190
}
192
191
193
- #[ instrument( level = "trace" , ret) ]
192
+ #[ instrument( level = "trace" , skip ( _cx ) , ret) ]
194
193
async fn list_keys (
195
194
& self ,
196
195
_cx : Option < HeaderMap > ,
@@ -213,7 +212,7 @@ impl bindings::exports::wasi::keyvalue::store::HandlerBucket<Option<HeaderMap>>
213
212
214
213
impl bindings:: exports:: wasi:: keyvalue:: store:: Handler < Option < HeaderMap > > for Server {
215
214
// NOTE: Resource handle returned is just the `identifier` itself
216
- #[ instrument( level = "trace" , ret) ]
215
+ #[ instrument( level = "trace" , skip ( _cx ) , ret) ]
217
216
async fn open (
218
217
& self ,
219
218
_cx : Option < HeaderMap > ,
0 commit comments