don't import all drivers by default, too much baggage

rusenask · Apr 12, 2020 · c748510 · c748510
1 parent 8bef40f
commit c748510
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/watcher.go b/watcher.go
@@ -11,15 +11,6 @@ import (
 
 	"github.com/casbin/casbin/persist"
 	"gocloud.dev/pubsub"
-
-	// Import the pubsub driver packages we want to be able to open.
-	_ "gocloud.dev/pubsub/awssnssqs"
-	_ "gocloud.dev/pubsub/azuresb"
-	_ "gocloud.dev/pubsub/gcppubsub"
-	_ "gocloud.dev/pubsub/kafkapubsub"
-	_ "gocloud.dev/pubsub/mempubsub"
-	_ "gocloud.dev/pubsub/natspubsub"
-	_ "gocloud.dev/pubsub/rabbitpubsub"
 )
 
 // check interface compatibility

diff --git a/watcher_test.go b/watcher_test.go
@@ -10,6 +10,10 @@ import (
 	"github.com/casbin/casbin"
 	gnatsd "github.com/nats-io/nats-server/v2/test"
 	"github.com/nats-io/nats.go"
+
+	// Enable inmemory and NATS drivers
+	_ "github.com/rusenask/casbin-go-cloud-watcher/drivers/mempubsub"
+	_ "github.com/rusenask/casbin-go-cloud-watcher/drivers/natspubsub"
 )
 
 func TestNATSWatcher(t *testing.T) {