cpu_monitor.py
Go to the documentation of this file.
00001 #!/usr/bin/env python
00002 #
00003 # Software License Agreement (BSD License)
00004 #
00005 # Copyright (c) 2017, TNO IVS, Helmond, Netherlands
00006 # All rights reserved.
00007 #
00008 # Redistribution and use in source and binary forms, with or without
00009 # modification, are permitted provided that the following conditions
00010 # are met:
00011 #
00012 #  * Redistributions of source code must retain the above copyright
00013 #    notice, this list of conditions and the following disclaimer.
00014 #  * Redistributions in binary form must reproduce the above
00015 #    copyright notice, this list of conditions and the following
00016 #    disclaimer in the documentation and/or other materials provided
00017 #    with the distribution.
00018 #  * Neither the name of the TNO IVS nor the names of its
00019 #    contributors may be used to endorse or promote products derived
00020 #    from this software without specific prior written permission.
00021 #
00022 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
00023 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
00024 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
00025 # FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
00026 # COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
00027 # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
00028 # BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
00029 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
00030 # CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
00031 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
00032 # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
00033 # POSSIBILITY OF SUCH DAMAGE.
00034 
00035 # \author Rein Appeldoorn
00036 
00037 import rospy
00038 from diagnostic_updater import DiagnosticTask, Updater
00039 from diagnostic_msgs.msg import DiagnosticStatus
00040 
00041 import psutil
00042 import socket
00043 
00044 
00045 class CpuTask(DiagnosticTask):
00046     def __init__(self, warning_percentage):
00047         DiagnosticTask.__init__(self, "CPU Information")
00048         self._warning_percentage = int(warning_percentage)
00049 
00050     def run(self, stat):
00051         cpu_percentages = psutil.cpu_percent(percpu=True)
00052         cpu_average = sum(cpu_percentages) / len(cpu_percentages)
00053 
00054         stat.add("CPU Load Average", cpu_average)
00055 
00056         warn = False
00057         for idx, val in enumerate(cpu_percentages):
00058             stat.add("CPU {} Load".format(idx), "{}".format(val))
00059             if val > self._warning_percentage:
00060                 warn = True
00061 
00062         if warn:
00063             stat.summary(DiagnosticStatus.WARN, "At least one CPU exceeds %d percent" % self._warning_percentage)
00064         else:
00065             stat.summary(DiagnosticStatus.OK, "CPU Average %.1f percent" % cpu_average)
00066 
00067         return stat
00068 
00069 
00070 def main():
00071     hostname = socket.gethostname()
00072     rospy.init_node('cpu_monitor_%s' % hostname.replace("-", "_"))
00073 
00074     updater = Updater()
00075     updater.setHardwareID(hostname)
00076     updater.add(CpuTask(rospy.get_param("~warning_percentage", 90)))
00077 
00078     rate = rospy.Rate(rospy.get_param("~rate", 1))
00079     while not rospy.is_shutdown():
00080         rate.sleep()
00081         updater.update()
00082 
00083 
00084 if __name__ == '__main__':
00085     main()


diagnostic_common_diagnostics
Author(s): Brice Rebsamen
autogenerated on Tue Mar 26 2019 03:09:46