Expose SharedData keys & remove methods in Context

Author: jamesmulcahy

include/proxy-wasm/context.h

@@ -353,6 +353,9 @@ class ContextBase : public RootInterface,
   WasmResult getSharedData(std::string_view key,
                            std::pair<std::string, uint32_t /* cas */> *data) override;
   WasmResult setSharedData(std::string_view key, std::string_view value, uint32_t cas) override;
+  WasmResult getSharedDataKeys(std::string_view key_prefix, std::vector<std::string> *result) override;
+  WasmResult removeSharedDataKey(std::string_view key, uint32_t cas) override;
+
 
   // Shared Queue
   WasmResult registerSharedQueue(std::string_view queue_name,

include/proxy-wasm/context_interface.h

@@ -621,6 +621,21 @@ struct SharedDataInterface {
    * @param data is a location to store the returned value.
    */
   virtual WasmResult setSharedData(std::string_view key, std::string_view value, uint32_t cas) = 0;
+
+  /**
+   * Return all the keys which match the given key_prefix
+   * @param key_prefix is used to restrict the results to keys matching the given prefix
+   * @param data is a location to store the returned value.
+   */
+  virtual WasmResult getSharedDataKeys(std::string_view key_prefix, std::vector<std::string> *result) = 0;
+
+  /**
+   * Removes the given key from the data shared between VMs.
+   * @param key is a proxy-wide key mapping to the shared data value.
+   * @param cas is a compare-and-swap value. If it is zero it is ignored, otherwise it must match
+   * the cas associated with the value.
+   */
+  virtual WasmResult removeSharedDataKey(std::string_view key, uint32_t cas) = 0;
 }; // namespace proxy_wasm
 
 struct SharedQueueInterface {

src/context.cc

@@ -192,6 +192,16 @@ WasmResult ContextBase::setSharedData(std::string_view key, std::string_view val
   return getGlobalSharedData().set(wasm_->vm_id(), key, value, cas);
 }
 
+WasmResult ContextBase::getSharedDataKeys(std::string_view key_prefix,
+                                          std::vector<std::string> *result) {
+  return getGlobalSharedData().keys(wasm_->vm_id(), key_prefix, result);
+}
+
+WasmResult ContextBase::removeSharedDataKey(std::string_view key, uint32_t cas) {
+  return getGlobalSharedData().remove(wasm_->vm_id(), key, cas);
+}
+
+
 // Shared Queue
 
 WasmResult ContextBase::registerSharedQueue(std::string_view queue_name,