Skip to content

Commit e653f7a

Browse files
A few fixes for gwf exec
1 parent 048819f commit e653f7a

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/gwf/exec/__main__.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from gwf.executors import deserialize
99

1010

11-
logger = logging.getLogger("gwf_exec")
11+
logger = logging.getLogger(__name__)
1212

1313

1414
async def forward(src, dst, bufsize=2**16, flush_sec=10):
@@ -30,13 +30,14 @@ async def forward(src, dst, bufsize=2**16, flush_sec=10):
3030
done = True
3131
buf[buf_used : buf_used + len(data)] = data
3232
buf_used += len(data)
33-
except TimeoutError:
33+
except asyncio.TimeoutError:
3434
pass
3535

3636
now = time.monotonic()
3737
if done or buf_used == bufsize or now - last_flush > flush_sec:
38-
dst.write(buf[:buf_used])
39-
dst.flush()
38+
if buf_used:
39+
dst.write(buf[:buf_used])
40+
dst.flush()
4041
last_flush = now
4142
buf_used = 0
4243

0 commit comments

Comments
 (0)