【指令碼】如何確保應用程式的唯一性
簡介
相信大家在開發指令碼或者寫程式的時候 ,大多會遇到如何判斷已經有程式在執行的情況。比如設計備份binlog ,由於某個例項產生的binlog 數量大於備份的速度,在下一個時間點,會啟動一個新的程式對binlog進行備份。那我們要怎麼解決呢,本文分別從 shell和python的角度提出我的解決方法,同時也推薦《 》,這裡有比較詳細的討論。
一 shell 指令碼的解決方法
利用mkdir 的特性 建立已經存在的檔案目錄則會失敗。程式第一次執行的時候可以建立一個 /tmp/lock資料夾,標示當前已經執行一個程式,當啟動第二個程式時,mkdir /tmp/lock 便會失敗。
注意 linux中的trap命令是防止指令碼異常終止 :被kill (不是kill -9) ,crtl+c 中斷 比較詳細的資料 《Linux命令之trap - 在指令碼中處理訊號》http://codingstandards.iteye.com/blog/836588
二 python 指令碼的解決方法
網上搜尋python 鎖定檔案的時候,都會提示 fcntl 模組。Python的檔案鎖目前使用的是fcntl這個庫,它實際上為 Unix上的ioctl,flock和fcntl 函式提供了一個介面。
fcntl模組的函式flock(file_handle, operation)
其中 file_handle 表示檔案描述符,operation 指要進行的鎖操作,有如下幾種:
fcntl.LOCK_UN 解鎖:刪除floc()函式建立的鎖
fcntl.LOCK_EX 排他鎖:除加鎖程式外其他程式沒有對已加鎖檔案讀寫訪問許可權。
fcntl.LOCK_SH 共享鎖:所有程式沒有寫訪問許可權,即使是加鎖程式也沒有。所有程式有讀訪問許可權。
fcntl.LOCK_NB 非阻塞鎖: 此引數意味著函式不能獲得檔案鎖就立即返回,否則,如果使用LOCK_EX/LOCK_SH請求加鎖不成功,則當前程式會等待獲得檔案鎖。使用LOCK_NB可以在獲得這個排他鎖的情況下不阻塞該程式,LOCK_NB 也可以同LOCK_SH或LOCK_NB進行按位或(|)運算操作,比如fcnt.flock(file_handle,fcntl.LOCK_EX|fcntl.LOCK_NB),此時系統便不會阻塞當前的程式。
注意:
1. 對於檔案的f.close() 操作會使檔案鎖失效;
2. 主程式結束後檔案鎖失效;
3. flock()的LOCK_EX是"勸告鎖",系統核心不會強制檢查鎖的狀態,需要在程式碼中進行檔案操作的地方顯式檢查才能生效。
測試指令碼
指令碼中使用is_running 函式對檔案加鎖,time.sleep(10) 模擬長時間執行的程式,第一次執行lock.py 成功加鎖,在程式執行期間 再次執行lock.py ,獲取鎖時會失敗,並且及時退出程式。
測試例子:
會話一
會話二
三 小結
其實還可以有很多其他的方式 比如 最容易想到的 application_name.pid 或者ps application_name | wc -l 來判斷,不過使用ps 命令時,遇到和系統其他命令關鍵字一樣的時候 ,就會不準。中比較推薦使用pid ,各位讀者朋友也可以提出自己的見解。歡迎討論。
相信大家在開發指令碼或者寫程式的時候 ,大多會遇到如何判斷已經有程式在執行的情況。比如設計備份binlog ,由於某個例項產生的binlog 數量大於備份的速度,在下一個時間點,會啟動一個新的程式對binlog進行備份。那我們要怎麼解決呢,本文分別從 shell和python的角度提出我的解決方法,同時也推薦《 》,這裡有比較詳細的討論。
一 shell 指令碼的解決方法
利用mkdir 的特性 建立已經存在的檔案目錄則會失敗。程式第一次執行的時候可以建立一個 /tmp/lock資料夾,標示當前已經執行一個程式,當啟動第二個程式時,mkdir /tmp/lock 便會失敗。
-
#!/bin/bash
-
mkdir /tmp/lock
-
if [ $? -ne 0 ];then
-
echo "there is tr script running .. "
-
exit 1
-
fi
-
trap "rm -fr /tmp/lock " SIGINT SIGTERM
-
sleep 50
-
if [ -d /tmp/lock ];then
-
rm -fr /tmp/lock
-
echo "rm -fr /tmp/lock"
- fi
二 python 指令碼的解決方法
網上搜尋python 鎖定檔案的時候,都會提示 fcntl 模組。Python的檔案鎖目前使用的是fcntl這個庫,它實際上為 Unix上的ioctl,flock和fcntl 函式提供了一個介面。
fcntl模組的函式flock(file_handle, operation)
其中 file_handle 表示檔案描述符,operation 指要進行的鎖操作,有如下幾種:
fcntl.LOCK_UN 解鎖:刪除floc()函式建立的鎖
fcntl.LOCK_EX 排他鎖:除加鎖程式外其他程式沒有對已加鎖檔案讀寫訪問許可權。
fcntl.LOCK_SH 共享鎖:所有程式沒有寫訪問許可權,即使是加鎖程式也沒有。所有程式有讀訪問許可權。
fcntl.LOCK_NB 非阻塞鎖: 此引數意味著函式不能獲得檔案鎖就立即返回,否則,如果使用LOCK_EX/LOCK_SH請求加鎖不成功,則當前程式會等待獲得檔案鎖。使用LOCK_NB可以在獲得這個排他鎖的情況下不阻塞該程式,LOCK_NB 也可以同LOCK_SH或LOCK_NB進行按位或(|)運算操作,比如fcnt.flock(file_handle,fcntl.LOCK_EX|fcntl.LOCK_NB),此時系統便不會阻塞當前的程式。
注意:
1. 對於檔案的f.close() 操作會使檔案鎖失效;
2. 主程式結束後檔案鎖失效;
3. flock()的LOCK_EX是"勸告鎖",系統核心不會強制檢查鎖的狀態,需要在程式碼中進行檔案操作的地方顯式檢查才能生效。
測試指令碼
指令碼中使用is_running 函式對檔案加鎖,time.sleep(10) 模擬長時間執行的程式,第一次執行lock.py 成功加鎖,在程式執行期間 再次執行lock.py ,獲取鎖時會失敗,並且及時退出程式。
-
#!/usr/bin/python2.6
-
#coding:utf8
-
import time
-
import fcntl
-
import sys
-
def is_running(file):
-
lock_file=open(file,"w")
-
try:
-
fcntl.lockf(lock_file,fcntl.LOCK_EX|fcntl.LOCK_NB)
-
print "給檔案加鎖 ,請等待10s..."
-
except :
-
print '檔案加鎖,無法執行,請稍後執行。'
-
return None
-
return lock_file
-
-
if __name__ == "__main__":
-
lockfile="/tmp/rsync_is_running"
-
a=is_running(lockfile)
-
if a is None :
-
print "lock file failed , rsync is running .quit ..."
-
sys.exit(0)
-
else :
-
print "lock file successed !!! "
- time.sleep(10)
會話一
會話二
三 小結
其實還可以有很多其他的方式 比如 最容易想到的 application_name.pid 或者ps application_name | wc -l 來判斷,不過使用ps 命令時,遇到和系統其他命令關鍵字一樣的時候 ,就會不準。中比較推薦使用pid ,各位讀者朋友也可以提出自己的見解。歡迎討論。
來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/22664653/viewspace-2110638/,如需轉載,請註明出處,否則將追究法律責任。
相關文章
- 確保應用程式安全性
- 確保物件的唯一性——單例模式 (五)物件單例模式
- 確保物件的唯一性——單例模式 (四)物件單例模式
- 確保物件的唯一性——單例模式 (三)物件單例模式
- 確保物件的唯一性——單例模式 (二)物件單例模式
- 確保Web應用程式安全應該考慮哪些事項Web
- CodeArts如何保證客戶程式碼和應用安全?
- 程式碼混淆工具ipaguard:如何使用ipaguard保護和混淆iOS應用程式程式碼iOS
- 用於管理應用程式得shell指令碼指令碼
- 如何讓我們的Android應用程式保活?Android
- blender python api 使用指令碼對骨骼進行重新命名(保證唯一性)PythonAPI指令碼
- 雲應用程式如何加強隱私保護
- AbsInt — 確保程式碼安全的靜態效能分析工具
- BeanShell桌面---Java應用程式指令碼引擎 (轉)BeanJava指令碼
- 設計模式系列之單例模式(Singleton Pattern)——確保物件的唯一性設計模式單例物件
- 如何保護在Autodesk應用程式商店的應用不被盜版 - 1
- 從應用安全到程式碼安全 確保軟體安全不能忽視檢測API漏洞!API
- Shell指令碼應用(三)指令碼
- 構建應用程式的正確方法
- puppet確保程式執行
- 如何提升 Web 應用的程式碼質量Web
- 如何解決"應用程式無法啟動,因為應用程式的並行配置不正確"問題並行
- Python的學習(八)----py指令碼中呼叫應用程式Python指令碼
- 使用OWASPTopTen保證Web應用程式的安全Web
- 如何改進你的指令碼程式指令碼
- appium 安卓應用指令碼APP安卓指令碼
- 如何確保按時交付專案? 有哪些實用方法
- 如何組織大型JavaScript應用中的程式碼?JavaScript
- 直指要害用技術確保網站密碼安全網站密碼
- 確保頁面裝載完畢之後再執行客戶端指令碼客戶端指令碼
- shell指令碼中的變數及應用指令碼變數
- 使用shell指令碼在Linux中管理Java應用程式指令碼LinuxJava
- 9個實用的Javascript程式碼高亮指令碼JavaScript指令碼
- Java 如何確保 JS 不被快取JavaJS快取
- Shell指令碼應用兩個例子指令碼
- 應用市場中包名(package name)的唯一性Package
- 使用Shell指令碼查詢程式對應的程式ID指令碼
- 如何正確的開始用 Go 程式設計Go程式設計