Remove heartbeat | Add Deregister

cmd/skywire-visor/commands/systray.go

@@ -4,6 +4,7 @@ package commands
 
 import (
 	"context"
+
 	"github.com/getlantern/systray"
 	"github.com/skycoin/skycoin/src/util/logging"
 

go.sum

@@ -91,6 +91,7 @@ github.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V
 github.com/go-ole/go-ole v1.2.4 h1:nNBDSCOigTSiarFpYE9J/KtEA1IOW4CNeqT9TQDqCxI=
 github.com/go-ole/go-ole v1.2.4/go.mod h1:XCwSNxSkXRo4vlyPy93sltvi/qJq0jqQhjqQNIwKuxM=
 github.com/go-redis/redis v6.15.8+incompatible/go.mod h1:NAIEuMOZ/fxfXJIrKDQDz8wamY7mA7PouImQ2Jvg6kA=
+github.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk=
 github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
 github.com/gobwas/httphead v0.0.0-20180130184737-2c6c146eadee h1:s+21KNqlpePfkah2I+gwHF8xmJWRjooY+5248k6m4A0=
 github.com/gobwas/httphead v0.0.0-20180130184737-2c6c146eadee/go.mod h1:L0fX3K22YWvt/FAX9NnzrNzcI4wNYi9Yku4O0LKYflo=
@@ -184,12 +185,7 @@ github.com/klauspost/reedsolomon v1.9.9/go.mod h1:O7yFFHiQwDR6b2t63KPUpccPtNdp5A
 github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
 github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
-<<<<<<< HEAD
-github.com/kr/pretty v0.2.0/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
-=======
-github.com/kr/pretty v0.2.1 h1:Fmg33tUaq4/8ym9TJN1x7sLJnHVwhP33CNkpYV/7rwI=
 github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
->>>>>>> upstream/develop
 github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
 github.com/kr/pty v1.1.3/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
 github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
@@ -474,11 +470,8 @@ golang.org/x/sys v0.0.0-20200519105757-fe76b779f299/go.mod h1:h1NjWce9XRLGQEsW7w
 golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20201231184435-2d18734c6014/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
-<<<<<<< HEAD
-golang.org/x/sys v0.0.0-20210217105451-b926d437f341/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
-=======
 golang.org/x/sys v0.0.0-20210112080510-489259a85091/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
->>>>>>> upstream/develop
+golang.org/x/sys v0.0.0-20210217105451-b926d437f341/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20210608053332-aa57babbf139 h1:C+AwYEtBp/VQwoLntUmQ/yx3MS9vmZaKNdw5eOpoQe8=

internal/gui/gui_darwin.go

@@ -13,7 +13,7 @@ const (
 	pngIconPath     = "/Applications/Skywire.app/Contents/Resources/icon.png"
 	iconPath        = "/Applications/Skywire.app/Contents/Resources/tray_icon.tiff"
 	deinstallerPath = "/Applications/Skywire.app/Contents/deinstaller"
-	appPath = "/Applications/Skywire.app"
+	appPath         = "/Applications/Skywire.app"
 )
 
 func preReadIcon() error {

pkg/app/appserver/proc.go

@@ -17,7 +17,7 @@ import (
 
 	"github.com/skycoin/skywire/pkg/app/appcommon"
 	"github.com/skycoin/skywire/pkg/app/appnet"
-	"github.com/skycoin/skywire/pkg/app/updatedisc"
+	"github.com/skycoin/skywire/pkg/app/managedisc"
 )
 
 var (
@@ -29,7 +29,7 @@ var (
 // communication.
 // TODO(evanlinjin): In the future, we will implement the ability to run multiple instances (procs) of a single app.
 type Proc struct {
-	disc updatedisc.Updater // app discovery client
+	disc managedisc.Manager // app discovery client
 	conf appcommon.ProcConfig
 	log  *logging.Logger
 
@@ -57,7 +57,7 @@ type Proc struct {
 }
 
 // NewProc constructs `Proc`.
-func NewProc(mLog *logging.MasterLogger, conf appcommon.ProcConfig, disc updatedisc.Updater, m ProcManager,
+func NewProc(mLog *logging.MasterLogger, conf appcommon.ProcConfig, disc managedisc.Manager, m ProcManager,
 	appName string) *Proc {
 	if mLog == nil {
 		mLog = logging.NewMasterLogger()
@@ -139,8 +139,8 @@ func (p *Proc) awaitConn() bool {
 	connDelta := p.rpcGW.cm.AddDeltaInformer()
 	go func() {
 		for n := range connDelta.Chan() {
-			if err := p.disc.ChangeValue(updatedisc.ConnCountValue, []byte(strconv.Itoa(n))); err != nil {
-				p.log.WithError(err).WithField("value", updatedisc.ConnCountValue).
+			if err := p.disc.ChangeValue(managedisc.ConnCountValue, []byte(strconv.Itoa(n))); err != nil {
+				p.log.WithError(err).WithField("value", managedisc.ConnCountValue).
 					Error("Failed to change app discovery value.")
 			}
 		}
@@ -149,8 +149,8 @@ func (p *Proc) awaitConn() bool {
 	lisDelta := p.rpcGW.lm.AddDeltaInformer()
 	go func() {
 		for n := range lisDelta.Chan() {
-			if err := p.disc.ChangeValue(updatedisc.ListenerCountValue, []byte(strconv.Itoa(n))); err != nil {
-				p.log.WithError(err).WithField("value", updatedisc.ListenerCountValue).
+			if err := p.disc.ChangeValue(managedisc.ListenerCountValue, []byte(strconv.Itoa(n))); err != nil {
+				p.log.WithError(err).WithField("value", managedisc.ListenerCountValue).
 					Error("Failed to change app discovery value.")
 			}
 		}
@@ -259,6 +259,9 @@ func (p *Proc) Stop() error {
 		}
 	}
 
+	// deregister discovery service
+	p.disc.Stop()
+
 	// the lock will be acquired as soon as the cmd finishes its work
 	p.waitMx.Lock()
 	defer func() {

pkg/app/appserver/proc_manager.go

@@ -14,7 +14,7 @@ import (
 
 	"github.com/skycoin/skywire/pkg/app/appcommon"
 	"github.com/skycoin/skywire/pkg/app/appevent"
-	"github.com/skycoin/skywire/pkg/app/updatedisc"
+	"github.com/skycoin/skywire/pkg/app/managedisc"
 )
 
 //go:generate mockery -name ProcManager -case underscore -inpkg
@@ -57,7 +57,7 @@ type procManager struct {
 	conns   map[string]net.Conn
 	connsWG sync.WaitGroup
 
-	discF      *updatedisc.Factory
+	discF      *managedisc.Factory
 	procs      map[string]*Proc
 	procsByKey map[appcommon.ProcKey]*Proc
 
@@ -69,12 +69,12 @@ type procManager struct {
 }
 
 // NewProcManager constructs `ProcManager`.
-func NewProcManager(mLog *logging.MasterLogger, discF *updatedisc.Factory, eb *appevent.Broadcaster, addr string) (ProcManager, error) {
+func NewProcManager(mLog *logging.MasterLogger, discF *managedisc.Factory, eb *appevent.Broadcaster, addr string) (ProcManager, error) {
 	if mLog == nil {
 		mLog = logging.NewMasterLogger()
 	}
 	if discF == nil {
-		discF = new(updatedisc.Factory)
+		discF = new(managedisc.Factory)
 	}
 	if eb == nil {
 		eb = appevent.NewBroadcaster(mLog.PackageLogger("event_broadcaster"), time.Second)
@@ -193,7 +193,7 @@ func (m *procManager) Start(conf appcommon.ProcConfig) (appcommon.ProcID, error)
 		break
 	}
 
-	disc, ok := m.discF.AppUpdater(conf)
+	disc, ok := m.discF.AppManager(conf)
 	if !ok {
 		log.WithField("appName", conf.AppName).
 			Debug("No app discovery associated with app.")

pkg/app/updatedisc/const.go → pkg/app/managedisc/const.go

@@ -1,4 +1,4 @@
-package updatedisc
+package managedisc
 
 // ChangeValue keys. Each key changes a different value for serviceUpdater.ChangeValue
 const (

pkg/app/managedisc/discovery_manager.go

@@ -0,0 +1,67 @@
+package managedisc
+
+import (
+	"context"
+	"strconv"
+	"sync"
+
+	"github.com/skycoin/skywire/pkg/servicedisc"
+)
+
+// Manager manages the associated app discovery
+type Manager interface {
+
+	// Start starts the manager.
+	Start()
+
+	// Stop stops the manager.
+	Stop()
+
+	// ChangeValue changes the associated value of the discovery entry.
+	ChangeValue(name string, v []byte) error
+}
+
+// emptyManager is for apps that do not require discovery manages.
+type emptyManager struct{}
+
+func (emptyManager) Start()                                  {}
+func (emptyManager) Stop()                                   {}
+func (emptyManager) ChangeValue(name string, v []byte) error { return nil }
+
+// serviceManager manages service-discovery entry of locally running App.
+type serviceManager struct {
+	client *servicedisc.HTTPClient
+	mu     sync.Mutex
+}
+
+func (u *serviceManager) Start() {
+	u.mu.Lock()
+	defer u.mu.Unlock()
+
+	ctx := context.Background()
+	if err := u.client.RegisterEntry(ctx); err != nil {
+		return
+	}
+}
+
+func (u *serviceManager) Stop() {
+	u.mu.Lock()
+	defer u.mu.Unlock()
+
+	ctx := context.Background()
+	if err := u.client.DeregisterEntry(ctx); err != nil {
+		return
+	}
+}
+
+func (u *serviceManager) ChangeValue(name string, v []byte) error {
+	switch name {
+	case ConnCountValue:
+		n, err := strconv.Atoi(string(v))
+		if err != nil {
+			return err
+		}
+		go u.client.UpdateStats(servicedisc.Stats{ConnectedClients: n})
+	}
+	return nil
+}

pkg/app/updatedisc/factory.go → pkg/app/managedisc/factory.go

@@ -1,8 +1,6 @@
-package updatedisc
+package managedisc
 
 import (
-	"time"
-
 	"github.com/sirupsen/logrus"
 	"github.com/skycoin/dmsg/cipher"
 	"github.com/skycoin/skycoin/src/util/logging"
@@ -14,30 +12,26 @@ import (
 
 // Factory creates appdisc.Updater instances based on the app name.
 type Factory struct {
-	Log            logrus.FieldLogger
-	PK             cipher.PubKey
-	SK             cipher.SecKey
-	UpdateInterval time.Duration
-	ServiceDisc    string // Address of service-discovery
+	Log         logrus.FieldLogger
+	PK          cipher.PubKey
+	SK          cipher.SecKey
+	ServiceDisc string // Address of service-discovery
 }
 
 func (f *Factory) setDefaults() {
 	if f.Log == nil {
 		f.Log = logging.MustGetLogger("appdisc")
 	}
-	if f.UpdateInterval == 0 {
-		f.UpdateInterval = skyenv.ServiceDiscUpdateInterval
-	}
 	if f.ServiceDisc == "" {
 		f.ServiceDisc = skyenv.DefaultServiceDiscAddr
 	}
 }
 
-// VisorUpdater obtains a visor updater.
-func (f *Factory) VisorUpdater(port uint16) Updater {
-	// Always return empty updater if keys are not set.
+// VisorManager obtains a visor manager.
+func (f *Factory) VisorManager(port uint16) Manager {
+	// Always return empty manager if keys are not set.
 	if f.setDefaults(); f.PK.Null() || f.SK.Null() {
-		return &emptyUpdater{}
+		return &emptyManager{}
 	}
 
 	conf := servicedisc.Config{
@@ -48,24 +42,23 @@ func (f *Factory) VisorUpdater(port uint16) Updater {
 		DiscAddr: f.ServiceDisc,
 	}
 
-	return &serviceUpdater{
-		client:   servicedisc.NewClient(f.Log, conf),
-		interval: f.UpdateInterval,
+	return &serviceManager{
+		client: servicedisc.NewClient(f.Log, conf),
 	}
 }
 
-// AppUpdater obtains an app updater based on the app name and configuration.
-func (f *Factory) AppUpdater(conf appcommon.ProcConfig) (Updater, bool) {
-	// Always return empty updater if keys are not set.
+// AppManager obtains an app manager based on the app name and configuration.
+func (f *Factory) AppManager(conf appcommon.ProcConfig) (Manager, bool) {
+	// Always return empty manager if keys are not set.
 	if f.setDefaults(); f.PK.Null() || f.SK.Null() {
-		return &emptyUpdater{}, false
+		return &emptyManager{}, false
 	}
 
 	log := f.Log.WithField("appName", conf.AppName)
 
-	// Do not update in service discovery if passcode-protected.
+	// Do not manage in service discovery if passcode-protected.
 	if conf.ContainsFlag("passcode") && conf.ArgVal("passcode") != "" {
-		return &emptyUpdater{}, false
+		return &emptyManager{}, false
 	}
 
 	getServiceDiscConf := func(conf appcommon.ProcConfig, sType string) servicedisc.Config {
@@ -80,11 +73,10 @@ func (f *Factory) AppUpdater(conf appcommon.ProcConfig) (Updater, bool) {
 
 	switch conf.AppName {
 	case skyenv.VPNServerName:
-		return &serviceUpdater{
-			client:   servicedisc.NewClient(log, getServiceDiscConf(conf, servicedisc.ServiceTypeVPN)),
-			interval: f.UpdateInterval,
+		return &serviceManager{
+			client: servicedisc.NewClient(log, getServiceDiscConf(conf, servicedisc.ServiceTypeVPN)),
 		}, true
 	default:
-		return &emptyUpdater{}, false
+		return &emptyManager{}, false
 	}
 }