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
 |