Merge branch 'master' of https://github.com/spacycoder/cosmos-casbin-adapter

spacycoder · spacycoder · commit f6394ab25222 · 2021-10-28T17:07:48.000+02:00
diff --git a/adapter.go b/adapter.go
@@ -54,7 +54,7 @@ func NewAdapter(connectionString string, options ...Option) persist.Adapter {
 		option(a)
 	}
 
-	db := client.Database(a.databaseName)
+	db := a.client.Database(a.databaseName)
 	a.createDatabaseIfNotExist(db)
 	a.db = db
 
@@ -377,3 +377,9 @@ func Collection(coll string) Option {
 		a.collectionName = coll
 	}
 }
+
+func CosmosClient(client *cosmos.Client) Option {
+	return func(a *adapter) {
+		a.client = client
+	}
+}

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,7 @@ func NewAdapter(connectionString string, options ...Option) persist.Adapter {`
`54`	`54`	`option(a)`
`55`	`55`	`}`
`56`	`56`
`57`		`- db := client.Database(a.databaseName)`
	`57`	`+ db := a.client.Database(a.databaseName)`
`58`	`58`	`a.createDatabaseIfNotExist(db)`
`59`	`59`	`a.db = db`
`60`	`60`
`@@ -377,3 +377,9 @@ func Collection(coll string) Option {`
`377`	`377`	`a.collectionName = coll`
`378`	`378`	`}`
`379`	`379`	`}`
	`380`	`+`
	`381`	`+func CosmosClient(client *cosmos.Client) Option {`
	`382`	`+ return func(a *adapter) {`
	`383`	`+ a.client = client`
	`384`	`+ }`
	`385`	`+}`