NEW: cfsystem_service type

andvgal · andvgal · commit 53b0cccb6aa5 · 2018-11-25T20:37:51.000+02:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,7 @@ project adheres to [Semantic Versioning](http://semver.org/).
 - CHANGED: updated for Ubuntu 18.04 Bionic support
 - NEW: cfsystem::add_handy_tools to control additional package setup
 - NEW: FreeIPA NTP support
+- NEW: cfsystem_service type
 
 ## 1.1.0 (2018-05-02)
 - NEW: cfsystem::metric type as a sort of IoC to integrate cfmetrics
diff --git a/README.md b/README.md
@@ -338,3 +338,7 @@ Therefore, this drop-in replacement has been provided.
 [cfnetwork]: https://codingfuture.net/docs/cfnetwork
 [cfauth]: https://codingfuture.net/docs/cfauth
 [cffirehol]: https://codingfuture.net/docs/cffirehol
+
+## `cfsystem_service` type
+
+Helper type to create cfsystem-integrated services.
diff --git a/lib/puppet/provider/cfsystem_service/cfprovider.rb b/lib/puppet/provider/cfsystem_service/cfprovider.rb
@@ -0,0 +1,114 @@
+#
+# Copyright 2018 (c) Andrey Galkin
+#
+
+
+begin
+    require File.expand_path( '../../../../puppet_x/cf_system', __FILE__ )
+rescue LoadError
+    require File.expand_path( '../../../../../../cfsystem/lib/puppet_x/cf_system', __FILE__ )
+end
+
+Puppet::Type.type(:cfsystem_service).provide(
+    :cfprov,
+    :parent => PuppetX::CfSystem::ProviderBase
+) do
+    desc "Provider for cfsystem_service"
+    
+    commands :systemctl => PuppetX::CfSystem::SYSTEMD_CTL
+        
+    def self.get_config_index
+        'cf91service'
+    end
+
+    def self.get_generator_version
+        cf_system().makeVersion(__FILE__)
+    end
+    
+    def self.check_exists(params)
+        debug("check_exists: #{params}")
+        begin
+            systemctl(['status', "#{params[:name]}.service"])
+        rescue => e
+            warning(e)
+            #warning(e.backtrace)
+            false
+        end
+    end
+
+    def self.on_config_change(newconf)
+        debug('on_config_change')
+
+        new_services = []
+
+        newconf.each do |name, conf|
+            new_services << name
+
+            begin
+                self.send("create_svc", name, conf)
+            rescue => e
+                warning(e)
+                #warning(e.backtrace)
+                err("Transition error in setup")
+            end
+        end
+ 
+        begin
+            cf_system.cleanupSystemD("cfsvc-", new_services)
+        rescue => e
+            warning(e)
+            #warning(e.backtrace)
+            err("Transition error in setup")
+        end
+    end
+
+    def self.create_svc(service_name, conf)
+        debug('on_config_change')
+        
+        user = conf[:user]
+        root_dir = conf[:root_dir]
+        settings_tune = (conf[:settings_tune] or {})
+        command = conf[:command]
+        allow_restart = conf[:allow_restart]
+        
+        avail_mem = cf_system.getMemory(service_name)
+
+        # Service File
+        #==================================================
+        content_ini = {
+            'Unit' => {
+                'Description' => "Service: #{service_name}",
+            },
+            'Service' => {
+                'ExecStart' => command,
+                'LimitNOFILE' => '16384',
+                'WorkingDirectory' => root_dir,
+            },
+        }
+        
+        settings_tune.each { |k, v|
+            content_ini.setdefault(k, {})
+            content_ini[k].merge! v
+        }
+
+        service_changed = self.cf_system().createService({
+            :service_name => service_name,
+            :user => user,
+            :content_ini => content_ini,
+            :cpu_weight => conf[:cpu_weight],
+            :io_weight => conf[:io_weight],
+            :mem_limit => avail_mem,
+            :mem_lock => true,
+        })
+
+        #==================================================
+        
+        if service_changed and allow_restart
+            warning(">> reloading #{service_name}")
+            systemctl('restart', "#{service_name}.service")
+        else
+            systemctl('start', "#{service_name}.service")
+        end        
+    end
+end
+
diff --git a/lib/puppet/type/cfsystem_service.rb b/lib/puppet/type/cfsystem_service.rb
@@ -0,0 +1,88 @@
+#
+# Copyright 2018 (c) Andrey Galkin
+#
+
+Puppet::Type.newtype(:cfsystem_service) do
+    desc "DO NOT USE DIRECTLY."
+    
+    autorequire(:cfsystem_flush_config) do
+        ['begin']
+    end
+    autorequire(:cfsystem_memory_calc) do
+        ['total']
+    end
+    autonotify(:cfsystem_flush_config) do
+        ['commit']
+    end
+    
+    ensurable do
+        defaultvalues
+        defaultto :absent
+    end
+    
+    
+    newparam(:name) do
+        isnamevar
+        
+        validate do |value|
+            unless value =~ /^[a-z0-9_@-]+$/i
+                raise ArgumentError, "%s is not a valid service name" % value
+            end
+        end
+    end
+    
+    newproperty(:user) do
+        validate do |value|
+            unless value =~ /^[a-zA-Z_][a-zA-Z0-9_-]*$/
+                raise ArgumentError, "%s is not valid user" % value
+            end
+        end
+    end
+    
+    newproperty(:root_dir) do
+        validate do |value|
+            unless value =~ /^(\/[a-z0-9_]+)+$/i
+                raise ArgumentError, "%s is not a valid path" % value
+            end
+        end
+    end
+
+    newproperty(:memory_weight) do
+        validate do |value|
+            unless value.is_a? Integer and value > 0
+                raise ArgumentError, "%s is not a valid positive integer" % value
+            end
+        end
+    end
+
+    newproperty(:cpu_weight) do
+        validate do |value|
+            unless value.is_a? Integer and value > 0
+                raise ArgumentError, "%s is not a valid positive integer" % value
+            end
+        end
+    end
+    
+    newproperty(:io_weight) do
+        validate do |value|
+            unless value.is_a? Integer and value > 0
+                raise ArgumentError, "%s is not a valid positive integer" % value
+            end
+        end
+    end
+    
+    newproperty(:settings_tune) do
+        validate do |value|
+            value.is_a? Hash
+        end
+    end
+
+    newproperty(:command) do
+        isrequired
+    end
+    
+    newproperty(:allow_restart) do
+        defaultto true
+    end
+end
+
