X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=govpp.go;h=d66d5dc564f68aa04a4258934f2951e57d0abaa9;hb=e6cca4bfd215aba51d58bf0bb046c16c081e42a3;hp=ff45b78fe90d9e063cb7fa82c52fa0fe73a9de1f;hpb=3da2b73ec5ce77abf796ebeae6dfc149c8d3c698;p=govpp.git

diff --git a/govpp.go b/govpp.go
index ff45b78..d66d5dc 100644
--- a/govpp.go
+++ b/govpp.go
@@ -18,18 +18,31 @@ import (
 	"git.fd.io/govpp.git/adapter"
 	"git.fd.io/govpp.git/adapter/vppapiclient"
 	"git.fd.io/govpp.git/core"
+	"time"
 )
 
-var vppAdapter adapter.VppAdapter // VPP Adapter that will be used in the subsequent Connect calls
+var (
+	// VPP binary API adapter that will be used in the subsequent Connect calls
+	vppAdapter adapter.VppAPI
+)
+
+func getVppAdapter(shm string) adapter.VppAPI {
+	if vppAdapter == nil {
+		vppAdapter = vppapiclient.NewVppClient(shm)
+	}
+	return vppAdapter
+}
+
+// SetVppAdapter sets the adapter that will be used for connections to VPP in the subsequent `Connect` calls.
+func SetVppAdapter(a adapter.VppAPI) {
+	vppAdapter = a
+}
 
 // Connect connects the govpp core to VPP either using the default VPP Adapter, or using the adapter previously
 // set by SetAdapter (useful mostly just for unit/integration tests with mocked VPP adapter).
 // This call blocks until VPP is connected, or an error occurs. Only one connection attempt will be performed.
 func Connect(shm string) (*core.Connection, error) {
-	if vppAdapter == nil {
-		vppAdapter = vppapiclient.NewVppAdapter(shm)
-	}
-	return core.Connect(vppAdapter)
+	return core.Connect(getVppAdapter(shm))
 }
 
 // AsyncConnect asynchronously connects the govpp core to VPP either using the default VPP Adapter,
@@ -37,14 +50,6 @@ func Connect(shm string) (*core.Connection, error) {
 // This call does not block until connection is established, it returns immediately. The caller is
 // supposed to watch the returned ConnectionState channel for Connected/Disconnected events.
 // In case of disconnect, the library will asynchronously try to reconnect.
-func AsyncConnect(shm string) (*core.Connection, chan core.ConnectionEvent, error) {
-	if vppAdapter == nil {
-		vppAdapter = vppapiclient.NewVppAdapter(shm)
-	}
-	return core.AsyncConnect(vppAdapter)
+func AsyncConnect(shm string, attempts int, interval time.Duration) (*core.Connection, chan core.ConnectionEvent, error) {
+	return core.AsyncConnect(getVppAdapter(shm), attempts, interval)
 }
-
-// SetAdapter sets the adapter that will be used for connections to VPP in the subsequent `Connect` calls.
-func SetAdapter(ad adapter.VppAdapter) {
-	vppAdapter = ad
-}
\ No newline at end of file