OLD | NEW |
1 // Copyright 2015 The LUCI Authors. All rights reserved. | 1 // Copyright 2015 The LUCI Authors. All rights reserved. |
2 // Use of this source code is governed under the Apache License, Version 2.0 | 2 // Use of this source code is governed under the Apache License, Version 2.0 |
3 // that can be found in the LICENSE file. | 3 // that can be found in the LICENSE file. |
4 | 4 |
5 package mutate | 5 package mutate |
6 | 6 |
7 import ( | 7 import ( |
8 ds "github.com/luci/gae/service/datastore" | 8 ds "github.com/luci/gae/service/datastore" |
9 "github.com/luci/luci-go/common/data/bit_field" | 9 "github.com/luci/luci-go/common/data/bit_field" |
| 10 "github.com/luci/luci-go/common/errors" |
10 "github.com/luci/luci-go/common/logging" | 11 "github.com/luci/luci-go/common/logging" |
11 "github.com/luci/luci-go/dm/api/service/v1" | 12 "github.com/luci/luci-go/dm/api/service/v1" |
12 "github.com/luci/luci-go/dm/appengine/model" | 13 "github.com/luci/luci-go/dm/appengine/model" |
13 "github.com/luci/luci-go/grpc/grpcutil" | 14 "github.com/luci/luci-go/grpc/grpcutil" |
14 "github.com/luci/luci-go/tumble" | 15 "github.com/luci/luci-go/tumble" |
15 "golang.org/x/net/context" | 16 "golang.org/x/net/context" |
16 "google.golang.org/grpc/codes" | 17 "google.golang.org/grpc/codes" |
17 ) | 18 ) |
18 | 19 |
19 // AddDeps transactionally stops the current execution and adds one or more | 20 // AddDeps transactionally stops the current execution and adds one or more |
(...skipping 23 matching lines...) Expand all Loading... |
43 if err != nil { | 44 if err != nil { |
44 return | 45 return |
45 } | 46 } |
46 | 47 |
47 if err = ResetExecutionTimeout(c, ex); err != nil { | 48 if err = ResetExecutionTimeout(c, ex); err != nil { |
48 logging.WithError(err).Errorf(c, "could not reset timeout") | 49 logging.WithError(err).Errorf(c, "could not reset timeout") |
49 return | 50 return |
50 } | 51 } |
51 | 52 |
52 fwdDeps, err := filterExisting(c, model.FwdDepsFromList(c, a.Auth.Id.Att
emptID(), a.Deps)) | 53 fwdDeps, err := filterExisting(c, model.FwdDepsFromList(c, a.Auth.Id.Att
emptID(), a.Deps)) |
53 » err = grpcutil.Annotate(err, codes.Internal).Reason("while filtering dep
s").Err() | 54 » err = errors.Annotate(err).Tag(grpcutil.Tag.With(codes.Internal)). |
| 55 » » Reason("while filtering deps").Err() |
54 if err != nil || len(fwdDeps) == 0 { | 56 if err != nil || len(fwdDeps) == 0 { |
55 return | 57 return |
56 } | 58 } |
57 | 59 |
58 logging.Fields{"aid": atmpt.ID, "count": len(fwdDeps)}.Infof(c, "added d
eps") | 60 logging.Fields{"aid": atmpt.ID, "count": len(fwdDeps)}.Infof(c, "added d
eps") |
59 atmpt.DepMap = bit_field.Make(uint32(len(fwdDeps))) | 61 atmpt.DepMap = bit_field.Make(uint32(len(fwdDeps))) |
60 | 62 |
61 for i, fdp := range fwdDeps { | 63 for i, fdp := range fwdDeps { |
62 fdp.BitIndex = uint32(i) | 64 fdp.BitIndex = uint32(i) |
63 fdp.ForExecution = atmpt.CurExecution | 65 fdp.ForExecution = atmpt.CurExecution |
64 } | 66 } |
65 | 67 |
66 if err = ds.Put(c, fwdDeps, atmpt, ex); err != nil { | 68 if err = ds.Put(c, fwdDeps, atmpt, ex); err != nil { |
67 » » err = grpcutil.Annotate(err, codes.Internal).Reason("putting stu
ff").Err() | 69 » » err = errors.Annotate(err).Tag(grpcutil.Tag.With(codes.Internal)
). |
| 70 » » » Reason("putting stuff").Err() |
68 return | 71 return |
69 } | 72 } |
70 | 73 |
71 mergeQuestMap := map[string]*MergeQuest(nil) | 74 mergeQuestMap := map[string]*MergeQuest(nil) |
72 if len(a.Quests) > 0 { | 75 if len(a.Quests) > 0 { |
73 mergeQuestMap = make(map[string]*MergeQuest, len(a.Quests)) | 76 mergeQuestMap = make(map[string]*MergeQuest, len(a.Quests)) |
74 for _, q := range a.Quests { | 77 for _, q := range a.Quests { |
75 mergeQuestMap[q.ID] = &MergeQuest{Quest: q} | 78 mergeQuestMap[q.ID] = &MergeQuest{Quest: q} |
76 } | 79 } |
77 } | 80 } |
(...skipping 25 matching lines...) Expand all Loading... |
103 for _, mut := range mergeQuestMap { | 106 for _, mut := range mergeQuestMap { |
104 muts = append(muts, mut) | 107 muts = append(muts, mut) |
105 } | 108 } |
106 | 109 |
107 return | 110 return |
108 } | 111 } |
109 | 112 |
110 func init() { | 113 func init() { |
111 tumble.Register((*AddDeps)(nil)) | 114 tumble.Register((*AddDeps)(nil)) |
112 } | 115 } |
OLD | NEW |