Skip to content

Commit 753461e

Browse files
committed
Improve AsyncSonyFlake performance
1 parent e120101 commit 753461e

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

src/sonyflake_turbo/__init__.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,8 @@ async def __call__(self, n):
9191

9292
ids, to_sleep = self.sf._raw(n)
9393

94-
await self.sleep(to_sleep)
94+
if to_sleep > 0:
95+
await self.sleep(to_sleep)
9596

9697
return ids
9798

@@ -100,7 +101,8 @@ def __await__(self):
100101

101102
id_, to_sleep = self.sf._raw(None)
102103

103-
yield from self.sleep(to_sleep).__await__()
104+
if to_sleep > 0:
105+
yield from self.sleep(to_sleep).__await__()
104106

105107
return id_
106108

@@ -114,5 +116,8 @@ async def _gen(self):
114116

115117
while True:
116118
id_, to_sleep = self.sf._raw(None)
117-
await self.sleep(to_sleep)
119+
120+
if to_sleep > 0:
121+
await self.sleep(to_sleep)
122+
118123
yield id_

0 commit comments

Comments
 (0)