Module: Env

Included in:
EnvTest, MCBasePatternHelper, Stub
Defined in:
lib/common/socket/env.rb

Overview

This class supply enviroment related local/remote operation e.g. modify configration, operate process…

Author

chenjie

Data

2009-4-30

Instance Method Summary collapse

Instance Method Details

#bak_conf(host = HOST, user = USERNAME, password = PASSWORD, conf_path = CONFPATH, context_key) ⇒ Object

bak_conf “conf1”



116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
 
# File 'lib/common/socket/env.rb', line 116

def bak_conf host=HOST, user=USERNAME, password=PASSWORD, conf_path=CONFPATH, context_key
	conf_dir=`dirname #{conf_path}`
	conf_name=`basename #{conf_path}`
	conf_dir.chop!
	conf_name.chop!
       now = Time.now.to_datetime
       bak_cmd = "cd #{conf_dir} && cp #{conf_name} #{conf_name}.#{now}"
       if Util.is_localhost? host
      		system "#{bak_cmd}"
       else
		Util.exe_ssh_cmd! host, bak_cmd, user, password
       end
       Context.set "#{context_key}", "#{conf_name}.#{now}"
       $log.debug "bak conf which context key is #{context_key}"
end

#copy_conf(host = HOST, user = USERNAME, password = PASSWORD, source_conf, target_conf) ⇒ Object

copy_conf “/home/space/space/conf/conf1”, “/home/space/space/conf/conf2”



167
168
169
170
171
172
173
174
175
 
# File 'lib/common/socket/env.rb', line 167

def copy_conf host=HOST, user=USERNAME, password=PASSWORD, source_conf, target_conf
       copy_cmd = "cp #{source_conf} #{target_conf}"
       if Util.is_localhost?(host) then
           	system "#{copy_cmd}"
       else
		Util.exe_ssh_cmd! host, copy_cmd, user, password
       end
       $log.debug "Copy conf from #{source_conf} to #{target_conf}."
end

#get_conf(host = HOST, user = USERNAME, password = PASSWORD, conf_path = CONFPATH, key) ⇒ Object

功能

获取配置文件的value

参数

  • host

  • conf_path

  • key

return if key existed, return conf value string; if not, return empty string “”

Example:

get_conf “name”

检查的方法:

assert_body pattern



85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
 
# File 'lib/common/socket/env.rb', line 85

def get_conf host=HOST, user=USERNAME, password=PASSWORD, conf_path=CONFPATH, key
	if Util.is_localhost? host
		cm = ConfModifier.new conf_path
		ret = cm.get_conf key
		Context.set("IOBODY", ret)
		return ret
      	else
  		#lib_path = Util.get_lib_path host
       		#cmd_str = "cd #{lib_path} && ruby conf_modifier.rb get_conf #{conf_path} #{key}"
		#value = Util.exe_ssh_cmd! host, cmd_str, user, password
		#Context.set("IOBODY", value)
		#return value
		Util.change_in_local(host, conf_path, user, password) { | local_path |
			cm = ConfModifier.new local_path
			ret = cm.get_conf key
			Context.set("IOBODY", ret)
		}
		return ret
      	end
end

#get_thread_num(host, module_name, user = USERNAME, password = PASSWORD) ⇒ Object

功能

获取模块进程线程数

参数

  • host

  • module_name

return thead number or 0(process not running)

example

get_thread_num host, “friend”



259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
 
# File 'lib/common/socket/env.rb', line 259

def get_thread_num host, module_name, user=USERNAME, password=PASSWORD
  if Util.is_localhost? host
	str = `pstree | grep #{module_name}`
               mdata = str.scan /.*---(\d+).*/
               if mdata.size > 0 then
                       return mdata[0][0].to_i
               else
                       return 0
               end
  else
       	cmd_str = "pstree | grep #{module_name}"
	str = Util.exe_ssh_cmd! host, cmd_str, user, password
       	mdata = str.scan /.*---(\d+).*/
       	if mdata.size > 0 then
         		return mdata[0][0].to_i
       	else
         		return 0
       	end		
  end
end

#is_process_running?(host, module_name, user = USERNAME, password = PASSWORD) ⇒ Boolean

example

is_process_running? host, “friend”

Returns:

  • (Boolean) 


232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
 
# File 'lib/common/socket/env.rb', line 232

def is_process_running? host, module_name, user=USERNAME, password=PASSWORD
 		if Util.is_localhost?(host) then
value = `pstree | grep #{module_name} | wc -l`
if value.to_i > 0 then
                             return true
                     else
                             return false
                     end
 		else
cmd_str = "pstree | grep #{module_name} | wc -l"
value = Util.exe_ssh_cmd! host, cmd_str, user, password
if value.to_i > 0 then
  			return true
		else
  			return false
	end
 		end
end

#kill_process(host, module_name, user = USERNAME, password = PASSWORD) ⇒ Object

example

kill_process host, “friend”



217
218
219
220
221
222
223
224
225
 
# File 'lib/common/socket/env.rb', line 217

def kill_process host, module_name, user=USERNAME, password=PASSWORD
	if Util.is_localhost?(host) then
`killall -9 #{module_name}`
		else
cmd_str = "killall -9 #{module_name}"
Util.exe_ssh_cmd! host, cmd_str, user, password
 		end
 		sleep 1
end

#modify_conf(host = HOST, user = USERNAME, password = PASSWORD, conf_path = CONFPATH, hash) ⇒ Object

功能

修改配置文件

参数

  • host

  • conf_path

  • hash 欲修改的配置hash

example

hh = “path”=>newpath modify_conf hh



59
60
61
62
63
64
65
66
67
68
69
 
# File 'lib/common/socket/env.rb', line 59

def modify_conf host=HOST, user=USERNAME, password=PASSWORD, conf_path=CONFPATH, hash
	if Util.is_localhost? host
   		cm = ConfModifier.new conf_path
    		cm.modifyall hash
    		cm.writeconf
      	else
        	hash.each{|k,v|
      	  		Env.modify_conf_single host, user, password, conf_path, k, v
        	}
      	end
end

#modify_conf_single(host = HOST, user = USERNAME, password = PASSWORD, conf_path = CONFPATH, key, new_value) ⇒ Object

功能

修改配置文件

参数

  • host host ip(example: 127.0.0.1 stands for localhost)

  • conf_path 配置文件路径

  • key 欲修改的key

  • new_value 新的value值

example

modify_conf_single “name”, newname



33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/common/socket/env.rb', line 33

def modify_conf_single host=HOST, user=USERNAME, password=PASSWORD, conf_path=CONFPATH, key, new_value
	if Util.is_localhost? host
    		cm = ConfModifier.new conf_path
    		cm.modify key, new_value
    		cm.writeconf
	else
    		#lib_path = Util.get_lib_path host
        	#cmd_str = "cd #{lib_path} && ruby conf_modifier.rb modify_conf_single #{conf_path} #{key} #{new_value.to_s}"
		#Util.exe_ssh_cmd! host, cmd_str, user, password
		Util.change_in_local(host, conf_path, user, password) { | local_path |
			cm = ConfModifier.new local_path
			cm.modify key, new_value
			cm.writeconf
		}
      	end
end

#recover_conf(host = HOST, user = USERNAME, password = PASSWORD, conf_path = CONFPATH, context_key) ⇒ Object

recover_conf “conf1”



142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
 
# File 'lib/common/socket/env.rb', line 142

def recover_conf host=HOST, user=USERNAME, password=PASSWORD, conf_path=CONFPATH, context_key
	conf_dir=`dirname #{conf_path}`
	conf_name=`basename #{conf_path}`
	conf_dir.chop!
	conf_name.chop!
       bak_conf = Context.get "#{context_key}"
       recover_cmd = "cd #{conf_dir} && mv #{bak_conf} #{conf_name}"
       if Util.is_localhost?(host) then
           	system "#{recover_cmd}"
       else
		Util.exe_ssh_cmd! host, recover_cmd, user, password
       end
       Context.delete "#{context_key}"
       $log.debug "Recover conf which context key is #{context_key}."
end

#start_process(host, module_name, super_path = "/home/space/space/", child_path = "./bin/", user = USERNAME, password = PASSWORD) ⇒ Object

功能

启动模块进程, 运用脚本

参数

  • host

  • module_name module name, e.g.: appui

  • script_path default “/home/space/space”

  • script_name default “loadspace.sh”

return true/false

Example

start_process '10.241.14.56', 'mlstar', '/home/space/mlstar/'



194
195
196
197
198
199
200
201
202
203
204
205
206
207
 
# File 'lib/common/socket/env.rb', line 194

def start_process host, module_name, super_path="/home/space/space/", child_path="./bin/", user=USERNAME, password=PASSWORD

  	if Util.is_localhost? host
		`killall -9 #{module_name}`
		`cd #{super_path}; nohup #{child_path}#{module_name} >/dev/null 2>&1 &`
  	else
		cmd_str = "killall -9 #{module_name};cd #{super_path}; nohup #{child_path}#{module_name} >/dev/null 2>&1 &"
		Util.exe_ssh_cmd! host, cmd_str, user, password
      	end

	sleep 2
      	value = is_process_running? host, module_name, user, password
      	return value
end