Initial distributor implementation

common/api/dm/service/v1/walk_graph_normalize.go

 // Copyright 2016 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package dm
 
 import (
         "github.com/luci/luci-go/common/errors"
+        "math"
 )
 
 const (
         // DefaultLimitMaxDataSize is the default MaxDataSize value (16MB).
         DefaultLimitMaxDataSize = 16 * 1024 * 1024
 
         // MaxLimitMaxDataSize is the maximum MaxDataSize value (30MB).
         MaxLimitMaxDataSize = 30 * 1024 * 1024
 )
 
 // Normalize returns an error iff the WalkGraphReq is invalid.
 func (w *WalkGraphReq) Normalize() error {
+        if w.Auth != nil {
+                if err := w.Auth.Normalize(); err != nil {
+                        return err
+                }
+        }
+
         if w.Query == nil {
                 return errors.New("must specify a Query")
         }
         if err := w.Query.Normalize(); err != nil {
                 return err
         }
 
         if w.Mode == nil {
                 w.Mode = &WalkGraphReq_Mode{}
         }
@@ skipping 11 matching lines @@
         if w.Limit.MaxDataSize == 0 {
                 w.Limit.MaxDataSize = DefaultLimitMaxDataSize
         }
         if w.Limit.MaxDataSize > MaxLimitMaxDataSize {
                 w.Limit.MaxDataSize = MaxLimitMaxDataSize
         }
 
         if w.Include == nil {
                 w.Include = &WalkGraphReq_Include{}
         } else {
-                if w.Include.AttemptResult {
+                if w.Include.All {
+                        w.Include.ObjectIds = true
+                        w.Include.QuestData = true
                         w.Include.AttemptData = true
+                        w.Include.AttemptResult = true
+                        w.Include.ExpiredAttempts = true
+                        w.Include.NumExecutions = math.MaxUint32
+                        w.Include.ExecutionInfoUrl = true
+                        w.Include.FwdDeps = true
+                        w.Include.BackDeps = true
+                } else {
+                        if w.Include.AttemptResult {
+                                w.Include.AttemptData = true
+                        }
+                        if w.Include.NumExecutions == 0 {
+                                w.Include.ExecutionInfoUrl = false
+                        }
                 }
         }
         return nil
 }