source: nutchez-0.2/src/test/nutchez @ 214

Last change on this file since 214 was 210, checked in by rock, 14 years ago

新增回到主選單

File size: 13.3 KB
Line 
1#!/bin/bash
2# Program:
3#   NutchEz management interface
4# Author:
5#   Waue, Shunfa, Rock {waue, shunfa, rock}@nchc.org.tw
6# Version:
7#    1.0
8# History:
9#   2010/06/07  Rock    First release (1.0)
10
11# [Variables Declaration] #
12DIALOG="dialog"
13main_choice=
14lang=
15
16# [PATH Declaration] #
17NutchEZ_Install_PATH="/opt/nutchez"
18Tomcat_HOME="/opt/nutchez/tomcat"
19NutchEZ_HOME="/home/nutchuser/nutchez"
20Work_Path="./"
21Work_Path_J=0
22
23# Work Path setup
24echo $0 | grep '/' || Work_Path_J=1
25if [ "$Work_Path_J" == "0"  ]; then
26  Work_Path=$(echo $0 | sed 's/nutchez//')
27fi
28
29# [Env Declaration] #
30. $Work_Path/lang/lang_en_US_nutchez
31
32# [Functions Declaration] #
33## [Pre-check for language] ##
34function prepare_lang(){
35lang=$(locale | grep 'LANG=' | cut -d "=" -f2)
36
37# if lang=zh then source lang_zh_TW_nutchez
38echo $lang | grep 'zh' && source $Work_Path/lang/lang_zh_TW_nutchez
39}
40
41
42## [Pre-Check for nutch_nodes file] ##
43function prepare_check(){ 
44# 若無 nutchez_nodes 則跳出
45if [ ! -e "${NutchEZ_HOME}/system/nutch_nodes" ]; then
46    $DIALOG --clear --backtitle "$dia_back" --title "$dia_pre_check_title_1" \
47    --msgbox "$dia_pre_check_msg_1 $NutchEZ_HOME/system/nutch_nodes !!!" 10 40
48    exit
49fi
50
51# 判斷 nutch_nodes 和 nutch_nodes.back
52# 第一次執行則要求 user 更新 /etc/hosts
53if [ ! -e "${NutchEZ_HOME}/system/nutch_nodes.bak" ]; then
54    $DIALOG --clear --backtitle "$dia_back" --title "$dia_pre_check_title_2" \
55    --yesno "$dia_pre_check_yesno_1\n$dia_pre_check_yesno_2" 10 55
56
57    echo -e "$pre_check_echo_1"
58    ./duplicate_del "${NutchEZ_HOME}/system/nutch_nodes"
59    su root -c "./duplicate_del /etc/hosts && ./add_hosts ${NutchEZ_HOME}/system/nutch_nodes /etc/hosts"
60    # 若密碼輸入失敗則跳出
61    if [ $? == "1" ]; then
62        echo "$pre_check_echo_2"
63        exit
64    fi 
65# 若 nutch_nodes 被更新過,則要求更新 /etc/hosts
66elif [ ${NutchEZ_HOME}/system/nutch_nodes -nt ${NutchEZ_HOME}/system/nutch_nodes.bak ]; then
67    $DIALOG --clear --backtitle "$dia_back" --title "$dia_pre_check_title_2" \
68    --yesno "$dia_pre_check_yesno_3\n$dia_pre_check_yesno_2" 10 55
69
70    echo -e "$pre_check_echo_1"
71    ./duplicate_del "${NutchEZ_HOME}/system/nutch_nodes"
72    su root -c "./duplicate_del /etc/hosts && ./add_hosts ${NutchEZ_HOME}/system/nutch_nodes /etc/hosts"
73    # 若密碼輸入失敗則跳出                   
74    if [ $? == "1" ]; then                   
75        echo "$pre_check_echo_2="
76        exit                                 
77    fi
78fi
79}
80
81## [Main Menu] ##
82function main_menu(){
83main_choice="/tmp/main_choice"
84
85$DIALOG --clear --backtitle "$dia_back" \
86    --title "$dia_main_title_1" \
87        --menu "$dia_choose" 15 60 7 \
88        "cluster_status" "$dia_main_menu_1_1" \
89        "cluster_setup" "$dia_main_menu_1_2" \
90        "server_setup" "$dia_main_menu_1_3" \
91        "tomcat_switch" "$dia_main_menu_1_4" \
92        "tomcat_port" "$dia_main_menu_1_5" \
93        "lang_switch" "$dia_main_menu_1_6" \
94        "exit" "$dia_exit" 2>$main_choice
95}
96
97## [Cluster Status (datanode & tasktracker)] ##
98function cluster_status(){
99IP_list=$(cat $NutchEZ_HOME/system/nutch_nodes | grep -v '^$' | grep -v '#'  | awk '{print $1}')
100clusterStatus="/tmp/clusterStatus"
101rm $clusterStatus 2> /dev/null
102#printf '%16s\t %11s\t %10s\t %25s\n' "[IP] [Hostanme] [Network] [Dtatnode & Tasktracker]" >>$clusterStatus
103echo -e "$cluster_status_echo_1" >>$clusterStatus
104echo -e "------------------------------------------------------------------------" >>$clusterStatus
105echo -e "\n$cluster_status_echo_2"
106for ip in $IP_list
107do
108    # Check Network status
109    ip_status="online"
110    ping -c1 -w1 $ip 2>&1 > /dev/null || ip_status="offline"
111    # Check Hadoop/Nutch service through ssh
112    Task_Data_status="stop"
113    if [ $ip_status == "online" ]; then
114    Task_Data=$(ssh -o StrictHostKeyChecking=no $ip "jps |grep TaskTracker ; jps | grep  DataNode")
115    fi
116
117    [ -z "$Task_Data" ] || Task_Data_status="running"
118    host_name=$(cat $NutchEZ_HOME/system/nutch_nodes | grep $ip | awk '{print $2}')
119    echo -e "$ip  $host_name \t\t $ip_status \t $Task_Data_status" >>$clusterStatus
120#    printf '%16s\t %11s\t %10s\t %25s\n' "$ip $host_name $ip_status $Task_Data_status" >>$clusterStatus
121done
122
123read -p "$cluster_status_read_1"
124$DIALOG --clear --backtitle "$dia_back" \
125     --title "$dia_cluster_status_title_1" --textbox $clusterStatus 20 90
126
127    # Back to main menu
128    main_menu
129    menu_choose
130}
131
132
133## [Server Setup (namenode & jobtracker)] ##
134function server_setup(){
135serverSetup=/tmp/serverSetup
136pid_name=$(jps | grep NameNode)
137pid_job=$(jps | grep JobTracker)
138
139if [ -z "$pid_name" -a -z "$pid_job" ]; then                                                                                               
140    $DIALOG --clear --backtitle "$dia_back" --title "$dia_server_title_1" \
141    --msgbox "$dia_server_msg_1" 7 50
142elif [ -z "$pid_name" -a -n "$pid_job" ]; then
143    $DIALOG --clear --backtitle "$dia_back" --title "$dia_server_title_1" \
144    --msgbox "$dia_server_msg_2" 7 50
145elif [ -n "$pid_name" -a -z "$pid_job" ]; then
146    $DIALOG --clear --backtitle "$dia_back" --title "$dia_server_title_1" \
147    --msgbox "$dia_server_msg_3" 7 50
148else
149    $DIALOG --clear --backtitle "$dia_back" --title "$dia_server_title_1" \
150    --msgbox "$dia_server_msg_4" 7 50
151fi
152
153$DIALOG --clear --backtitle "$dia_back" --title "$dia_server_title_2" \
154    --menu "$dia_choose" 15 65 4 \
155    "start" "$dia_server_menu_1_1" \
156    "stop" "$dia_server_menu_1_2" \
157    "restart" "$dia_server_menu_1_3" \
158    "exit" "$dia_exit" 2>$serverSetup
159
160if [ "$(cat $serverSetup)" == "start" ]; then
161    $NutchEZ_Install_PATH/nutch/bin/start-dfs.sh
162    $NutchEZ_Install_PATH/nutch/bin/start-mapred.sh
163elif [ "$(cat $serverSetup)" == "stop" ]; then
164    $NutchEZ_Install_PATH/nutch/bin/stop-dfs.sh
165    $NutchEZ_Install_PATH/nutch/bin/stop-mapred.sh
166elif [ "$(cat $serverSetup)" == "restart" ]; then
167    $NutchEZ_Install_PATH/nutch/bin/stop-dfs.sh
168    $NutchEZ_Install_PATH/nutch/bin/stop-mapred.sh
169    $NutchEZ_Install_PATH/nutch/bin/start-dfs.sh
170    $NutchEZ_Install_PATH/nutch/bin/start-mapred.sh
171else
172    exit
173fi
174
175    # Back to main menu                                                                                   
176    main_menu       
177    menu_choose
178}
179
180
181## [Cluster Setup (datanode & tasktracker)] ##
182function cluster_setup(){
183# 從 nutch_nodes 讀出 ip 和 hostname
184#IP_list=$(cat $NutchEZ_HOME/system/nutch_nodes | grep -v '^$' | grep -v '#' | awk '{print $1}')
185#HOST_list=$(cat $NutchEZ_HOME/system/nutch_nodes | grep -v '^$' | grep -v '#' | awk '{print $2}')
186LIST_status="off"
187allpart="/tmp/allpart"
188runNodes="/tmp/runNodes"
189clusterSetup="/tmp/clusterSetup"
190
191$DIALOG --clear --backtitle "$dia_back" \
192    --title "$dia_cluster_setup_title_1" \
193    --menu "$dia_choose" 15 55 3 \
194    "All" "$dia_cluster_setup_menu_1_1" \
195    "Part" "$dia_cluster_setup_menu_1_2" \
196    "Exit" "$dia_exit" 2>$allpart
197
198# 判斷是否選 all 和 part,若是 all 直接到服務選單,若是 part 則先到機器選單
199if [ "$(cat $allpart)" == "All" ]; then
200     cat $NutchEZ_HOME/system/nutch_nodes | grep -v '^$' | grep -v '#'  | awk '{print $1}' >$runNodes
201elif [ "$(cat $allpart)" == "Part" ]; then
202    $DIALOG --clear --backtitle "$dia_back" --title "$dia_cluster_setup_title_2" \
203    --checklist "$dia_cluster_setup_check_1" 15 55 7 $(cat $NutchEZ_HOME/system/nutch_nodes | \
204    grep -v '^$' | grep -v '#'  | awk '{print $1 "\t" $2 "\t" "$LIST_status" }') 2>$runNodes
205else
206    exit
207fi
208
209if [ -z "$(cat $runNodes)" ]; then
210    exit
211else
212    $(cat $runNodes) | sed -i 's/"//g' $runNodes 
213fi
214
215$DIALOG --clear --backtitle "$dia_back" --title "$dia_cluster_setup_title_3" \
216    --menu "$dia_choose" 15 65 4 \
217    "start" "$dia_cluster_setup_menu_2_1" \
218    "stop" "$dia_cluster_setup_menu_2_2" \
219    "restart" "$dia_cluster_setup_menu_2_3" \
220    "exit" "$dia_exit" 2>$clusterSetup
221
222if [ "$(cat $clusterSetup)" == "start" ]; then
223    echo -e "\n$cluster_setup_echo_1"
224    for node in $(cat $runNodes) 
225    do
226        ssh -o StrictHostKeyChecking=no $node "$NutchEZ_Install_PATH/nutch/bin/hadoop-daemon.sh start datanode \
227        ; $NutchEZ_Install_PATH/nutch/bin/hadoop-daemon.sh start tasktracker" 2>/dev/null
228        [ $? == "0" ] || echo "$cluster_setup_echo_error $node !!!"
229    done
230elif [ "$(cat $clusterSetup)" == "stop" ]; then
231    echo -e "\n$cluster_setup_echo_3"
232    for node in $(cat $runNodes)
233    do
234        ssh -o StrictHostKeyChecking=no $node "$NutchEZ_Install_PATH/nutch/bin/hadoop-daemon.sh stop datanode \
235        ; $NutchEZ_Install_PATH/nutch/bin/hadoop-daemon.sh stop tasktracker" 2>/dev/null
236        [ $? == "0" ] || echo "$cluster_setup_echo_error $node !!!"
237    done
238elif [ "$(cat $clusterSetup)" == "restart"  ]; then
239    echo -e "\n$cluster_setup_echo_5"
240    for node in $(cat $runNodes)
241    do 
242        ssh -o StrictHostKeyChecking=no $node "$NutchEZ_Install_PATH/nutch/bin/hadoop-daemon.sh stop datanode \
243        ; $NutchEZ_Install_PATH/nutch/bin/hadoop-daemon.sh stop tasktracker \
244        ; $NutchEZ_Install_PATH/nutch/bin/hadoop-daemon.sh start datanode \
245        ; $NutchEZ_Install_PATH/nutch/bin/hadoop-daemon.sh start tasktracker" 2>/dev/null
246        [ $? == "0" ] || echo "$cluster_setup_echo_error $node !!!"
247    done
248else
249    exit
250fi
251
252    # Back to main menu                                                                                   
253    main_menu       
254    menu_choose
255}
256
257## [Tomcat Severice start/stop/restart] ##
258function tomcat_switch(){
259tom_pids=$(ps x | grep -v 'grep' | grep "tomcat" | awk '{print $1}')
260if [ -n "$tom_pids" ]; then
261    $DIALOG --clear --backtitle "$dia_back" --title "$dia_tomcat_switch_title_1" \
262        --msgbox "$dia_tomcat_switch_msg_1" 7 50
263else
264    $DIALOG --clear --backtitle "$dia_back" --title "$dia_tomcat_switch_title_1" \
265        --msgbox "$dia_tomcat_switch_msg_2" 7 50
266fi
267tomcatSwitch="/tmp/tomcatSwitch"
268$DIALOG --clear --backtitle "$dia_back" --title "$dia_tomcat_switch_title_2" \
269    --menu "$dia_choose" 15 55 4 \
270        "start" "$dia_tomcat_switch_menu_1_1" \
271        "stop" "$dia_tomcat_switch_menu_1_2" \
272        "restart" "$dia_tomcat_switch_menu_1_3" \
273        "exit" "$dia_exit" 2>$tomcatSwitch
274
275if [ "$(cat $tomcatSwitch)" == "exit" ]; then
276    exit 0
277fi
278
279# jude $Tomcat_HOME/bin/startup.sh
280if [ ! -e $Tomcat_HOME/bin/startup.sh ]; then
281    echo -e "\n$tomcat_switch_echo_1 $Tomcat_HOME/bin/startup.sh"
282    exit
283fi
284
285if [ ! -e $Tomcat_HOME/bin/shutdown.sh ]; then
286    echo -e "\n$tomcat_switch_echo_1 $Tomcat_HOME/bin/shutdown.sh"
287    exit
288fi
289
290if [ "$(cat $tomcatSwitch)" == "start" ]; then
291    echo "$tomcat_switch_echo_3"
292    $Tomcat_HOME/bin/startup.sh
293elif [ "$(cat $tomcatSwitch)" == "stop" ]; then
294    echo "$tomcat_switch_echo_4"
295    $Tomcat_HOME/bin/shutdown.sh
296    for tom_pid in $tom_pids
297    do
298        kill -9 $tom_pid 2>/dev/null
299    done
300elif [ "$(cat $tomcatSwitch)" == "restart" ]; then
301    echo "$tomcat_switch_echo_5"
302    $Tomcat_HOME/bin/shutdown.sh
303    for tom_pid in $tom_pids
304    do
305        kill -9 $tom_pid 2>/dev/null
306    done
307    $Tomcat_HOME/bin/startup.sh
308else
309    exit 0
310fi
311
312    # Back to main menu                                                                                   
313    main_menu       
314    menu_choose
315}
316
317## [Tomcat Port Change] ##
318function tomcat_port(){
319
320if [ ! -e $Tomcat_HOME/conf/server.xml ]; then
321    $DIALOG --clear --backtitle "$dia_back" --title "$dia_tomcat_port_title_1" \
322        --msgbox "$dia_tomcat_port_msg_1 $Tomcat_HOME/conf/server.xml !!!" 10 50
323    exit
324fi
325tomcatPort="/tmp/tomcatPort"
326$DIALOG --clear --backtitle "$dia_back" --title "$dia_tomcat_port_title_2" \
327    --inputbox "$dia_tomcat_port_input_1" 10 55 2>$tomcatPort
328
329if [ -z "$(cat $tomcatPort)" ]; then
330         # Back to main menu                                                                               
331         main_menu
332         menu_choose
333else
334    modify_line_nu=$(cat -n $Tomcat_HOME/conf/server.xml | grep -v SSL | grep 'HTTP/1.1' | grep '<Connector' | awk '{print $1}')
335    sed -i "${modify_line_nu}c <Connector port=\"$(cat $tomcatPort)\" protocol=\"HTTP/1.1\"" $Tomcat_HOME/conf/server.xml
336
337    tom_pids=$(ps x | grep -v 'grep' | grep "tomcat" | awk '{print $1}')
338    $Tomcat_HOME/bin/shutdown.sh
339    for tom_pid in $tom_pids
340       do
341           kill -9 $tom_pid 2>/dev/null
342       done
343    $Tomcat_HOME/bin/startup.sh
344fi
345
346    # Back to main menu                                                                                   
347    main_menu       
348    menu_choose
349}
350
351
352## [Language Change] ##
353function lang_switch(){
354langSwitch="/tmp/langSwitch"
355$DIALOG --clear --backtitle "$dia_back" --title "$dia_lang_title_1" \
356    --menu "$dia_choose" 15 55 3 \
357    "en_US" "$dia_lang_menu_1_1" \
358    "zh_TW" "$dia_lang_menu_1_2" \
359    "Exit" "$dia_exit" 2>$langSwitch
360#
361if [ "$(cat $langSwitch)" == "Exit" ]; then
362    exit
363elif [ "$(cat $langSwitch)" == "en_US" ]; then
364    source $Work_Path/lang/lang_en_US_nutchez
365    main_menu
366    menu_choose
367elif [ "$(cat $langSwitch)" == "zh_TW" ]; then
368    source $Work_Path/lang/lang_zh_TW_nutchez
369    main_menu
370    menu_choose
371fi
372}
373
374## [Menu_choose] ##
375function menu_choose(){
376case $(cat $main_choice) in
377    "cluster_status")
378        cluster_status;;
379    "cluster_setup")
380        cluster_setup;;
381    "server_setup")   
382        server_setup;;
383    "tomcat_switch")
384        tomcat_switch;;
385    "tomcat_port")
386        tomcat_port;;
387    "lang_switch")
388        lang_switch;;
389esac
390}
391
392# [Main Code] #
393if [ $USER != "nutchuser" ]; then
394    echo -e "\n$user_error"
395    exit
396fi
397
398prepare_lang
399prepare_check
400main_menu
401menu_choose
Note: See TracBrowser for help on using the repository browser.