tools: Add tools to quickly add measurements
Change-Id: I743f0b33f014411d7ac07d0c11f40870413bed27
This commit is contained in:
parent
b994a8668e
commit
996ade6c8a
@ -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"
|
||||
|
60
tools/measures_injector.py
Executable file
60
tools/measures_injector.py
Executable file
@ -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()
|
Loading…
Reference in New Issue
Block a user