nginx 源码学习笔记（八）——基本容器——array数组

60 阅读 0 评论 40 点赞

我是靠谱客的博主温暖手机，最近开发中收集的这篇文章主要介绍nginx 源码学习笔记（八）——基本容器——array数组，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

对应文件为core/ngx_array.{c|h}

ngx_array是nginx内部封装的，使用ngx_pool_t对内存池进行分配的数组容器，其中的数据是在一整片内存区中连续存放的。更新数据时只能在尾部压入1个或多个元素。这里单纯的觉得和数组没有差别。

数组的实现结构为：

[cpp]  
   view plain 
   copy 
   print 
   ? 
    
   
 <span style="font-size:16px;">struct ngx_array_s {  
     void        *elts;   //具体的数据区域的指针  
     ngx_uint_t   nelts;   //数组实际包含的元素数量  
     size_t       size;    //数组单个元素的大小  
     ngx_uint_t   nalloc;   //数组容器预先(或者重新)分配的内存大小 ！！！这部分为预计的存储元素数量！！！  
     ngx_pool_t  *pool;    //分配的内存池  
 };  
 </span>  

常用操作有：

[cpp]  
   view plain 
   copy 
   print 
   ? 
    
   
 <span style="font-size:16px;">//创建数组  
 ngx_array_t *  
 ngx_array_create(ngx_pool_t *p, ngx_uint_t n, size_t size)  
 {  
     ngx_array_t *a;  
   
     a = ngx_palloc(p, sizeof(ngx_array_t));   //分配数组空间  
     if (a == NULL) {  
         return NULL;  
     }  
     a->elts = ngx_palloc(p, n * size);        //分配数据区域空间  
     if (a->elts == NULL) {  
         return NULL;  
     }  
     a->nelts = 0;                       //初始化信息  
     a->size = size;  
     a->nalloc = n;  
     a->pool = p;  
     return a;  
 }  
   
 //摧毁数组  
 void  
 ngx_array_destroy(ngx_array_t *a)  
 {  
     ngx_pool_t  *p;  
     p = a->pool;  
     if ((u_char *) a->elts + a->size * a->nalloc == p->d.last) {  //如果有数据 判断总数据量是否等于p追后地址  
         p->d.last -= a->size * a->nalloc;     //如果等于，指针迁移总数据量位数即可  
     }  
     if ((u_char *) a + sizeof(ngx_array_t) == p->d.last) {     //如果没有数据  
         p->d.last = (u_char *) a;                       //直接把指针移动到a的地址上  
     }  
 }  
   
 //  
 void *  
 ngx_array_push(ngx_array_t *a)  
 {  
     void        *elt, *new;  
     size_t       size;  
     ngx_pool_t  *p;  
     if (a->nelts == a->nalloc) {                         //数组已经存满  
         /* the array is full */  
         size = a->size * a->nalloc;  
         p = a->pool;  
         if ((u_char *) a->elts + size == p->d.last          //如果总数据量等于last地址，并且新加的数据在可用范围内  
             && p->d.last + a->size <= p->d.end)  
         {  
             /* 
              * the array allocation is the last in the pool 
              * and there is space for new allocation 
              */  
             p->d.last += a->size;                    //last后移  
             a->nalloc++;                          //总数据量增加  
         } else {  
             /* allocate a new array */               //如果不是，创建新的内存空间  
             new = ngx_palloc(p, 2 * size);            //创建一个两倍大小的内存空间  
             if (new == NULL) {  
                 return NULL;  
             }  
             ngx_memcpy(new, a->elts, size);        //复制到新的内存空间  
             a->elts = new;                       //指向新地址  
             a->nalloc *= 2;                       //可容纳总数量*2  
         }  
     }  
     elt = (u_char *) a->elts + a->size * a->nelts;        //新指针用于返回  
     a->nelts++;                                  //实际数量增加  
     return elt;                                  //返回指针指向的地址，这里只进行内存分配，还需要对返回的新指针进行赋值，等操作才可以实现数组的添加，这里要特殊注意  
 }  
 //void * ngx_array_push_n(ngx_array_t *a, ngx_uint_t n) //对于这个函数就不多讲了  
   
 </span>  

总结：

ngx_array_create 在未创建ngx_array_t结构体时创建数组
ngx_array_init 在创建ngx_array_t结构体后创建数组
ngx_array_destroy 销毁数组
ngx_array_push 在数组中压入一个数据，得到一个数据指针，在这里就不用再分配内存了，可以直接使用分配的内存

ngx_array_push_n 在数组中压入 n个数据，得到一个数据指针，在这里就不用再分配内存了，可以直接使用分配的内存