additional http api methods

Author: Pavel Safonov

go.mod

@@ -3,16 +3,16 @@ module github.com/pnsafonov/pind
 go 1.19
 
 require (
-	github.com/lrita/numa v1.0.2
+	github.com/lrita/numa v1.0.3
 	github.com/prometheus/procfs v0.12.0
 	github.com/sirupsen/logrus v1.9.3
+	github.com/t-tomalak/logrus-easy-formatter v0.0.0-20190827215021-c074f06c5816
 	golang.org/x/sys v0.12.0
 	gopkg.in/natefinch/lumberjack.v2 v2.2.1
 	gopkg.in/yaml.v3 v3.0.1
 )
 
 require (
 	github.com/intel-go/cpuid v0.0.0-20181003105527-1a4a6f06a1c6 // indirect
-	github.com/t-tomalak/logrus-easy-formatter v0.0.0-20190827215021-c074f06c5816 // indirect
 	golang.org/x/sync v0.3.0 // indirect
 )

go.sum

@@ -7,6 +7,8 @@ github.com/intel-go/cpuid v0.0.0-20181003105527-1a4a6f06a1c6/go.mod h1:RmeVYf9Xr
 github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
 github.com/lrita/numa v1.0.2 h1:RzJFNvgHXtzyf+RsGRP9XyMHjNUQFOEypv+2oXaJBj8=
 github.com/lrita/numa v1.0.2/go.mod h1:B0HvTeFKP4P+yb/BvVhww9Sx+q/zZEwTfzPL0LhF79M=
+github.com/lrita/numa v1.0.3 h1:9vx+77J1GSYdphcABftzenMTqzYLygLJa5EzZN1wa9w=
+github.com/lrita/numa v1.0.3/go.mod h1:B0HvTeFKP4P+yb/BvVhww9Sx+q/zZEwTfzPL0LhF79M=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/prometheus/procfs v0.12.0 h1:jluTpSng7V9hY0O2R9DzzJHYb2xULk9VTR1V1R/k6Bo=

pkg/http_api/http_api.go

@@ -65,10 +65,38 @@ func (x *HttpApi) serve(l net.Listener) {
 	_ = x.Server.Serve(l)
 }
 
+func writeStateBytes(w http.ResponseWriter, state *State) error {
+	stateBytes, err := json.MarshalIndent(state, "", "    ")
+	if err != nil {
+		log.Errorf("HttpApi writeStateBytes, json.Marshal err = %v", err)
+		return err
+	}
+
+	_, err = w.Write(stateBytes)
+	return err
+}
+
 // getApiState - /api/state
 func (x *HttpApi) getApiState(w http.ResponseWriter, r *http.Request) {
 	header := w.Header()
 	header.Set("Content-Type", "application/json; charset=utf-8")
 
+	// /api/state?vm_name=my_env_db
+	vmName := r.URL.Query().Get("vm_name")
+	if vmName != "" {
+		state1 := x.state.CloneWithFilter1(vmName)
+		_ = writeStateBytes(w, state1)
+		return
+	}
+
+	// /api/state?vm_prefix=my_env_
+	vmPrefix := r.URL.Query().Get("vm_prefix")
+	if vmPrefix != "" {
+		state1 := x.state.CloneWithFilter2(vmPrefix)
+		_ = writeStateBytes(w, state1)
+		return
+	}
+
+	// /api/state
 	_, _ = w.Write(x.stateBytes)
 }

pkg/http_api/state0.go

@@ -0,0 +1,47 @@
+package http_api
+
+import "strings"
+
+func (x *State) CloneWithFilter0(inFilter map[string]*Proc) *State {
+	procs0 := &Procs{
+		NotInFilter: x.Procs.NotInFilter,
+		InFilter:    inFilter,
+	}
+
+	state0 := &State{
+		Version: x.Version,
+		GitHash: x.GitHash,
+		Errors:  x.Errors,
+		Time:    x.Time,
+		Procs:   procs0,
+
+		Pool: x.Pool,
+		Numa: x.Numa,
+
+		Config: x.Config,
+	}
+
+	return state0
+}
+
+func (x *State) CloneWithFilter1(vmName string) *State {
+	inFilter := make(map[string]*Proc)
+	for k, v := range x.Procs.InFilter {
+		if v.VmName == vmName {
+			inFilter[k] = v
+		}
+	}
+
+	return x.CloneWithFilter0(inFilter)
+}
+
+func (x *State) CloneWithFilter2(vmPrefix string) *State {
+	inFilter := make(map[string]*Proc)
+	for k, v := range x.Procs.InFilter {
+		if strings.HasPrefix(v.VmName, vmPrefix) {
+			inFilter[k] = v
+		}
+	}
+
+	return x.CloneWithFilter0(inFilter)
+}