feat: Add Rockset as an OnlineStore

This commit adds Rockset as a contributed online store. It implements the Update, Teardown, Read and Write apis using the Rockset pythonn client. Signed-off-by: Daniel Lin <[email protected]>
feast-dev · Feb 4, 2023 · a799056 · a799056
1 parent ff66784
commit a799056
Show file tree

Hide file tree

Showing 18 changed files with 691 additions and 0 deletions.
diff --git a/docs/SUMMARY.md b/docs/SUMMARY.md
@@ -96,6 +96,7 @@
   * [PostgreSQL (contrib)](reference/online-stores/postgres.md)
   * [Cassandra + Astra DB (contrib)](reference/online-stores/cassandra.md) 
   * [MySQL (contrib)](reference/online-stores/mysql.md)
+  * [Rockset (contrib)](reference/online-stores/rockset.md)
 * [Providers](reference/providers/README.md)
   * [Local](reference/providers/local.md)
   * [Google Cloud Platform](reference/providers/google-cloud-platform.md)

diff --git a/docs/reference/online-stores/README.md b/docs/reference/online-stores/README.md
@@ -42,3 +42,7 @@ Please see [Online Store](../../getting-started/architecture-and-components/onli
 [mysql.md](mysql.md)
 {% endcontent-ref %}
 
+{% content-ref url="mysql.md" %}
+[rockset.md](rockset.md)
+{% endcontent-ref %}
+
diff --git a/docs/reference/online-stores/rockset.md b/docs/reference/online-stores/rockset.md
@@ -0,0 +1,38 @@
+# Rockset (contrib)
+
+## Description
+
+In Alpha Development.
+
+The [Rockset](https://rockset.com/demo-signup/) online store provides support for materializing feature values within a Rockset collection for serving online features in real-time.
+
+* Each document is uniquely identified by its '_id' value. Repeated inserts into the same document '_id' will result in an upsert.
+
+Rockset indexes all columns allowing for quick per feature look up and also allows for a dynamic typed schema that can change based on any new requirements. ApiKeys can be found in the console
+along with host urls which you can find in "View Region Endpoint Urls".
+
+Data Model Used Per Doc
+
+```
+{
+  "_id": (STRING) Unique Identifier for the feature document.
+  <key_name>: (STRING) Feature Values Mapped by Feature Name. Feature
+                       values stored as a serialized hex string.
+  ....
+  "event_ts": (STRING) ISO Stringified Timestamp.
+  "created_ts": (STRING) ISO Stringified Timestamp.
+}
+```
+
+
+## Example
+
+```yaml
+project: my_feature_app
+registry: data/registry.db
+provider: local
+online_stores
+    type: rockset
+    apikey: MY_APIKEY_HERE
+    host: api.usw2a1.rockset.com
+```
diff --git a/sdk/python/docs/source/feast.infra.online_stores.contrib.rockset_online_store.rst b/sdk/python/docs/source/feast.infra.online_stores.contrib.rockset_online_store.rst
@@ -0,0 +1,21 @@
+feast.infra.online\_stores.contrib.rockset\_online\_store package
+=================================================================
+
+Submodules
+----------
+
+feast.infra.online\_stores.contrib.rockset\_online\_store.rockset module
+------------------------------------------------------------------------
+
+.. automodule:: feast.infra.online_stores.contrib.rockset_online_store.rockset
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+Module contents
+---------------
+
+.. automodule:: feast.infra.online_stores.contrib.rockset_online_store
+   :members:
+   :undoc-members:
+   :show-inheritance:
diff --git a/sdk/python/docs/source/feast.infra.online_stores.contrib.rst b/sdk/python/docs/source/feast.infra.online_stores.contrib.rst
@@ -10,6 +10,7 @@ Subpackages
    feast.infra.online_stores.contrib.cassandra_online_store
    feast.infra.online_stores.contrib.hbase_online_store
    feast.infra.online_stores.contrib.mysql_online_store
+   feast.infra.online_stores.contrib.rockset_online_store
 
 Submodules
 ----------
@@ -54,6 +55,14 @@ feast.infra.online\_stores.contrib.postgres\_repo\_configuration module
    :undoc-members:
    :show-inheritance:
 
+feast.infra.online\_stores.contrib.rockset\_repo\_configuration module
+----------------------------------------------------------------------
+
+.. automodule:: feast.infra.online_stores.contrib.rockset_repo_configuration
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
 Module contents
 ---------------
 

diff --git a/sdk/python/feast/infra/online_stores/contrib/rockset_online_store/__init__.py b/sdk/python/feast/infra/online_stores/contrib/rockset_online_store/__init__.py