在面向对象的编程语言中,很多语言会根据引用计数器来实现对对象的自动回收机制,就和Unix/Linux中的文件系统那样,当一个对象被另一个对象持有时,被持有的对象自身的引用计数器会加1。只要对象还被引用,即引用计数器还未被清0,那么我们对对象执行释放操作,只是将对象的引用计数器减1。直到对象的引用计数器变为0后,该对象才真正被释放。
下面我们根据以上描述的原理,用C来尝试实现对象的创建、引用和释放。
#include <stdlib.h>
#include <pthread.h>
struct foo{
int f_count;
pthread_mutex_t f_lock;
int f_id;
/*...more stuff here... */
};
struct foo *foo_alloc(int id) /* 申请并分配对象空间 */
{
struct foo *fp;
if ((fp = malloc(sizeof(struct foo)))!= NULL){
fp->f_count = 1;
fp->f_id = id;
if (pthread_mutex_init(&fp->f_lock, NULL)!= 0){
free(fp);
return (NULL);
}
/* ... continue initialization ...*/
}
return (fp);
}
void foo_hold(struct foo *fp) /* 对对象增加了一个引用 */
{
pthread_mutex_lock(&fp->f_lock); //因为可能有多个线程创建并持有对象,所以需要加互斥量
fp->f_count++; /* 引用计数加1 */
pthread_mutex_unlock(&fp->f_lock);
}
void foo_rele(struct foo *fp)/* 释放引用对象 */
{
pthread_mutex_lock(&fp->f_lock);
if (--fp->f_count == 0){ /* last reference */
pthread_mutex_unlock(&fp->f_lock);
pthread_mutex_destroy(&fp->f_lock);
free(fp);
}else{
pthread_mutex_unlock(&fp->f_lock);
}
}