Chromium Code Reviews| Index: filter/count/user.go |
| diff --git a/filter/count/user.go b/filter/count/user.go |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..a12a5fcae4743b58400e1e375d5eb081d21ee338 |
| --- /dev/null |
| +++ b/filter/count/user.go |
| @@ -0,0 +1,76 @@ |
| +// Copyright 2015 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +package count |
| + |
| +import ( |
| + "github.com/luci/gae/service/user" |
| + "golang.org/x/net/context" |
| +) |
| + |
| +// UserCounter is the counter object for the User service. |
| +type UserCounter struct { |
| + Counter Entry |
| + CurrentOAuth Entry |
| + IsAdmin Entry |
| + LoginURL Entry |
| + LoginURLFederated Entry |
| + LogoutURL Entry |
| + OAuthConsumerKey Entry |
| +} |
| + |
| +type userCounter struct { |
| + c *UserCounter |
| + |
| + u user.Interface |
| +} |
| + |
| +var _ user.Interface = (*userCounter)(nil) |
| + |
| +func (u *userCounter) Current() *user.User { |
| + u.c.Counter.up() |
|
dnj
2015/12/15 18:59:30
Why not named "Current"?
iannucci
2015/12/15 19:12:21
Lol. Boog.
|
| + return u.u.Current() |
| +} |
| + |
| +func (u *userCounter) CurrentOAuth(scopes ...string) (*user.User, error) { |
| + ret, err := u.u.CurrentOAuth(scopes...) |
| + return ret, u.c.CurrentOAuth.up(err) |
| +} |
| + |
| +func (u *userCounter) IsAdmin() bool { |
| + u.c.IsAdmin.up() |
| + return u.u.IsAdmin() |
| +} |
| + |
| +func (u *userCounter) LoginURL(dest string) (string, error) { |
| + ret, err := u.u.LoginURL(dest) |
| + return ret, u.c.LoginURL.up(err) |
| +} |
| + |
| +func (u *userCounter) LoginURLFederated(dest, identity string) (string, error) { |
| + ret, err := u.u.LoginURLFederated(dest, identity) |
| + return ret, u.c.LoginURLFederated.up(err) |
| +} |
| + |
| +func (u *userCounter) LogoutURL(dest string) (string, error) { |
| + ret, err := u.u.LogoutURL(dest) |
| + return ret, u.c.LogoutURL.up(err) |
| +} |
| + |
| +func (u *userCounter) OAuthConsumerKey() (string, error) { |
| + ret, err := u.u.OAuthConsumerKey() |
| + return ret, u.c.OAuthConsumerKey.up(err) |
| +} |
| + |
| +func (u *userCounter) Testable() user.Testable { |
| + return u.u.Testable() |
| +} |
| + |
| +// FilterUser installs a counter User filter in the context. |
| +func FilterUser(c context.Context) (context.Context, *UserCounter) { |
| + state := &UserCounter{} |
| + return user.AddFilters(c, func(ic context.Context, u user.Interface) user.Interface { |
| + return &userCounter{state, u} |
| + }), state |
| +} |