6add8c4fbf
Added method for creating array of samples to samples manager. Added shell command for creating array of samples. Added tests for this changes. It is made for Rally tests that use big data arrays. So now only one function may be called instead of numerous calls simple create function. Change-Id: I01ba4f0c0db40c95ef72dadaa34b3fafc034e417
103 lines
3.4 KiB
Python
103 lines
3.4 KiB
Python
#
|
|
# 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.
|
|
|
|
from ceilometerclient.common import base
|
|
from ceilometerclient.v2 import options
|
|
|
|
CREATION_ATTRIBUTES = ('source',
|
|
'counter_name',
|
|
'counter_type',
|
|
'counter_unit',
|
|
'counter_volume',
|
|
'user_id',
|
|
'project_id',
|
|
'resource_id',
|
|
'timestamp',
|
|
'resource_metadata')
|
|
|
|
|
|
class OldSample(base.Resource):
|
|
"""Represents API v2 OldSample object.
|
|
|
|
Model definition:
|
|
http://docs.openstack.org/developer/ceilometer/webapi/v2.html#OldSample
|
|
"""
|
|
def __repr__(self):
|
|
return "<OldSample %s>" % self._info
|
|
|
|
|
|
class OldSampleManager(base.Manager):
|
|
resource_class = OldSample
|
|
|
|
@staticmethod
|
|
def _path(counter_name=None):
|
|
return '/v2/meters/%s' % counter_name if counter_name else '/v2/meters'
|
|
|
|
def list(self, meter_name=None, q=None, limit=None):
|
|
path = self._path(counter_name=meter_name)
|
|
params = ['limit=%s' % str(limit)] if limit else None
|
|
return self._list(options.build_url(path, q, params))
|
|
|
|
def create(self, **kwargs):
|
|
new = dict((key, value) for (key, value) in kwargs.items()
|
|
if key in CREATION_ATTRIBUTES)
|
|
url = self._path(counter_name=kwargs['counter_name'])
|
|
body = self.api.post(url, json=[new]).json()
|
|
if body:
|
|
return [OldSample(self, b) for b in body]
|
|
|
|
def create_list(self, sample_list=None, **kwargs):
|
|
sample_dict = {}
|
|
|
|
for sample_body in sample_list:
|
|
sample = dict((key, value) for (key, value) in sample_body.items()
|
|
if key in CREATION_ATTRIBUTES)
|
|
sample_dict.setdefault(
|
|
sample_body["counter_name"], []
|
|
).append(sample)
|
|
|
|
sample_return_list = []
|
|
|
|
for (counter_name, sample_body) in sample_dict.items():
|
|
url = self._path(counter_name=counter_name)
|
|
body = self.api.post(url, json=sample_body).json()
|
|
|
|
if body:
|
|
sample_return_list.extend([OldSample(self, b) for b in body])
|
|
return sample_return_list
|
|
|
|
|
|
class Sample(base.Resource):
|
|
"""Represents API v2 Sample object.
|
|
|
|
Model definition:
|
|
http://docs.openstack.org/developer/ceilometer/webapi/v2.html#Sample
|
|
"""
|
|
def __repr__(self):
|
|
return "<Sample %s>" % self._info
|
|
|
|
|
|
class SampleManager(base.Manager):
|
|
resource_class = Sample
|
|
|
|
def list(self, q=None, limit=None):
|
|
params = ['limit=%s' % str(limit)] if limit else None
|
|
return self._list(options.build_url("/v2/samples", q, params))
|
|
|
|
def get(self, sample_id):
|
|
path = "/v2/samples/" + sample_id
|
|
try:
|
|
return self._list(path, expect_single=True)[0]
|
|
except IndexError:
|
|
return None
|