tools: Add tools to quickly add measurements

Change-Id: I743f0b33f014411d7ac07d0c11f40870413bed27

devstack/gate/post_test_hook.sh

@@ -43,6 +43,9 @@ export GNOCCHI_SERVICE_URL=$(openstack catalog show metric -c endpoints -f value
 
 curl -X GET ${GNOCCHI_SERVICE_URL}/v1/archive_policy -H "Content-Type: application/json"
 
+# Just ensure tools still works
+gnocchi metric create
+sudo -E -H -u stack $GNOCCHI_DIR/tools/measures_injector.py --metrics 1 --batch-of-measures 2 --measures-per-batch 2
 
 # Run tests
 echo "Running gnocchi functional test suite"

tools/measures_injector.py

@@ -0,0 +1,60 @@
+#!/usr/bin/env python
+# Copyright (c) 2016 Red Hat
+#
+# 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 datetime
+import random
+
+from concurrent import futures
+from oslo_config import cfg
+import six
+
+from gnocchi import indexer
+from gnocchi import service
+from gnocchi import storage
+from gnocchi import utils
+
+
+def injector():
+    conf = cfg.ConfigOpts()
+    conf.register_cli_opts([
+        cfg.IntOpt("metrics", default=None),
+        cfg.IntOpt("batch-of-measures", default=1000),
+        cfg.IntOpt("measures-per-batch", default=10),
+    ])
+    conf = service.prepare_service(conf=conf)
+    index = indexer.get_driver(conf)
+    index.connect()
+    s = storage.get_driver(conf)
+
+    metrics = index.list_metrics()
+    if conf.metrics:
+        metrics = metrics[:conf.metrics]
+
+    def todo(metric):
+        for _ in six.moves.range(conf.batch_of_measures):
+            measures = [
+                storage.Measure(utils.to_timestamp(datetime.datetime.now()),
+                                random.random())
+                for __ in six.moves.range(conf.measures_per_batch)]
+            s.add_measures(metric, measures)
+
+    with futures.ThreadPoolExecutor(max_workers=len(metrics)) as executor:
+        # We use 'list' to iterate all threads here to raise the first
+        # exception now, not much choice
+        list(executor.map(todo, metrics))
+
+
+if __name__ == '__main__':
+    injector()