test/acl
1
0
Fork 0
mirror of https://gitee.com/acl-dev/acl.git synced 2024-12-01 19:37:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

check state before wait

Browse Source
This commit is contained in:
白喵 2024-06-25 12:39:04 +08:00
parent 25b59ab47b
commit 919c8a68fa
1 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
lib_fiber/cpp/src/wait_group.cpp
View File

@ -17,8 +17,7 @@ wait_group::~wait_group(void)
void wait_group::add(int n)
{
state_ += (long long)n << 32;
long long state = state_;
long long state = state_.add_fetch((long long)n << 32);
int c = (int)(state >> 32);
uint32_t w = (uint32_t)state;
if(c < 0){
@ -52,22 +51,26 @@ void wait_group::done(void)
void wait_group::wait(void)
{
long long state = state_;
int c = (int)(state >> 32);
uint32_t w = (uint32_t)state;
if(c == 0) return;
state_++;
bool found;
for(;;){
long long state = state_;
int c = (int)(state >> 32);
uint32_t w = (uint32_t)state;
if(c == 0) return;
if(state_.cas(state, state + 1) == state){
bool found;
#ifdef _DEBUG
unsigned long* tid = box_->pop(-1, &found);
assert(found);
delete tid;
unsigned long* tid = box_->pop(-1, &found);
assert(found);
delete tid;
#else
(void) box_->pop(-1, &found);
assert(found);
(void) box_->pop(-1, &found);
assert(found);
#endif
if(state_ != 0){
acl_msg_fatal("wait_group: wait_group is reused before previous wait has returned");
if(state_ != 0){
acl_msg_fatal("wait_group: wait_group is reused before previous wait has returned");
}
return;
}
}
}