Support OwnerUsers filter in Hypervisor.ListVMs SRPC method.

Symantec · Feb 14, 2019 · bbc6d77 · bbc6d77
1 parent 4239249
commit bbc6d77
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 7 deletions.
diff --git a/hypervisor/httpd/listVMs.go b/hypervisor/httpd/listVMs.go
@@ -16,7 +16,7 @@ func (s state) listVMsHandler(w http.ResponseWriter, req *http.Request) {
 	parsedQuery := url.ParseQuery(req.URL)
 	writer := bufio.NewWriter(w)
 	defer writer.Flush()
-	ipAddrs := s.manager.ListVMs(true)
+	ipAddrs := s.manager.ListVMs(nil, true)
 	matchState := parsedQuery.Table["state"]
 	if parsedQuery.OutputType() == url.OutputTypeText && matchState == "" {
 		for _, ipAddr := range ipAddrs {

diff --git a/hypervisor/manager/api.go b/hypervisor/manager/api.go
@@ -233,8 +233,8 @@ func (m *Manager) ListSubnets(doSort bool) []proto.Subnet {
 	return m.listSubnets(doSort)
 }
 
-func (m *Manager) ListVMs(doSort bool) []string {
-	return m.listVMs(doSort)
+func (m *Manager) ListVMs(ownerUsers []string, doSort bool) []string {
+	return m.listVMs(ownerUsers, doSort)
 }
 
 func (m *Manager) ListVolumeDirectories() []string {

diff --git a/hypervisor/manager/vm.go b/hypervisor/manager/vm.go
@@ -1027,11 +1027,23 @@ func (m *Manager) importLocalVm(authInfo *srpc.AuthInformation,
 	return nil
 }
 
-func (m *Manager) listVMs(doSort bool) []string {
+func (m *Manager) listVMs(ownerUsers []string, doSort bool) []string {
 	m.mutex.RLock()
 	ipAddrs := make([]string, 0, len(m.vms))
-	for ipAddr := range m.vms {
-		ipAddrs = append(ipAddrs, ipAddr)
+	for ipAddr, vm := range m.vms {
+		include := true
+		if len(ownerUsers) > 0 {
+			include = false
+			for _, ownerUser := range ownerUsers {
+				if _, ok := vm.ownerUsers[ownerUser]; ok {
+					include = true
+					break
+				}
+			}
+		}
+		if include {
+			ipAddrs = append(ipAddrs, ipAddr)
+		}
 	}
 	m.mutex.RUnlock()
 	if doSort {

diff --git a/hypervisor/rpcd/listVMs.go b/hypervisor/rpcd/listVMs.go
@@ -10,7 +10,7 @@ import (
 func (t *srpcType) ListVMs(conn *srpc.Conn,
 	request hypervisor.ListVMsRequest,
 	reply *hypervisor.ListVMsResponse) error {
-	ipAddressStrings := t.manager.ListVMs(request.Sort)
+	ipAddressStrings := t.manager.ListVMs(request.OwnerUsers, request.Sort)
 	ipAddresses := make([]net.IP, 0, len(ipAddressStrings))
 	for _, ipAddressString := range ipAddressStrings {
 		ipAddress := net.ParseIP(ipAddressString)