其他分享
首页 > 其他分享> > 简单流控--漏桶与令牌桶

简单流控--漏桶与令牌桶

作者:互联网

漏桶示例:slowcat.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>

#define READLENTH 10  //设置一次读取长度
static volatile int loop =1;   //标志

static void alrm_handler(int s)  //闹钟事件处理
{
  alarm(1);  //循环设置闹钟
  loop = 0;
}

int main(int argc, char**argv)
{
  int fd;
  int readcnt;
  int writecnt;
  int pos;
  char readbuff[READLENTH];

  if(argc<2) //参数判断
  {
    fprintf(stderr,"Usage:%s <filename>\n",argv[0]);
    exit(1);
  }

  signal(SIGALRM,alrm_handler);   //注册信号处理函数
  alarm(1);  //创建闹钟信号

  while(1)
  {
    fd = open(argv[1],O_RDONLY);  //打开文件
    if(fd <0 && errno !=EINTR)   //当不是中断导致打开出错时
    {
      perror("open()");
      exit(1);
    }
    else if(fd >=0)  //正常打开
      break;
  }

  while(1)
  {
    while(loop)  //判断是否有闹钟事件到
      pause();

    loop = 1;
    while((readcnt = read(fd, readbuff, READLENTH))<0)  //读取数据
    {
      if(errno!=EINTR)  //不为中断导致读取出错时
      {
        perror("read()");
        exit(1);
      }
      else
        continue;
    }
    if(readcnt ==0)
      break;

    pos = 0;
    while(readcnt>0)
    {
      writecnt = write(1,readbuff+pos,readcnt); //向标准输出写入数据
      if(writecnt<0)  //写入错误时
      {
        if(errno == EINTR)
          continue;
        perror("write()");
        exit(1);
      }
      pos +=writecnt;
      readcnt -=writecnt;
    }
  }

  close(fd);
  exit(0);
}

 

令牌桶示例:slowcat2.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>

#define READLENTH 10
#define MAXTOKEN 100  //最大令牌数
static volatile int token =0;     //令牌数

static void alrm_handler(int s)  //闹钟信号处理函数
{
  alarm(1);  //循环产生闹钟事件
  token++;  //令牌数自加 
  if(token>=MAXTOKEN)    //限制令牌最大数
    token == MAXTOKEN;
}

int main(int argc, char**argv)
{
  int fd;
  int readcnt;
  int writecnt;
  int pos;
  char readbuff[READLENTH];

  if(argc<2)  //判断输入参数
  {
    fprintf(stderr,"Usage:%s <filename>\n",argv[0]);
    exit(1);
  }

  signal(SIGALRM,alrm_handler);  //注册闹钟处理函数
  alarm(1);  //创建闹钟

  while(1)
  {
    fd = open(argv[1],O_RDONLY);  //打开文件
    if(fd <0 && errno !=EINTR)
    {
      perror("open()");
      exit(1);
    }
    else if(fd >=0)
      break;
  }

  while(1)
  {
    while(token<=0)  //没有令牌数时
      pause();

    token--;  //令牌数自减
    while((readcnt = read(fd, readbuff, READLENTH))<0) //读取数据
    {
      if(errno!=EINTR)
      {
        perror("read()");
        exit(1);
      }
      else
        continue;
    }
    if(readcnt ==0)
      break;

    pos = 0;
    while(readcnt>0) //循环写入数据
    {
      writecnt = write(1,readbuff+pos,readcnt);
      if(writecnt<0)
      {
        if(errno == EINTR)
          continue;
        perror("write()");
        exit(1);
      }
      pos +=writecnt;
      readcnt -=writecnt;
    }
  }

  close(fd);
  exit(0);
}

标签:readcnt,流控,--,int,while,exit,fd,漏桶,include
来源: https://www.cnblogs.com/linux-learn/p/16485308.html