rust 如何使用文件鎖防止應用多開

更新時間：2024年04月03日 11:36:20 作者：會編程的大白熊

這篇文章主要介紹了rust 如何使用文件鎖防止應用多開,本文給出了進程只能單開的方法,結合實例代碼給大家介紹的非常詳細,感興趣的朋友跟隨小編一起看看吧

安裝依賴

cargo add fslock

使用方法

use fslock::LockFile;
// 打開pid文件，沒有則自動創(chuàng)建
let mut pid_lock =
        LockFile::open(&pid_path.clone().into_os_string()).unwrap();
// 非阻塞的鎖文件
if !pid_lock.try_lock_with_pid().unwrap() {
    // 如果文件已經(jīng)被鎖，則退出進程
    // ...
}
// 文件加鎖成功，則執(zhí)行業(yè)務邏輯
// ...

庫支持的系統(tǒng)

此庫支持 windows、linux 和 mac，通過如下代碼實現(xiàn)

#[cfg(unix)]
mod unix;
#[cfg(unix)]
use crate::unix as sys;
mod string;
mod fmt;
#[cfg(windows)]
mod windows;
#[cfg(windows)]
use crate::windows as sys;

原理簡介

1. 鎖操作

linux

調(diào)用 libc.so的接口實現(xiàn)，目的是不使用 std 庫。

pub fn try_lock(fd: FileDesc) -> Result<bool, Error> {
    let res = unsafe { libc::flock(fd, libc::LOCK_EX | libc::LOCK_NB) };
    if res >= 0 {
        Ok(true)
    } else {
        let err = errno();
        if err == libc::EWOULDBLOCK || err == libc::EINTR {
            Ok(false)
        } else {
            Err(Error::from_raw_os_error(err as i32))
        }
    }
}

windows

pub fn try_lock(handle: FileDesc) -> Result<bool, Error> {
    let mut overlapped = make_overlapped()?;
    let drop_handle = DropHandle { handle: overlapped.hEvent };
    let res = unsafe {
        LockFileEx(
            handle,
            LOCKFILE_EXCLUSIVE_LOCK | LOCKFILE_FAIL_IMMEDIATELY,
            0,
            1,
            1,
            &mut overlapped as LPOVERLAPPED,
        )
    };
    let ret = if res == TRUE {
        let res = unsafe { WaitForSingleObject(overlapped.hEvent, 0) };
        if res != WAIT_FAILED {
            Ok(true)
        } else {
            Err(Error::last_os_error())
        }
    } else {
        let err = unsafe { GetLastError() };
        if err == ERROR_LOCK_VIOLATION {
            Ok(false)
        } else {
            Err(Error::from_raw_os_error(err as i32))
        }
    };
    drop(drop_handle);
    ret
}

2. 進程退出解鎖清空文件內(nèi)容

impl Drop for LockFile {
    fn drop(&mut self) {
        if self.locked {
            let _ = self.unlock();
        }
        sys::close(self.desc);
    }
}
    pub fn unlock(&mut self) -> Result<(), sys::Error> {
        if !self.locked {
            panic!("Attempted to unlock already unlocked lockfile");
        }
        self.locked = false;
        sys::unlock(self.desc)?;
        // 解鎖后清空文件內(nèi)容
        sys::truncate(self.desc)?;
        Ok(())
    }

到此這篇關于rust 如何使用文件鎖防止應用多開的文章就介紹到這了,更多相關rust 文件鎖內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: