# -*- encoding: utf-8 -*-
#
# Copyright © 2012 eNovance <licensing@enovance.com>
# Copyright © 2012 Red Hat, Inc
#
# Author: Julien Danjou <julien@danjou.info>
# Author: Eoghan Glynn <eglynn@redhat.com>
#
# 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 ceilometer import counter
from ceilometer.compute import plugin
from ceilometer.compute.pollsters import util
from ceilometer.openstack.common import log

LOG = log.getLogger(__name__)


class CPUPollster(plugin.ComputePollster):

    def get_counters(self, manager, cache, instance):
        LOG.info('checking instance %s', instance.id)
        instance_name = util.instance_name(instance)
        try:
            cpu_info = manager.inspector.inspect_cpus(instance_name)
            LOG.info("CPUTIME USAGE: %s %d",
                     instance.__dict__, cpu_info.time)
            cpu_num = {'cpu_number': cpu_info.number}
            yield util.make_counter_from_instance(
                instance,
                name='cpu',
                type=counter.TYPE_CUMULATIVE,
                unit='ns',
                volume=cpu_info.time,
                additional_metadata=cpu_num,
            )
        except Exception as err:
            LOG.error('could not get CPU time for %s: %s',
                      instance.id, err)
            LOG.exception(err)