#!/usr/bin/env vpython3

# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
""" A metric implementation to count the number of inputs. """

from measure import Measure
from test_script_metrics_pb2 import TestScriptMetric


class Count(Measure):

  def __init__(self, name: str) -> None:
    self._name = name
    self._count = 0

  def record(self) -> None:
    self._count += 1

  def dump(self) -> TestScriptMetric:
    result = TestScriptMetric()
    result.name = self._name
    result.value = self._count
    return result