autocomplete api

nomad/resources_endpoint.go

@@ -26,7 +26,7 @@ type Resources struct {
 
 // getMatches extracts matches for an iterator, and returns a list of ids for
 // these matches.
-func getMatches(iter memdb.ResultIterator) ([]string, bool) {
+func (r *Resources) getMatches(iter memdb.ResultIterator) ([]string, bool) {
 	var matches []string
 	isTruncated := false
 
@@ -37,7 +37,7 @@ func getMatches(iter memdb.ResultIterator) ([]string, bool) {
 		}
 
 		var id string
-		switch raw.(type) {
+		switch t := raw.(type) {
 		case *structs.Job:
 			id = raw.(*structs.Job).ID
 		case *structs.Evaluation:
@@ -47,8 +47,7 @@ func getMatches(iter memdb.ResultIterator) ([]string, bool) {
 		case *structs.Node:
 			id = raw.(*structs.Node).ID
 		default:
-			//s.logger.Printf("[ERR] nomad: unexpected type for resources context; not a job, allocation, node, or evaluation")
-			// TODO
+			r.srv.logger.Printf("[ERR] nomad.resources: unexpected type for resources context; %T \n", t)
 			continue
 		}
 
@@ -109,25 +108,22 @@ func (r *Resources) List(args *structs.ResourceListRequest,
 
 			// Return matches for the given prefix
 			for k, v := range iters {
-				res, isTrunc := getMatches(v)
+				res, isTrunc := r.getMatches(v)
 				reply.Matches[k] = res
 				reply.Truncations[k] = isTrunc
 			}
 
 			// Set the index for the context. If the context has been specified, it
-			// is the only non-empty match set, and the index is set for it.
-			// If the context was not specified, we set the index to be the max index
-			// from  available contexts
+			// will be used as the index of the response. Otherwise, the
+			// maximum index from all resources will be used.
 			reply.Index = 0
-			for k, v := range reply.Matches {
-				if len(v) != 0 { // make sure matches exist for this context
-					index, err := state.Index(k)
-					if err != nil {
-						return err
-					}
-					if index > reply.Index {
-						reply.Index = index
-					}
+			for _, e := range contexts {
+				index, err := state.Index(e)
+				if err != nil {
+					return err
+				}
+				if index > reply.Index {
+					reply.Index = index
 				}
 			}
 

nomad/resources_endpoint_test.go

@@ -267,7 +267,7 @@ func TestResourcesEndpoint_List_NoContext(t *testing.T) {
 	assert.Equal(uint64(1000), resp.Index)
 }
 
-//// Tests that the top 20 matches are returned when no prefix is set
+// Tests that the top 20 matches are returned when no prefix is set
 func TestResourcesEndpoint_List_NoPrefix(t *testing.T) {
 	assert := assert.New(t)
 

nomad/structs/structs.go

@@ -235,8 +235,12 @@ type NodeSpecificRequest struct {
 // the match list is truncated specific to each type of context.
 type ResourceListResponse struct {
 	// Map of context types to resource ids which match a specified prefix
-	Matches     map[string][]string
+	Matches map[string][]string
+
+	// Truncations indicates whether the matches for a particular context have
+	// been truncated
 	Truncations map[string]bool
+
 	QueryMeta
 }
 
@@ -247,6 +251,7 @@ type ResourceListRequest struct {
 	// Prefix is what resources are matched to. I.e, if the given prefix were
 	// "a", potential matches might be "abcd" or "aabb"
 	Prefix string
+
 	// Context is the resource that can be matched. A context can be a job, node,
 	// evaluation, allocation, or empty (indicated every context should be
 	// matched)