| 26 | } |
| 27 | |
| 28 | func (wg *WireGuard) handleUserStats(ctx context.Context, request *common.StatRequest) (*common.StatResponse, error) { |
| 29 | var keys []string |
| 30 | if peer := wg.peerStore.GetByEmail(request.GetName()); peer != nil { |
| 31 | keys = []string{peer.PublicKey.String()} |
| 32 | } |
| 33 | if len(keys) == 0 { |
| 34 | return &common.StatResponse{Stats: []*common.Stat{}}, nil |
| 35 | } |
| 36 | |
| 37 | return wg.statsTracker.GetStats(ctx, keys, request.GetReset_()), nil |
| 38 | } |
| 39 | |
| 40 | func (wg *WireGuard) handleUsersStats(ctx context.Context, request *common.StatRequest) (*common.StatResponse, error) { |
| 41 | return wg.statsTracker.GetUsersStats(ctx, request.GetReset_()), nil |