update substrate gateway to allow updating sub connection

Author: Eslam-Nawara

pkg/api_gateway.go

@@ -10,6 +10,7 @@ import (
 //go:generate zbusc -module api-gateway -version 0.0.1 -name api-gateway -package stubs github.com/threefoldtech/zosbase/pkg+SubstrateGateway stubs/api_gateway_stub.go
 
 type SubstrateGateway interface {
+	UpdateSubstrateGatewayConnection(manager substrate.Manager) (err error)
 	CreateNode(node substrate.Node) (uint32, error)
 	CreateTwin(relay string, pk []byte) (uint32, error)
 	EnsureAccount(activationURL []string, termsAndConditionsLink string, termsAndConditionsHash string) (info substrate.AccountInfo, err error)

pkg/stubs/api_gateway_stub.go

@@ -387,3 +387,18 @@ func (s *SubstrateGatewayStub) UpdateNodeUptimeV2(ctx context.Context, arg0 uint
 	}
 	return
 }
+
+func (s *SubstrateGatewayStub) UpdateSubstrateGatewayConnection(ctx context.Context, arg0 tfchainclientgo.Manager) (ret0 error) {
+	args := []interface{}{arg0}
+	result, err := s.client.RequestContext(ctx, s.module, s.object, "UpdateSubstrateGatewayConnection", args...)
+	if err != nil {
+		panic(err)
+	}
+	result.PanicOnError()
+	ret0 = result.CallError()
+	loader := zbus.Loader{}
+	if err := result.Unmarshal(&loader); err != nil {
+		panic(err)
+	}
+	return
+}

pkg/substrate_gateway/substrate_gateway.go

@@ -31,6 +31,20 @@ func NewSubstrateGateway(manager substrate.Manager, identity substrate.Identity)
 	return gw, nil
 }
 
+// UpdateSubstrateGatewayConnection allow modules to update substrate manager so that the node can recover chain outage
+func (g *substrateGateway) UpdateSubstrateGatewayConnection(manager substrate.Manager) error {
+	sub, err := manager.Substrate()
+	if err != nil {
+		return err
+	}
+
+	// close the old connection
+	g.sub.Close()
+
+	g.sub = sub
+	return nil
+}
+
 func (g *substrateGateway) GetZosVersion() (string, error) {
 	log.Debug().Str("method", "GetZosVersion").Msg("method called")
 
@@ -207,6 +221,7 @@ func (g *substrateGateway) UpdateNodeUptimeV2(uptime uint64, timestampHint uint6
 	defer g.mu.Unlock()
 	return g.sub.UpdateNodeUptimeV2(g.identity, uptime, timestampHint)
 }
+
 func (g *substrateGateway) GetTime() (time.Time, error) {
 	log.Trace().Str("method", "Time").Msg("method called")