add OpenConnections function to primitives

Author: Eslam-Nawara

pkg/primitives/statistics.go

@@ -283,7 +283,6 @@ func (s *statsStream) ListGPUs() ([]pkg.GPUInfo, error) {
 		}
 		for _, dl := range active.Deployments {
 			for _, wl := range dl.Workloads {
-
 				if wl.Type == zos.ZMachineType || wl.Type == zos.ZMachineLightType {
 					var vm vmType
 					if err := json.Unmarshal(wl.Data, &vm); err != nil {
@@ -294,7 +293,6 @@ func (s *statsStream) ListGPUs() ([]pkg.GPUInfo, error) {
 						gpus[string(gpu)] = dl.ContractID
 					}
 				}
-
 			}
 		}
 		return gpus, nil
@@ -347,3 +345,7 @@ func (s *statsStream) openConnectionsCount() (int, error) {
 	}
 	return strconv.Atoi(strings.TrimSpace(string(out)))
 }
+
+func (s *statsStream) OpenConnections() ([]byte, error) {
+	return exec.Command("ss", "-ptn", "state", "established").Output()
+}

pkg/provision.go

@@ -20,6 +20,8 @@ type Provision interface {
 	Changes(twin uint32, contractID uint64) ([]gridtypes.Workload, error)
 	ListPublicIPs() ([]string, error)
 	ListPrivateIPs(twin uint32, network gridtypes.Name) ([]string, error)
+	Pause(twin uint32, id uint64) error
+	Resume(twin uint32, id uint64) error
 }
 
 type Statistics interface {
@@ -29,6 +31,7 @@ type Statistics interface {
 	Workloads() (int, error)
 	GetCounters() (Counters, error)
 	ListGPUs() ([]GPUInfo, error)
+	OpenConnections() ([]byte, error)
 }
 
 type Counters struct {

pkg/provision/engine.go

@@ -364,7 +364,7 @@ func (e *NativeEngine) Provision(ctx context.Context, deployment gridtypes.Deplo
 }
 
 // Pause deployment
-func (e *NativeEngine) Pause(ctx context.Context, twin uint32, id uint64) error {
+func (e *NativeEngine) Pause(twin uint32, id uint64) error {
 	deployment, err := e.storage.Get(twin, id)
 	if err != nil {
 		return err
@@ -384,7 +384,7 @@ func (e *NativeEngine) Pause(ctx context.Context, twin uint32, id uint64) error
 }
 
 // Resume deployment
-func (e *NativeEngine) Resume(ctx context.Context, twin uint32, id uint64) error {
+func (e *NativeEngine) Resume(twin uint32, id uint64) error {
 	deployment, err := e.storage.Get(twin, id)
 	if err != nil {
 		return err

pkg/provision/interface.go

@@ -19,8 +19,8 @@ type Engine interface {
 	// and will be processes later
 	Provision(ctx context.Context, wl gridtypes.Deployment) error
 	Deprovision(ctx context.Context, twin uint32, id uint64, reason string) error
-	Pause(ctx context.Context, twin uint32, id uint64) error
-	Resume(ctx context.Context, twin uint32, id uint64) error
+	Pause(twin uint32, id uint64) error
+	Resume(twin uint32, id uint64) error
 	Update(ctx context.Context, update gridtypes.Deployment) error
 	Storage() Storage
 	Twins() Twins
@@ -61,7 +61,7 @@ var (
 	// ErrDeploymentConflict returned if deployment cannot be stored because
 	// it conflicts with another deployment
 	ErrDeploymentConflict = fmt.Errorf("conflict")
-	//ErrDeploymentNotExists returned if object not exists
+	// ErrDeploymentNotExists returned if object not exists
 	ErrDeploymentNotExists = fmt.Errorf("deployment does not exist")
 	// ErrWorkloadNotExist returned by storage if workload does not exist
 	ErrWorkloadNotExist = fmt.Errorf("workload does not exist")
@@ -78,10 +78,12 @@ var (
 )
 
 // Field interface
-type Field interface{}
-type VersionField struct {
-	Version uint32
-}
+type (
+	Field        interface{}
+	VersionField struct {
+		Version uint32
+	}
+)
 
 type DescriptionField struct {
 	Description string

pkg/stubs/provision_stub.go

@@ -159,3 +159,33 @@ func (s *ProvisionStub) ListPublicIPs(ctx context.Context) (ret0 []string, ret1
 	}
 	return
 }
+
+func (s *ProvisionStub) Pause(ctx context.Context, arg0 uint32, arg1 uint64) (ret0 error) {
+	args := []interface{}{arg0, arg1}
+	result, err := s.client.RequestContext(ctx, s.module, s.object, "Pause", args...)
+	if err != nil {
+		panic(err)
+	}
+	result.PanicOnError()
+	ret0 = result.CallError()
+	loader := zbus.Loader{}
+	if err := result.Unmarshal(&loader); err != nil {
+		panic(err)
+	}
+	return
+}
+
+func (s *ProvisionStub) Resume(ctx context.Context, arg0 uint32, arg1 uint64) (ret0 error) {
+	args := []interface{}{arg0, arg1}
+	result, err := s.client.RequestContext(ctx, s.module, s.object, "Resume", 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/stubs/statistics_stub.go

@@ -79,6 +79,23 @@ func (s *StatisticsStub) ListGPUs(ctx context.Context) (ret0 []pkg.GPUInfo, ret1
 	return
 }
 
+func (s *StatisticsStub) OpenConnections(ctx context.Context) (ret0 []uint8, ret1 error) {
+	args := []interface{}{}
+	result, err := s.client.RequestContext(ctx, s.module, s.object, "OpenConnections", args...)
+	if err != nil {
+		panic(err)
+	}
+	result.PanicOnError()
+	ret1 = result.CallError()
+	loader := zbus.Loader{
+		&ret0,
+	}
+	if err := result.Unmarshal(&loader); err != nil {
+		panic(err)
+	}
+	return
+}
+
 func (s *StatisticsStub) ReservedStream(ctx context.Context) (<-chan gridtypes.Capacity, error) {
 	ch := make(chan gridtypes.Capacity, 1)
 	recv, err := s.client.Stream(ctx, s.module, s.object, "ReservedStream")

pkg/zinit/commands.go

@@ -45,9 +45,9 @@ const (
 	ServiceStateSuccess = "success"
 	// ServiceStateError is return when we a service exit with an error (exit code != 0)
 	ServiceStateError = "error"
-	//ServiceStateFailure is set of zinit can not spawn a service in the first place
-	//due to a missing executable for example. Unlike `error` which is returned if the
-	//service itself exits with an error.
+	// ServiceStateFailure is set of zinit can not spawn a service in the first place
+	// due to a missing executable for example. Unlike `error` which is returned if the
+	// service itself exits with an error.
 	ServiceStateFailure = "failure"
 )
 
@@ -204,7 +204,6 @@ type ServiceStatus struct {
 func (c *Client) List() (out map[string]ServiceState, err error) {
 	err = c.cmd("list", &out)
 	return
-
 }
 
 // Status returns the status of a service
@@ -611,3 +610,36 @@ func (c *Client) Destroy(timeout time.Duration, services ...string) error {
 
 	return nil
 }
+
+// List returns all the service monitored and their status
+func (c *Client) Log(n int) (out []byte, err error) {
+	cmd1 := exec.Command("zinit", "log", "-s")
+	cmd2 := exec.Command("tail", "-n", fmt.Sprint(n))
+
+	cmd2.Stdin, err = cmd1.StdoutPipe()
+	if err != nil {
+		return
+	}
+
+	err = cmd1.Start()
+	if err != nil {
+		return
+	}
+
+	output, err := cmd2.Output()
+	if err != nil {
+		return
+	}
+
+	err = cmd1.Wait()
+	if err != nil {
+		return
+	}
+
+	return output, err
+}
+
+// Restart restarts a service.
+func (c *Client) Restart(service string) error {
+	return c.cmd(fmt.Sprintf("restart %s", service), nil)
+}