gravitl/netmaker
1
1
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2024-09-20 23:36:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

added conn closes

Browse source
This commit is contained in:
worker-9 2021-09-23 09:04:33 -04:00
parent 0551e54dd6
commit 5cf69ebe5b
4 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
netclient/functions/checkin.go
View file

@ -171,6 +171,7 @@ func Pull(network string, manual bool) (*models.Node, error) {
ncutils.PrintLog("Cant dial GRPC server: "+err.Error(), 1)
return nil, err
}
defer conn.Close()
wcclient := nodepb.NewNodeServiceClient(conn)
ctx, err := auth.SetJWT(wcclient, network)
@ -253,6 +254,7 @@ func Push(network string) error {
ncutils.PrintLog("Cant dial GRPC server: "+err.Error(), 1)
return err
}
defer conn.Close()
wcclient = nodepb.NewNodeServiceClient(conn)
ctx, err := auth.SetJWT(wcclient, network)

1
netclient/functions/common.go
View file

@ -177,6 +177,7 @@ func LeaveNetwork(network string) error {
if err != nil {
log.Printf("Unable to establish client connection to "+servercfg.GRPCAddress+": %v", err)
} else {
defer conn.Close()
wcclient = nodepb.NewNodeServiceClient(conn)
ctx, err := auth.SetJWT(wcclient, network)

2
netclient/functions/join.go
View file

@ -90,7 +90,7 @@ func JoinNetwork(cfg config.ClientConfig, privateKey string) error {
if err != nil {
log.Fatalf("Unable to establish client connection to "+cfg.Server.GRPCAddress+": %v", err)
}
defer conn.Close()
wcclient = nodepb.NewNodeServiceClient(conn)
postnode := &models.Node{

3
netclient/server/grpc.go
View file

@ -29,6 +29,7 @@ func getGrpcClient(cfg *config.ClientConfig) (nodepb.NodeServiceClient, error) {
if err != nil {
return nil, err
}
defer conn.Close()
wcclient = nodepb.NewNodeServiceClient(conn)
return wcclient, nil
}
@ -138,6 +139,7 @@ func GetPeers(macaddress string, network string, server string, dualstack bool,
if err != nil {
log.Fatalf("Unable to establish client connection to localhost:50051: %v", err)
}
defer conn.Close()
// Instantiate the BlogServiceClient with our client connection to the server
wcclient = nodepb.NewNodeServiceClient(conn)
@ -293,6 +295,7 @@ func GetExtPeers(macaddress string, network string, server string, dualstack boo
if err != nil {
log.Fatalf("Unable to establish client connection to localhost:50051: %v", err)
}
defer conn.Close()
// Instantiate the BlogServiceClient with our client connection to the server
wcclient = nodepb.NewNodeServiceClient(conn)