Add a simple stats framework to gRPC C core

1 files changed, 102 insertions, 0 deletions

diff --git a/tools/codegen/core/gen_stats_data.py b/tools/codegen/core/gen_stats_data.py
new file mode 100755
index 0000000000..bc601a89a7
--- /dev/null
+++ b/tools/codegen/core/gen_stats_data.py
@@ -0,0 +1,102 @@
+#!/usr/bin/env python2.7
+
+# Copyright 2017 gRPC authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import collections
+import sys
+import yaml
+
+with open('src/core/lib/debug/stats_data.yaml') as f:
+  attrs = yaml.load(f.read())
+
+Counter = collections.namedtuple('Counter', 'name')
+
+counters = []
+
+for attr in attrs:
+  if 'counter' in attr:
+    counters.append(Counter(name=attr['counter']))
+  else:
+    print 'Error: bad attr %r' % attr
+
+# utility: print a big comment block into a set of files
+def put_banner(files, banner):
+  for f in files:
+    print >>f, '/*'
+    for line in banner:
+      print >>f, ' * %s' % line
+    print >>f, ' */'
+    print >>f
+
+with open('src/core/lib/debug/stats_data.h', 'w') as H:
+  # copy-paste copyright notice from this file
+  with open(sys.argv[0]) as my_source:
+    copyright = []
+    for line in my_source:
+      if line[0] != '#': break
+    for line in my_source:
+      if line[0] == '#':
+        copyright.append(line)
+        break
+    for line in my_source:
+      if line[0] != '#':
+        break
+      copyright.append(line)
+    put_banner([H], [line[2:].rstrip() for line in copyright])
+
+  put_banner([H], ["Automatically generated by tools/codegen/core/gen_stats_data.py"])
+
+  print >>H, "#ifndef GRPC_CORE_LIB_DEBUG_STATS_DATA_H"
+  print >>H, "#define GRPC_CORE_LIB_DEBUG_STATS_DATA_H"
+  print >>H
+
+  print >>H, "typedef enum {"
+  for ctr in counters:
+    print >>H, "  GRPC_STATS_COUNTER_%s," % ctr.name.upper()
+  print >>H, "  GRPC_STATS_COUNTER_COUNT"
+  print >>H, "} grpc_stats_counters;"
+
+  for ctr in counters:
+    print >>H, "#define GRPC_STATS_INC_%s(exec_ctx) GRPC_STATS_INC_COUNTER((exec_ctx), GRPC_STATS_COUNTER_%s)" % (ctr.name.upper(), ctr.name.upper())
+
+  print >>H, "extern const char *grpc_stats_counter_name[GRPC_STATS_COUNTER_COUNT];"
+
+  print >>H
+  print >>H, "#endif /* GRPC_CORE_LIB_DEBUG_STATS_DATA_H */"
+
+with open('src/core/lib/debug/stats_data.c', 'w') as C:
+  # copy-paste copyright notice from this file
+  with open(sys.argv[0]) as my_source:
+    copyright = []
+    for line in my_source:
+      if line[0] != '#': break
+    for line in my_source:
+      if line[0] == '#':
+        copyright.append(line)
+        break
+    for line in my_source:
+      if line[0] != '#':
+        break
+      copyright.append(line)
+    put_banner([C], [line[2:].rstrip() for line in copyright])
+
+  put_banner([C], ["Automatically generated by tools/codegen/core/gen_stats_data.py"])
+
+  print >>C, "#include \"src/core/lib/debug/stats_data.h\""
+
+  print >>C, "const char *grpc_stats_counter_name[GRPC_STATS_COUNTER_COUNT] = {";
+  for ctr in counters:
+    print >>C, "  \"%s\"," % ctr.name
+  print >>C, "};"