provider.go

/*
Copyright SecureKey Technologies Inc. All Rights Reserved.

SPDX-License-Identifier: Apache-2.0
*/

package wrapper

import (
	"github.com/hyperledger/aries-framework-go/spi/storage"
	"go.mongodb.org/mongo-driver/mongo"
)

type provider interface {
	storage.Provider

	Ping() error
}

type mongoDBProvider interface {
	provider

	CreateCustomIndexes(storeName string, model ...mongo.IndexModel) error
}

// ProviderWrapper wrap aries provider.
type ProviderWrapper struct {
	p       provider
	dbType  string
	metrics metricsProvider
}

// NewProvider return new store provider wrapper.
func NewProvider(p provider, dbType string, metrics metricsProvider) *ProviderWrapper {
	return &ProviderWrapper{p: p, dbType: dbType, metrics: metrics}
}

// OpenStore open store.
func (prov *ProviderWrapper) OpenStore(name string) (storage.Store, error) {
	s, err := prov.p.OpenStore(name)
	if err != nil {
		return nil, err
	}

	return NewStore(s, prov.dbType, prov.metrics), nil
}

// SetStoreConfig set store config.
func (prov *ProviderWrapper) SetStoreConfig(name string, config storage.StoreConfiguration) error {
	return prov.p.SetStoreConfig(name, config)
}

// GetStoreConfig get store config.
func (prov *ProviderWrapper) GetStoreConfig(name string) (storage.StoreConfiguration, error) {
	return prov.p.GetStoreConfig(name)
}

// GetOpenStores get stores.
func (prov *ProviderWrapper) GetOpenStores() []storage.Store {
	return prov.p.GetOpenStores()
}

// Ping db.
func (prov *ProviderWrapper) Ping() error {
	return prov.p.Ping()
}

// Close provider.
func (prov *ProviderWrapper) Close() error {
	return prov.p.Close()
}

// MongoDBProviderWrapper wraps a MongoDB aries provider.
type MongoDBProviderWrapper struct {
	*ProviderWrapper

	mp mongoDBProvider
}

// NewMongoDBProvider return new store provider wrapper.
func NewMongoDBProvider(p mongoDBProvider, metrics metricsProvider) *MongoDBProviderWrapper {
	return &MongoDBProviderWrapper{
		ProviderWrapper: NewProvider(p, "MongoDB", metrics),
		mp:              p,
	}
}

// OpenStore open store.
func (prov *MongoDBProviderWrapper) OpenStore(name string) (storage.Store, error) {
	s, err := prov.p.OpenStore(name)
	if err != nil {
		return nil, err
	}

	return NewMongoDBStore(s, prov.metrics), nil
}

// CreateCustomIndexes creates MongoDB indexes.
func (prov *MongoDBProviderWrapper) CreateCustomIndexes(storeName string, model ...mongo.IndexModel) error {
	return prov.mp.CreateCustomIndexes(storeName, model...)
}