signoz/pkg/authz/openfgaauthz/sqlstore.go

package openfgaauthz

import (
	"github.com/SigNoz/signoz/pkg/errors"
	"github.com/SigNoz/signoz/pkg/sqlstore"
	"github.com/openfga/openfga/pkg/storage"
	"github.com/openfga/openfga/pkg/storage/postgres"
	"github.com/openfga/openfga/pkg/storage/sqlcommon"
)

func NewSQLStore(sqlstore sqlstore.SQLStore) (storage.OpenFGADatastore, error) {
	switch sqlstore.BunDB().Dialect().Name().String() {
	// use the NewWithDB for sqlite once https://github.com/openfga/openfga/pull/2679 is merged and released else will figure out something else.
	case "sqlite":
	case "pg":
		return postgres.NewWithDB(sqlstore.SQLDB(), nil, &sqlcommon.Config{
			MaxTuplesPerWriteField: 100,
			MaxTypesPerModelField:  100,
		})
	}
	return nil, errors.Newf(errors.TypeInvalidInput, errors.CodeInvalidInput, "invalid store type: %s", sqlstore.BunDB().Dialect().Name().String())
}