21 lines
728 B
Ruby
21 lines
728 B
Ruby
|
namespace :gitlab do
|
||
|
|
desc "GitLab | Check TCP connectivity to a specific host and port"
|
||
|
|
task :tcp_check, [:host, :port] => :environment do |_t, args|
|
||
|
|
unless args.host && args.port
|
||
|
|
puts "Please specify a host and port: `rake gitlab:tcp_check[example.com,80]`".color(:red)
|
||
|
|
exit 1
|
||
|
|
end
|
||
|
|
|
||
|
|
checker = Gitlab::TcpChecker.new(args.host, args.port)
|
||
|
|
|
||
|
|
if checker.check
|
||
|
|
puts "TCP connection from #{checker.local} to #{checker.remote} succeeded".color(:green)
|
||
|
|
else
|
||
|
|
puts "TCP connection to #{checker.remote} failed: #{checker.error}".color(:red)
|
||
|
|
puts
|
||
|
|
puts 'Check that host and port are correct, and that the traffic is permitted through any firewalls.'
|
||
|
|
exit 1
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|