| OLD | NEW |
| 1 // Copyright 2017 The LUCI Authors. All rights reserved. | 1 // Copyright 2017 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 revocation | 5 package revocation |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "golang.org/x/net/context" | 8 "golang.org/x/net/context" |
| 9 | 9 |
| 10 "github.com/luci/gae/service/datastore" | 10 "github.com/luci/gae/service/datastore" |
| 11 » "github.com/luci/luci-go/common/errors" | 11 » "github.com/luci/luci-go/common/retry" |
| 12 ) | 12 ) |
| 13 | 13 |
| 14 // GenerateTokenID produces an int64 that can be used as a token identifier. | 14 // GenerateTokenID produces an int64 that can be used as a token identifier. |
| 15 // | 15 // |
| 16 // We reuse datastore ID generator to produce token ids. The tokens are not | 16 // We reuse datastore ID generator to produce token ids. The tokens are not |
| 17 // actually stored in the datastore. The generated ID sequence is associated | 17 // actually stored in the datastore. The generated ID sequence is associated |
| 18 // with some entity kind (indicated via 'kind'). If we ever need to restart the | 18 // with some entity kind (indicated via 'kind'). If we ever need to restart the |
| 19 // ID sequence, this kind can be changed. | 19 // ID sequence, this kind can be changed. |
| 20 func GenerateTokenID(c context.Context, kind string) (int64, error) { | 20 func GenerateTokenID(c context.Context, kind string) (int64, error) { |
| 21 // Note: AllocateIDs modifies passed slice in place, by replacing the ke
ys | 21 // Note: AllocateIDs modifies passed slice in place, by replacing the ke
ys |
| 22 // there. | 22 // there. |
| 23 keys := []*datastore.Key{ | 23 keys := []*datastore.Key{ |
| 24 datastore.NewKey(c, kind, "", 0, nil), | 24 datastore.NewKey(c, kind, "", 0, nil), |
| 25 } | 25 } |
| 26 if err := datastore.AllocateIDs(c, keys); err != nil { | 26 if err := datastore.AllocateIDs(c, keys); err != nil { |
| 27 » » return 0, errors.WrapTransient(err) | 27 » » return 0, retry.Tag.Apply(err) |
| 28 } | 28 } |
| 29 return keys[0].IntID(), nil | 29 return keys[0].IntID(), nil |
| 30 } | 30 } |
| OLD | NEW |