欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

C基础 如何让代码只执行一次

程序员文章站 2022-06-24 09:15:05
1.0 最简单, 最高效的方式 C 代码运行起点 main 就是个大单例函数. 如果把函数注册在其里面, 那么一定很可以 :) 是不是, 很轻松的完成了初始化工作. 不妨赠送一个好用的宏, 用于处理这类事情 用起来更简单, 可以插在代码的任何一处 2.0 多线程模式, 如何搞起呢 继续看下面例子 o ......
1.0 最简单, 最高效的方式
    C 代码运行起点 main 就是个大单例函数. 如果把函数注册在其里面, 那么一定很可以 :)
// 某个库需要初始化的函数
void log_init(void) {
    ... ... 
}

int main(int argc, char * argv[]) {
    ... ...
    extern void log_init(void);
    log_init()
    ... ...    
    return 0;
}
是不是, 很轻松的完成了初始化工作.
不妨赠送一个好用的宏, 用于处理这类事情
//
// EXTERN_RUN - 简单的声明, 并立即使用的宏
// ftest    : 需要执行的函数名称
// ...      : 可变参数, 保留
//
#define EXTERN_RUN(ftest, ...)  \
do {                            \
    extern void ftest();        \
    ftest (__VA_ARGS__);        \
} while(0)
用起来更简单, 可以插在代码的任何一处
EXTERN_RUN(log_run);
2.0 多线程模式, 如何搞起呢
    继续看下面例子 once.c 
#include <stdio.h>
#include <pthread.h>

static void _once(void) {
    static long _cnt;

    printf("_once _cnt = %ld\n", ++_cnt);
}

//
// pthread_once 感受
//
int main(int argc, char * argv[]) {
    pthread_once_t once = PTHREAD_ONCE_INIT;

    puts("pthread_once 感受开始 ... ");
    pthread_once(&once, _once);
    pthread_once(&once, _once);
    puts("pthread_once 感受结束 ... ");

    return 0;
}
gcc -g -Wall -o once.out once.c -lpthread
最终运行结果, 也是如我们所料那样

C基础 如何让代码只执行一次

pthread_once 实际开发中多用于初始化线程私有变量. 其内部实现加锁的.
不妨问个小问题, 如果需要你去实现 pthread_once 你会怎么分析呢 ? 

这个问题好解答也不好解答. 
核心亮点在于 pthread_once 运行的函数实体崩溃了. 多线程之间如何避免死锁. 
不妨参照下面 winds 上面 pthread_once 一位大佬的实现:
#include "pthread.h"
#include "implement.h"

/* [i_a] simple wrapper ensures correct calling convention for all */
static void PTW32_CDECL
ptw32_mcs_lock_cleanup(void *args)
{
    ptw32_mcs_local_node_t *node = (ptw32_mcs_local_node_t *)args;
    ptw32_mcs_lock_release(node);
}

int
pthread_once (pthread_once_t * once_control, void (PTW32_CDECL *init_routine) (void))
{
  if (once_control == NULL || init_routine == NULL)
    {
      return EINVAL;
    }

  if ((PTW32_INTERLOCKED_LONG)PTW32_FALSE ==
      (PTW32_INTERLOCKED_LONG)PTW32_INTERLOCKED_EXCHANGE_ADD_LONG((PTW32_INTERLOCKED_LONGPTR)&once_control->done,
                                                                  (PTW32_INTERLOCKED_LONG)0)) /* MBR fence */
    {
      ptw32_mcs_local_node_t node;

      ptw32_mcs_lock_acquire((ptw32_mcs_lock_t *)&once_control->lock, &node);

      if (!once_control->done)
    {

#if defined(PTW32_CONFIG_MSVC7)
#pragma inline_depth(0)
#endif

      pthread_cleanup_push(ptw32_mcs_lock_cleanup, &node);
      (*init_routine)();
      pthread_cleanup_pop(0);

#if defined(PTW32_CONFIG_MSVC7)
#pragma inline_depth()
#endif

      once_control->done = PTW32_TRUE;
    }

      ptw32_mcs_lock_release(&node);
    }

  return 0;

}               /* pthread_once */
核心是通过 pthread_cleanup_push 和  pthread_cleanup_pop 解决崩溃死锁问题.
当然还有一种思路, 可以解决上面问题. 不妨往下看.
3.0 跳过锁问题, 尝试原子操作
    先举个小例子 
#include <stdio.h>

static void _once(void) {
    static long _cnt;

    printf("_once _cnt = %ld\n", ++_cnt);
}

//
// run once 感受
//
int main(int argc, char * argv[]) {
    puts("run once 感受开始 ... ");

    {
        static int _done;
        if (__sync_bool_compare_and_swap(&_done, 0, 1)) {
            _once();
        }

        if (__sync_bool_compare_and_swap(&_done, 0, 1)) {
            _once();
        }
    }

    puts("run once 感受结束 ... ");
    return 0;
}
运行展示:

C基础 如何让代码只执行一次

这里通过 GCC 提供的 原子交换 __sync_bool_compare_and_swap 解决的. 

不妨继续赠送的封装宏, 来完成上面操作
#define ONCE_RUN(code) {                                    \
    static int _done;                                       \
    if (!_done) {                                           \
        if (__sync_bool_compare_and_swap(&_done, 0, 1)) {   \
            code                                            \
        }                                                   \
    }                                                       \
}
因为是原子操作, 没有锁那么重, 自然出了问题也不会引起死锁问题.  

当然有人说 pthread, __sync_xxx 都是和 GCC 绑定的, 那么 CL 能不能使用了. 当然也是可以的.

pthread 跨平台 - https://github.com/wangzhione/structc/blob/master/structc/system/thread.h

atomic 跨平台 - https://github.com/wangzhione/atomic/blob/master/atomic/atom.h

通过上面基础封装库支持, 用 C 写系统应用相关代码还是很好搞的.

把酒满上, 还能再写几行代码

    有问题, 是肯定的. 欢迎指正, 更新是共同提高的过程.

老鼠爱大米 - http://music.163.com/m/song?id=308789&userid=16529894

回头看看, 时间好快呀.  那些个一块补习, 为了玩的伙伴们, 二胎孩子都快上学了. 
哈哈. 
咱们一线秃头兵还在为了 穷 而冲锋陷阵 (°ー°〃)