blob: 629c62a14e66cb1a2ebc1d4b00ac3658b4d8fc41 (
plain)
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
60
61
62
63
|
#!/usr/bin/env ruby
require 'tempfile'
require 'socket'
ADDR = 'socket'.freeze
def main(gitaly_dir)
gitaly_dir = File.realpath(gitaly_dir)
Dir.mktmpdir do |dir|
Dir.chdir(dir)
File.write('config.toml', <<~CONFIG
socket_path = "#{ADDR}"
bin_dir = "#{gitaly_dir}"
[[storage]]
name = "default"
path = "#{dir}"
[gitaly-ruby]
dir = "#{gitaly_dir}/ruby"
[gitlab-shell]
dir = "#{gitaly_dir}"
CONFIG
)
pid = nil
begin
start = Time.now
pid = spawn(File.join(gitaly_dir, 'gitaly'), 'config.toml')
wait_connect
puts
puts "\n\nconnection established after #{Time.now - start} seconds\n\n"
ensure
if pid
Process.kill("KILL", pid)
Process.wait(pid)
end
end
end
end
def wait_connect
loop do
begin
Socket.unix(ADDR)
return
rescue # rubocop:disable Lint/RescueWithoutErrorClass
print '.'
sleep(0.1)
end
end
end
unless ARGV.count == 1
abort "Usage: #{$PROGRAM_NAME} GITALY_DIR"
end
main(ARGV.first)
|