思路

• 將讀寫鎖和讀寫鎖的屬性以及一個(gè)用于存儲(chǔ)共享內(nèi)存的地址的int型變量三者封裝成一個(gè)struct結(jié)構(gòu)
• 將這個(gè)結(jié)構(gòu)體放到共享內(nèi)存中，以及將讀寫鎖的屬性設(shè)置成全局性質(zhì)，然后使用這個(gè)屬性初始化鎖，以及將鎖的地址關(guān)聯(lián)到結(jié)構(gòu)體的內(nèi)存地址這個(gè)變量
• 定義一個(gè)銷毀函數(shù)，用于退出進(jìn)程的時(shí)候，將內(nèi)存空間銷毀，以及刪除對(duì)應(yīng)的變量
• 定義讀寫函數(shù)，讀函數(shù)讀取文件的內(nèi)容，輸出文件的內(nèi)容；寫函數(shù)，以追加的方式，向指定文件中寫入數(shù)據(jù)
• 測(cè)試：使用fork創(chuàng)建父子兩個(gè)進(jìn)程，定義一個(gè)for循環(huán)，循環(huán)5次之后，退出循環(huán)并且執(zhí)行銷毀函數(shù)。
• 在使用讀寫函數(shù)之前需要對(duì)于鎖執(zhí)行相關(guān)的操作，比如加讀鎖，加寫鎖，以及執(zhí)行完讀寫文件函數(shù)之后去掉鎖

代碼

#include <unistd.h> #include <sys/types.h> #include <sys/ipc.h> #include <sys/shm.h> #include <cassert> #include <pthread.h> #include <cstdio> #include <cstdlib> #include <fstream> #include <iostream> #include <sstream> #include <cstring>/*** 返回一片共享內(nèi)存標(biāo)識(shí)符，用于后續(xù)獲取該共享內(nèi)存，以及銷毀該共享內(nèi)存*/ namespace hhh{namespace hhh{class mutex_function{public:mutex_function() = default;// 定義進(jìn)程鎖結(jié)構(gòu)體typedef struct mutex_struct{pthread_rwlock_t lock;pthread_rwlockattr_t lock_attr;int memory_addr;}mutex_struct_t;int create_share_memory_id(const int index_of_key,size_t md5){const char* file_path = "./";key_t id = ftok(file_path,md5);const int share_memory_id = shmget(id,md5,IPC_CREAT | 0666);if (share_memory_id == -1){std::cout << "share_memory_id error!\n" << std::endl;exit(EXIT_FAILURE);}else{return share_memory_id;}}// 初始化進(jìn)程鎖結(jié)構(gòu)體int init_mutex_struct(mutex_struct_t* mutex_ptr){pthread_rwlockattr_init(&(mutex_ptr->lock_attr));pthread_rwlockattr_setpshared(&(mutex_ptr->lock_attr),PTHREAD_PROCESS_SHARED);pthread_rwlock_init(&(mutex_ptr->lock),&(mutex_ptr->lock_attr));return 0;}// 在共享內(nèi)存上定義進(jìn)程鎖結(jié)構(gòu)體并且返回其位置mutex_struct_t* mutex_struct_addr(const int index){int memory_id = create_share_memory_id(index,sizeof (mutex_struct_t));mutex_struct_t * share_ptr = (mutex_struct_t* )shmat(memory_id, 0, 0);if (share_ptr == (void *)-1){std::cout << "share_ptr error!\n" << std::endl;exit(EXIT_FAILURE);}share_ptr->memory_addr = memory_id;assert(init_mutex_struct(share_ptr)== 0);return share_ptr;}// 銷毀進(jìn)程鎖結(jié)構(gòu)體，利用其memory_addr變量索引到其占用的共享內(nèi)存并銷毀const int destroy_mutex_memory(mutex_struct_t * mutex_struct){pthread_rwlock_destroy(&(mutex_struct->lock));pthread_rwlockattr_destroy(&(mutex_struct->lock_attr));assert(shmctl(mutex_struct->memory_addr,IPC_RMID, nullptr) == 0);return 0;}public:mutex_struct_t* mutex_struct_ptr;};}//namespace hhh }//namespace hhh//read file void read_file(const std::string file_name,int id){std::ifstream fp(file_name,std::ios::binary);std::stringstream ss;ss << fp.rdbuf();std::cout << "線程"<< id << "搶占了資源,輸出:" << ss.str() << std::endl;sleep(3);std::cout << "線程"<< id << "釋放了資源...\n" << std::endl;fp.close(); } //write file void write_file(const char *file_name,char *str,int id){FILE *fd = fopen(file_name,"a+");if (fd == nullptr){printf("fd is nullptr and open file fail\n");} else{std::cout <<"線程"<< id << "搶占了資源" << std::endl;fwrite(str,strlen(str),1,fd);char *next = "\0";fwrite(next,strlen(next),1,fd);sleep(5);std::cout << "線程"<< id << "釋放了資源...\n" << std::endl;}fclose(fd); } int main(){std::string file_name = "/home/gsc/Projects/1.txt";char * str1 = "progress 1";// 創(chuàng)建自定義進(jìn)程鎖hsm::hal::mutex_function mutex_function_object;mutex_function_object.mutex_struct_ptr = mutex_function_object.mutex_struct_addr(111);// 創(chuàng)建新進(jìn)程int id = fork();assert(id >= 0);// 設(shè)置循環(huán)次數(shù)for (int i = 0; i < 5; ++i) {if (id > 0) { // 父進(jìn)程// 讀鎖 // pthread_rwlock_rdlock(&(mutex_function_object.mutex_struct_ptr->lock)); // read_file(file_name,getpid()); // pthread_rwlock_unlock(&(mutex_function_object.mutex_struct_ptr->lock));// 寫鎖pthread_rwlock_wrlock(&(mutex_function_object.mutex_struct_ptr->lock));write_file(file_name.c_str(),str1,getpid());pthread_rwlock_unlock(&(mutex_function_object.mutex_struct_ptr->lock));} else { // 子進(jìn)程// 讀鎖 // pthread_rwlock_rdlock(&(mutex_function_object.mutex_struct_ptr->lock)); // read_file(file_name,getpid()); // pthread_rwlock_unlock(&(mutex_function_object.mutex_struct_ptr->lock));// 寫鎖pthread_rwlock_wrlock(&(mutex_function_object.mutex_struct_ptr->lock));write_file(file_name.c_str(),str1,getpid());pthread_rwlock_unlock(&(mutex_function_object.mutex_struct_ptr->lock));}}// 等待循環(huán)完畢sleep(1);// 循環(huán)完畢,銷毀進(jìn)程鎖,釋放申請(qǐng)的共享內(nèi)存if (id > 0) { // 父進(jìn)程mutex_function_object.destroy_mutex_memory((mutex_function_object.mutex_struct_ptr));mutex_function_object.mutex_struct_ptr = nullptr;printf("父進(jìn)程釋放資源完畢\n");}return 0; }

?

總結(jié)

以上是生活随笔為你收集整理的将读写锁放到共享内存，实现进程之间对于同一文件的读写操作的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。