blob: 1b119507a1cb9685cd2f870f56a260b9aac3c0df (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
// Copyright 2024 Staysail Systems, Inc. <info@staysail.tech>
//
// This software is supplied under the terms of the MIT License, a
// copy of which should be located in the distribution where this
// file was obtained (LICENSE.txt). A copy of the license may also be
// found online at https://opensource.org/licenses/MIT.
//
#include <core/refcnt.h>
void
nni_refcnt_init(
nni_refcnt *rc, unsigned value, void *data, void (*fini)(void *))
{
nni_atomic_init(&rc->rc_cnt);
nni_atomic_set(&rc->rc_cnt, value);
rc->rc_data = data;
rc->rc_fini = fini;
}
void
nni_refcnt_hold(nni_refcnt *rc)
{
nni_atomic_inc(&rc->rc_cnt);
}
void
nni_refcnt_rele(nni_refcnt *rc)
{
if (nni_atomic_dec_nv(&rc->rc_cnt) == 0) {
rc->rc_fini(rc->rc_data);
}
}
|
