Add method to retrieve DataStore from ProtoStore (#58)

Added `GetDataStore` method to `ProtoStore`
vmware-labs · Aug 21, 2023 · 6bf6d1e · 6bf6d1e
1 parent 5d1b86a
commit 6bf6d1e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/protostore/protostore.go b/pkg/protostore/protostore.go
@@ -63,6 +63,7 @@ type ProtoStore interface {
 	MsgToFilter(ctx context.Context, id string, msg proto.Message) (pMsg *ProtoStoreMsg, err error)
 	MsgToPersist(ctx context.Context, id string, msg proto.Message, md Metadata) (pMsg *ProtoStoreMsg, err error)
 
+	GetDataStore() datastore.DataStore
 	GetAuthorizer() authorizer.Authorizer
 	DropTables(msgs ...proto.Message) error
 }
@@ -145,6 +146,10 @@ func GetProtoStore(logger *logrus.Entry, ds datastore.DataStore) ProtoStore {
 	return p
 }
 
+func (p ProtobufDataStore) GetDataStore() datastore.DataStore {
+	return p.ds
+}
+
 func (p ProtobufDataStore) GetAuthorizer() authorizer.Authorizer {
 	return p.ds.GetAuthorizer()
 }