Class: SidekiqAutoscale::KubernetesAdapter

Inherits:
Object
  • Object
show all
Defined in:
lib/sidekiq_autoscale/adapters/kubernetes_adapter.rb

Overview

Scale a Kubernetes deployment object.

Instance Method Summary collapse

Constructor Details

#initialize ⇒ KubernetesAdapter

Returns a new instance of KubernetesAdapter.



6
7
8
9
10
11
12
13
14
 
# File 'lib/sidekiq_autoscale/adapters/kubernetes_adapter.rb', line 6

def initialize
  require "k8s-ruby"

  namespace = File.read("/run/secrets/kubernetes.io/serviceaccount/namespace")
  client = K8s::Client.autoconfig

  @deployment_name = SidekiqAutoscale.adapter_config[:deployment_name]
  @resources = client.api("apps/v1").resource("deployments", namespace: namespace)
end

Instance Method Details

#worker_count ⇒ Object 



16
17
18
19
20
21
 
# File 'lib/sidekiq_autoscale/adapters/kubernetes_adapter.rb', line 16

def worker_count
  @resources.get(@deployment_name).spec.replicas
rescue Excon::Errors::Error, K8s::Error, K8s::Error::Forbidden => e
  SidekiqAutoscale.on_scaling_error(e)
  0
end

#worker_count=(val) ⇒ Object 



23
24
25
26
27
28
29
30
 
# File 'lib/sidekiq_autoscale/adapters/kubernetes_adapter.rb', line 23

def worker_count=(val)
  return if val == worker_count

  SidekiqAutoscale.logger.info("[SIDEKIQ_SCALE][KUBERNETES_ACTION] Setting new worker count to #{val} (is currenly #{worker_count})")
  @resources.merge_patch(@deployment_name, spec: {replicas: val})
rescue Excon::Errors::Error, K8s::Error, K8s::Error::Forbidden => e
  SidekiqAutoscale.on_scaling_error(e)
end