Class: Util

Inherits:
Object
  • Object
show all
Defined in:
lib/common/socket/util.rb

Overview

============================================================ 基础库辅助类

Author

chenjie

Data

2009-4-30

============================================================

Class Method Summary collapse

Class Method Details

.change_in_local(host, remote_path, user = USERNAME, password = PASSWORD) {|local_path| ... } ⇒ Object

===功能 获取文件，操作文件，放回文件 ===example Util.change_in_local host, remotepath

Yields:

  • (local_path) 


231
232
233
234
235
236
237
238
 
# File 'lib/common/socket/util.rb', line 231

def Util.change_in_local host, remote_path, user=USERNAME, password=PASSWORD
	sfix = Util.suffix
	local_path="/tmp/localfile.#{sfix}"
	Util.download host, remote_path, local_path, user, password
	yield local_path
	Util.upload host, local_path, remote_path, user, password
	File.delete(local_path)
end

.download(host, remote_path, local_path, user = USERNAME, password = PASSWORD) ⇒ Object

===功能 scp远端文件到本机 ===example Util.download host, remote, local



155
156
157
158
159
 
# File 'lib/common/socket/util.rb', line 155

def Util.download host, remote_path, local_path, user=USERNAME, password=PASSWORD
	Net::SCP.start(host, user, :password=>password) do | scp |
		scp.download! remote_path, local_path
	end
end

.exe_ssh_cmd(host, cmd_str, user = USERNAME, password = PASSWORD) ⇒ Object

===功能 在远程机器上执行命令 封装Net::SSH::Connection::Session.exe "Note that this method returns immediately, and requires an event loop (see Session#loop) in order for the command to actually execute." ===example Util.exe_ssh_cmd host, "ls"



130
131
132
133
134
135
136
137
138
139
140
 
# File 'lib/common/socket/util.rb', line 130

def Util.exe_ssh_cmd host, cmd_str, user=USERNAME, password=PASSWORD
 		ssh_cmd = "source ~/.bash_profile && #{cmd_str}"
	$log.debug "SSH CMD = #{ssh_cmd}"
		res = ''
		Net::SSH.start(host, user, :password=>password) do |session|
res = session.exec ssh_cmd
		end
		res = res.to_s.strip.chomp
		$log.debug "SSH CMD RES = #{res}"
		return res
end

.exe_ssh_cmd!(host, cmd_str, user = USERNAME, password = PASSWORD) ⇒ Object

===功能 在远程机器上执行命令 封装Net::SSH::Connection::Session.exe! Same as exec, except this will block until the command finishes ===example Util.exe_ssh_cmd! host, "killall -9 friend"



112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/common/socket/util.rb', line 112

def Util.exe_ssh_cmd! host, cmd_str, user=USERNAME, password=PASSWORD
 		ssh_cmd = "source ~/.bash_profile && #{cmd_str}"
	$log.debug "SSH CMD = #{ssh_cmd}"
		res = ''
		Net::SSH.start(host, user, :password=>password) do |session|
res = session.exec! ssh_cmd
		end
		res = res.to_s.strip.chomp
		$log.debug "SSH CMD RES = #{res}"
		return res
end

.get_lib_path(host, user = USERNAME, password = PASSWORD) ⇒ Object

===功能 返回指定机器的lib路径, 执行staf命令相关

  • ip 目标机器ip, string return string, 目标机器上部署的lib路径 ===example Util.get_lib_path host


95
96
97
98
99
100
101
102
103
104
 
# File 'lib/common/socket/util.rb', line 95

def Util.get_lib_path host, user=USERNAME, password=PASSWORD
      if Util.is_localhost? host then 
    		return LIB_ROOT
  	else
    		cmd_str = "echo $HOME"
        	home = Util.exe_ssh_cmd! host, cmd_str, user, password
        	#return "#{home}/tools/app-test/search/space/module-test/lib/"
        	return "#{home}/tools/rlib/"
  	end
end

.get_local_ip ⇒ Object

===功能 返回本机IP（非127.0.0.1） ===example lip = Util.get_local_ip



146
147
148
149
 
# File 'lib/common/socket/util.rb', line 146

def Util.get_local_ip
	ip = `hostname -i`
	return ip
end

.is_localhost?(ip) ⇒ Boolean

===功能 判断传入ip是否是本机ip return: true/false ===example Util.is_localhost? "10.4.34.23"

Returns:

  • (Boolean) 


21
22
23
24
25
26
27
28
29
30
 
# File 'lib/common/socket/util.rb', line 21

def Util.is_localhost? ip
	#get localhost ip
	local_ip = `hostname -i`.chomp

  	if ( (ip == nil) || (ip == local_ip) || (ip == "127.0.0.1") || (ip == "localhost") ) then
      		return true
	else
		return false
  	end
end

.scpfromto(host_a, remote_path_a, host_b, remote_path_b, user_a, password_a, user_b, password_b) ⇒ Object

===功能 scp文件从机器a到机器b 如果远程多层子目录不存在则自动的进行创建 ===example Util.scpfromto hosta, remotea, hostb, remoteb, usera, passa, userb, passb



176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
 
# File 'lib/common/socket/util.rb', line 176

def Util.scpfromto host_a, remote_path_a, host_b, remote_path_b, user_a, password_a, user_b, password_b
	sfix = Util.suffix
	local_path="/tmp/localfile.#{sfix}"
	Net::SCP.start(host_a, user_a, :password=>password_a) do | scp |
		scp.download! remote_path_a, local_path do |ch, name, sent, total|
			print "."
		end
		puts ""
	end
	if(remote_path_b=~/(\/.*)\/[^\/]*/)
		puts "Make sure dir exist #{$1}"
		`mkdir -p #{$1}`
	end
	Net::SCP.start(host_b, user_b, :password=>password_b) do | scp |
		scp.upload! local_path, remote_path_b do |ch, name, sent, total|
			print "."
		end
		puts ""
	end
	File.delete(local_path)
end

.scpfromtod(host_a, remote_path_a, host_b, remote_path_b, user_a, password_a, user_b, password_b) ⇒ Object

===功能 scp文件夹从机器a到机器b 如果远程多层子目录不存在则自动的进行创建 ===example Util.scpfromtod hosta, remotea, hostb, remoteb, usera, passa, userb, passb



203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
 
# File 'lib/common/socket/util.rb', line 203

def Util.scpfromtod host_a, remote_path_a, host_b, remote_path_b, user_a, password_a, user_b, password_b
	sfix = Util.suffix
	dirname = `basename #{remote_path_b}`.chop!
	local_path="/tmp/localfile.#{sfix}"
	Net::SCP.start(host_a, user_a, :password=>password_a) do | scp |
		scp.download! remote_path_a, local_path, :recursive=>"yes" do |ch, name, sent, total|
			print "."
		end
		puts ""
	end
	if(remote_path_b=~/(\/.*)\/[^\/]*/)
		puts "Make sure dir exist #{$1}"
		`mkdir -p #{$1}`
	end
	Net::SCP.start(host_b, user_b, :password=>password_b) do | scp |
		scp.upload! local_path+"/#{dirname}", remote_path_b, :recursive=>"yes" do |ch, name, sent, total|
			print "."
		end
		puts ""
	end
	#Dir.rmdir(local_path)
	`rm -fr #{local_path}`
end

.suffix ⇒ Object

===功能 获取全局唯一的后缀 ===example su = Util.suffix



244
245
246
 
# File 'lib/common/socket/util.rb', line 244

def Util.suffix
	return "#{Time.now.tv_sec}#{Time.now.tv_usec}"
end

.upload(host, local_path, remote_path, user = USERNAME, password = PASSWORD) ⇒ Object

===功能 scp本机文件到远端 ===example Util.upload host, local, remote



165
166
167
168
169
 
# File 'lib/common/socket/util.rb', line 165

def Util.upload host, local_path, remote_path, user=USERNAME, password=PASSWORD
	Net::SCP.start(host, user, :password=>password) do | scp |
		scp.upload! local_path, remote_path
	end
end