sshkit-custom-dsl 0.0.6 → 0.0.7
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/.gitignore +1 -0
- data/.rubocop_todo.yml +11 -0
- data/README.md +33 -4
- data/Rakefile +3 -4
- data/lib/core_ext/sshkit/backend/abstract.rb +4 -11
- data/lib/core_ext/sshkit.rb +1 -1
- data/lib/sshkit/custom/config/store.rb +11 -10
- data/lib/sshkit/custom/dsl/config_statements.rb +14 -9
- data/lib/sshkit/custom/dsl/exec_statements.rb +2 -3
- data/lib/sshkit/custom/dsl/helper.rb +3 -3
- data/lib/sshkit/custom/dsl/version.rb +1 -1
- data/lib/sshkit/custom/dsl.rb +4 -4
- data/lib/sshkit/custom/runner/abstract.rb +88 -0
- data/lib/sshkit/custom/runner/group.rb +34 -0
- data/lib/sshkit/custom/runner/parallel.rb +30 -0
- data/lib/sshkit/custom/runner/sequential.rb +14 -0
- data/spec/spec_helper.rb +0 -2
- data/spec/unit/sshkit/custom/config/store_spec.rb +116 -0
- data/spec/unit/sshkit/custom/dsl/config_statements_spec.rb +147 -0
- data/spec/unit/sshkit/custom/dsl/exec_statements_spec.rb +74 -0
- data/spec/unit/sshkit/custom/dsl/helper_spec.rb +57 -0
- data/spec/unit/sshkit/custom/runner/abstract_spec.rb +123 -0
- data/spec/unit/sshkit/custom/runner/group_spec.rb +48 -0
- data/spec/unit/sshkit/custom/runner/parallel_spec.rb +55 -0
- data/spec/unit/sshkit/custom/runner/sequential_spec.rb +27 -0
- data/sshkit-custom-dsl.gemspec +10 -10
- metadata +22 -6
- data/lib/sshkit/custom/config/runner/abstract.rb +0 -100
- data/lib/sshkit/custom/config/runner/group.rb +0 -31
- data/lib/sshkit/custom/config/runner/parallel.rb +0 -35
- data/lib/sshkit/custom/config/runner/sequential.rb +0 -18
|
@@ -1,100 +0,0 @@
|
|
|
1
|
-
module SSHKit
|
|
2
|
-
module Custom
|
|
3
|
-
module Config
|
|
4
|
-
module Runner
|
|
5
|
-
|
|
6
|
-
ExecuteError = SSHKit::Runner::ExecuteError
|
|
7
|
-
|
|
8
|
-
class Abstract
|
|
9
|
-
attr_accessor :backends
|
|
10
|
-
attr_reader :options
|
|
11
|
-
attr_writer :wait_interval
|
|
12
|
-
|
|
13
|
-
def self.create_runner(opts)
|
|
14
|
-
opts_with_defaults = { in: :parallel }.merge(opts)
|
|
15
|
-
|
|
16
|
-
case opts_with_defaults[:in]
|
|
17
|
-
when :parallel then Parallel
|
|
18
|
-
when :sequence then Sequential
|
|
19
|
-
when :groups then Group
|
|
20
|
-
else
|
|
21
|
-
raise RuntimeError, "Don't know how to handle run style #{opts_with_defaults[:in].inspect}"
|
|
22
|
-
end.new(opts_with_defaults)
|
|
23
|
-
|
|
24
|
-
end
|
|
25
|
-
|
|
26
|
-
def self.scope_storage
|
|
27
|
-
ScopedStorage::ThreadLocalStorage
|
|
28
|
-
end
|
|
29
|
-
|
|
30
|
-
def self.scope
|
|
31
|
-
@scope ||= ScopedStorage::Scope.new('sshkit_runner', scope_storage)
|
|
32
|
-
end
|
|
33
|
-
|
|
34
|
-
def self.active_backend
|
|
35
|
-
scope[:active_backend]
|
|
36
|
-
end
|
|
37
|
-
|
|
38
|
-
def self.active_backend=(new_backend)
|
|
39
|
-
scope[:active_backend]=new_backend
|
|
40
|
-
end
|
|
41
|
-
|
|
42
|
-
def initialize(options = nil)
|
|
43
|
-
@options = options || {}
|
|
44
|
-
end
|
|
45
|
-
|
|
46
|
-
def active_backend
|
|
47
|
-
self.class.active_backend
|
|
48
|
-
end
|
|
49
|
-
|
|
50
|
-
def active_backend=(new_backend)
|
|
51
|
-
self.class.active_backend=new_backend
|
|
52
|
-
end
|
|
53
|
-
|
|
54
|
-
def send_cmd(cmd, *args, &block)
|
|
55
|
-
|
|
56
|
-
begin
|
|
57
|
-
|
|
58
|
-
if block
|
|
59
|
-
args = Array(block.call(active_backend.host))
|
|
60
|
-
end
|
|
61
|
-
|
|
62
|
-
active_backend.send(cmd, *args)
|
|
63
|
-
|
|
64
|
-
rescue Exception => e
|
|
65
|
-
e2 = ExecuteError.new e
|
|
66
|
-
raise e2, "Exception while executing on host #{active_backend.host}: #{e.message}"
|
|
67
|
-
end
|
|
68
|
-
|
|
69
|
-
end
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
def apply_to_bck(backend, &block)
|
|
73
|
-
begin
|
|
74
|
-
|
|
75
|
-
self.active_backend = backend
|
|
76
|
-
block.call(backend.host)
|
|
77
|
-
|
|
78
|
-
rescue Exception => e
|
|
79
|
-
e2 = ExecuteError.new e
|
|
80
|
-
raise e2, "Exception while executing on host #{backend.host}: #{e.message}"
|
|
81
|
-
ensure
|
|
82
|
-
self.active_backend = nil
|
|
83
|
-
end
|
|
84
|
-
end
|
|
85
|
-
|
|
86
|
-
protected
|
|
87
|
-
|
|
88
|
-
def do_wait
|
|
89
|
-
sleep wait_interval
|
|
90
|
-
end
|
|
91
|
-
|
|
92
|
-
def wait_interval
|
|
93
|
-
@wait_interval || options[:wait] || 2
|
|
94
|
-
end
|
|
95
|
-
|
|
96
|
-
end
|
|
97
|
-
end
|
|
98
|
-
end
|
|
99
|
-
end
|
|
100
|
-
end
|
|
@@ -1,31 +0,0 @@
|
|
|
1
|
-
module SSHKit
|
|
2
|
-
module Custom
|
|
3
|
-
module Config
|
|
4
|
-
module Runner
|
|
5
|
-
|
|
6
|
-
class Group < Abstract
|
|
7
|
-
attr_writer :group_size
|
|
8
|
-
|
|
9
|
-
def apply_block_to_bcks(&block)
|
|
10
|
-
backends.each_slice(group_size).collect do |group_backends|
|
|
11
|
-
|
|
12
|
-
Parallel.new(options).tap do |runner|
|
|
13
|
-
runner.backends = group_backends
|
|
14
|
-
runner.apply_block_to_bcks(&block)
|
|
15
|
-
end
|
|
16
|
-
|
|
17
|
-
do_wait
|
|
18
|
-
|
|
19
|
-
end.flatten
|
|
20
|
-
end
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
def group_size
|
|
24
|
-
@group_size ||= options[:limit] || 2
|
|
25
|
-
end
|
|
26
|
-
|
|
27
|
-
end
|
|
28
|
-
end
|
|
29
|
-
end
|
|
30
|
-
end
|
|
31
|
-
end
|
|
@@ -1,35 +0,0 @@
|
|
|
1
|
-
module SSHKit
|
|
2
|
-
module Custom
|
|
3
|
-
module Config
|
|
4
|
-
module Runner
|
|
5
|
-
|
|
6
|
-
require 'rake'
|
|
7
|
-
|
|
8
|
-
class Parallel < Abstract
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
def thread_pool
|
|
13
|
-
@thread_pool ||= Rake::ThreadPool.new(thread_count)
|
|
14
|
-
end
|
|
15
|
-
|
|
16
|
-
def apply_block_to_bcks(&block)
|
|
17
|
-
|
|
18
|
-
futures = backends.map do |b|
|
|
19
|
-
thread_pool.future(b) do |fb|
|
|
20
|
-
apply_to_bck fb, &block
|
|
21
|
-
end
|
|
22
|
-
end
|
|
23
|
-
|
|
24
|
-
futures.each { |f| f.value }
|
|
25
|
-
end
|
|
26
|
-
|
|
27
|
-
def thread_count
|
|
28
|
-
@thread_count ||= options[:thread_count] || Rake.suggested_thread_count-1
|
|
29
|
-
end
|
|
30
|
-
|
|
31
|
-
end
|
|
32
|
-
end
|
|
33
|
-
end
|
|
34
|
-
end
|
|
35
|
-
end
|
|
@@ -1,18 +0,0 @@
|
|
|
1
|
-
module SSHKit
|
|
2
|
-
module Custom
|
|
3
|
-
module Config
|
|
4
|
-
module Runner
|
|
5
|
-
class Sequential < Abstract
|
|
6
|
-
|
|
7
|
-
def apply_block_to_bcks(&block)
|
|
8
|
-
backends.each do |backend|
|
|
9
|
-
apply_to_bck backend, &block
|
|
10
|
-
do_wait
|
|
11
|
-
end
|
|
12
|
-
end
|
|
13
|
-
|
|
14
|
-
end
|
|
15
|
-
end
|
|
16
|
-
end
|
|
17
|
-
end
|
|
18
|
-
end
|