文章目錄

• 項目介紹
• • 開發(fā)語言
  • 開發(fā)環(huán)境
  • 項目簡介
  • 項目特點
  • 適用場景
  • 發(fā)布鏈接
• 使用介紹
• • 上下文環(huán)境
  • 宏
  • 協(xié)程狀態(tài)
  • 協(xié)程與調(diào)度器結(jié)構(gòu)體
  • 接口
• 示范用例
• • 使用協(xié)程實現(xiàn)一個TCP服務(wù)器

項目介紹

開發(fā)語言

C

---

開發(fā)環(huán)境

CentOS7、vim、gcc、gdb、git、MakeFile

---

項目簡介

“協(xié)程”即用戶態(tài)下的非搶占式的輕量級線程，是一種在程序開發(fā)中處理多任務(wù)的組件。
由于在C/C++中并沒有引入?yún)f(xié)程這一概念，而大部分開源的庫又過于重量，所以我基于ucontext組件實現(xiàn)了一個簡單的協(xié)程庫

---

項目特點

• 用戶態(tài)實現(xiàn)協(xié)程的調(diào)度切換，減少了內(nèi)核切換的開銷。
• 非搶占式，用戶自己實現(xiàn)調(diào)度，同一時間只能有一個協(xié)程在執(zhí)行，由協(xié)程主動交出控制權(quán)。
• 基于非對稱(asymmetric)模式, 控制流更加簡單，程序更加結(jié)構(gòu)化。
• 協(xié)程具有獨立的棧，確保運行效率。

---

適用場景

協(xié)程主要適用于I/O密集型的場景，如示例中的TCP服務(wù)器。在傳統(tǒng)的多路復用+多線程/多進程的做法，每并發(fā)一個進程/線程就會消耗內(nèi)存，并且最嚴重的問題就是由系統(tǒng)來進行調(diào)度切換帶來的嚴重損耗，而協(xié)程剛好能夠解決這些問題。

---

發(fā)布鏈接

具體的實現(xiàn)請參考源代碼
協(xié)程

---

使用介紹

上下文環(huán)境

對于上下文環(huán)境的切換可以使用很多方法

• 匯編
• C語言庫函數(shù)setjmp， longjmp
• glibc的ucontext組

我在這里使用的是ucontext組件，如果不了解這個組件的使用方法，可以參考我的另一篇博客
ucontext族函數(shù)的使用及原理分析

由于ucontext只支持posix，如果需要移植到windows，只需要將api換成fiber的即可

---

宏

#define STACK_SIZE (1024 * 1024) //協(xié)程函數(shù)棧大小 #define COROUTINE_SIZE (1024) //協(xié)程最大數(shù)量

---

協(xié)程狀態(tài)

協(xié)程共設(shè)置了四種狀態(tài)，READY, RUNNING, DEAD, SUSPEND。下圖描述了所有狀態(tài)即狀態(tài)之間的轉(zhuǎn)換關(guān)系。

---

協(xié)程與調(diào)度器結(jié)構(gòu)體

typedef struct coroutine {void* (*call_back)(struct schedule* s, void* args); //回調(diào)函數(shù)void* args; //回調(diào)函數(shù)的參數(shù)ucontext_t ctx; //協(xié)程的上下文char stack[STACK_SIZE]; //協(xié)程的函數(shù)棧enum State state; //協(xié)程狀態(tài)}coroutine;typedef struct schedule {coroutine** coroutines; //協(xié)程數(shù)組int cur_id; //正在運行的協(xié)程下標int max_id; //數(shù)組中最大的下標ucontext_t main; //主流程上下文 }schedule;

---

接口

//創(chuàng)建協(xié)程序調(diào)度器 schedule* schedule_create();//創(chuàng)建協(xié)程, 并返回協(xié)程所處下標 int coroutine_create(schedule* s, void* (*call_back)(schedule*, void* ), void* args);//協(xié)程讓出CPU，返回主流程上下文 void coroutine_yield(schedule* s);//恢復協(xié)程上下文 void coroutine_resume(schedule* s, int id);//運行協(xié)程 void coroutine_running(schedule* s, int id);//銷毀協(xié)程調(diào)度器 void schedule_destroy(schedule* s);//判斷協(xié)程調(diào)度器中的協(xié)程是否全部結(jié)束, 結(jié)束返回1, 沒結(jié)束返回0 int schedule_finished(schedule* s);

---

示范用例

使用協(xié)程實現(xiàn)一個TCP服務(wù)器

思路圖

#include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <sys/socket.h> #include <arpa/inet.h> #include <netinet/in.h> #include <fcntl.h>#include "coroutine.h" static int co_ids[COROUTINE_SIZE];void SetNoBlock(int fd) {int flag = fcntl(fd, F_GETFL, 0);flag |= O_NONBLOCK;fcntl(fd, F_SETFL, flag); }int socket_init() {int lst_fd = socket(AF_INET, SOCK_STREAM, 0);if(lst_fd == -1){perror("socket.\n");exit(1);}int op = 1;setsockopt(lst_fd, SOL_SOCKET, SO_REUSEADDR, &op, sizeof(op));struct sockaddr_in addr;addr.sin_family = AF_INET;addr.sin_port = htons(9200);addr.sin_addr.s_addr = inet_addr("192.168.0.128");if(bind(lst_fd, (struct sockaddr*)&addr, sizeof(addr)) < 0){perror("bind.\n");exit(1);}if(listen(lst_fd, SOMAXCONN) < 0){ perror("listen.\n");exit(1);}return lst_fd; }void accept_conn(int lst_fd, schedule *s, int* co_ids, void *(*call_back)(schedule *s, void *args)) {while(1){int new_fd = accept(lst_fd, NULL, NULL);//如果有新連接到來則創(chuàng)建一個協(xié)程來管理這個連接if(new_fd > 0){SetNoBlock(new_fd);int args[] = {lst_fd, new_fd};int cid = coroutine_create(s, call_back, args);if(cid >= COROUTINE_SIZE){perror("too many connections.\n");return;}co_ids[cid] = 1;coroutine_running(s, cid);}//如果當前沒有連接，則切換至協(xié)程上下文中繼續(xù)運行else{int i = 0;for(i = 0; i < COROUTINE_SIZE; i++){if(co_ids[i] == -1){continue;}coroutine_resume(s, i);}}}}void *handle(schedule *s, void *args) {int* arr = (int*)args;int cfd = arr[1];char buf[1024] = { 0 };while(1){memset(buf, 0, sizeof(buf));int ret = recv(cfd, buf, 1024, 0);if(ret < 0){//如果此時沒有數(shù)據(jù)，則不再等待，直接切換回主流程coroutine_yield(s);}else if(ret == 0){//通信結(jié)束co_ids[s->cur_id] = -1;break;}else{printf("=> : %s\n", buf);if(strncasecmp(buf, "exit", 4) == 0){co_ids[s->cur_id] = -1;break;}send(cfd, buf, ret, 0);}} }int main() {static int co_ids[COROUTINE_SIZE];int lst_fd = socket_init();SetNoBlock(lst_fd);schedule* s = schedule_create();int i;for(i = 0; i < COROUTINE_SIZE; i++){co_ids[i] = -1;}accept_conn(lst_fd, s, co_ids, handle);schedule_destroy(s); }

可以看到該服務(wù)器可以快速的響應(yīng)多個連接

總結(jié)

以上是生活随笔為你收集整理的【项目介绍】协程——C语言实现的用户态非抢占式轻量级线程的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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