/
cuda.pp
59 lines (52 loc) · 1.96 KB
/
cuda.pp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# Class: collectd::plugin::cuda
#
# @see https://pypi.python.org/pypi/collectd-cuda
#
# Configures cuda metrics collection. Optionally installs the plugin
# Note, it is up to you to support package installation and sources
#
# @param ensure Optional[String] Passed to package and collectd::plugin resources (both). Default: present
# @param manage_package Optional[Boolean] Toggles installation of plugin. Default: undef
# @param package_name Optional[String] Name of plugin package to install. Default: collectd-cuda
# @param package_provider Optional[String] Passed to package resource. Default: pip
# @param provider_proxy Optional[String] Proxy for provider. Default: undef
class collectd::plugin::cuda (
Optional[String] $ensure = 'present',
Optional[Boolean] $manage_package = undef,
Optional[String] $package_name = 'collectd-cuda',
Optional[String] $package_provider = 'pip',
Optional[String] $provider_proxy = undef,
) {
include ::collectd
$_manage_package = pick($manage_package, $::collectd::manage_package)
if ($_manage_package) {
if (!defined(Package['python-pip'])) {
package { 'python-pip': ensure => 'present', }
Package[$package_name] {
require => Package['python-pip'],
}
if $facts['os']['family'] == 'RedHat' {
# Epel is installed in install.pp if manage_repo is true
# python-pip doesn't exist in base for RedHat. Need epel installed first
if (defined(Class['::epel'])) {
Package['python-pip'] {
require => Class['::epel'],
}
}
}
}
}
if ($_manage_package) and ($provider_proxy) {
$install_options = [{'--proxy' => $provider_proxy}]
} else {
$install_options = undef
}
package { $package_name:
ensure => $ensure,
provider => $package_provider,
install_options => $install_options,
}
collectd::plugin::python::module { 'collectd_cuda.collectd_plugin':
ensure => $ensure,
}
}