what
0f4a4ded53
Invoke 之前完全依赖调用方传入的 ctx 控制超时，池子被占满或 Python 侧 handler 阻塞时，未设置 deadline 的调用会永久阻塞且不报错。新增 WithDefaultTimeout 选项，仅在 ctx 未设置 deadline 时兜底生效，调用方 显式设置的超时优先级更高。 同时补充 example 中的阻塞/超时演示（demoTimeout、demoBlocking）和 pool_test.go 集成测试，覆盖池占满排队、默认超时、显式 deadline 优先级、 流式输出超时后 channel 静默关闭等场景。
57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
package gobridge
|
||
|
||
import (
|
||
"context"
|
||
"net"
|
||
"time"
|
||
)
|
||
|
||
// singleWorkerPool 是 Pool 的包装，所有请求固定路由到同一个 worker 进程。
|
||
type singleWorkerPool struct {
|
||
pool *pool
|
||
workerIdx int
|
||
}
|
||
|
||
// NewSession 返回一个固定到某个 worker 进程的 Pool 视图。
|
||
// 通过该 Pool 发起的所有 Invoke 调用始终路由到同一 Python 进程，
|
||
// 适用于多次调用之间需要共享 Python 侧状态的场景。
|
||
//
|
||
// session := gobridge.NewSession(pool)
|
||
// gobridge.Invoke(ctx, session, "init", arg)
|
||
// gobridge.Invoke(ctx, session, "next_step") // 与上一行走同一进程
|
||
func NewSession(p Pool) Pool {
|
||
inner, ok := p.(*pool)
|
||
if !ok {
|
||
// 已经是 session 或其他实现，直接返回
|
||
return p
|
||
}
|
||
n := uint64(len(inner.workers))
|
||
if n == 0 {
|
||
return p
|
||
}
|
||
idx := inner.idx.Add(1) % n
|
||
return &singleWorkerPool{pool: inner, workerIdx: int(idx)}
|
||
}
|
||
|
||
func (s *singleWorkerPool) acquire(ctx context.Context) (net.Conn, *worker, error) {
|
||
idx := s.workerIdx % len(s.pool.workers)
|
||
w := s.pool.workers[idx]
|
||
conn, err := w.acquire(ctx)
|
||
return conn, w, err
|
||
}
|
||
|
||
// Close 不关闭底层 pool，session 不拥有 pool 的生命周期。
|
||
func (s *singleWorkerPool) Close() {}
|
||
|
||
func (s *singleWorkerPool) nextReqID() uint64 {
|
||
return s.pool.nextReqID()
|
||
}
|
||
|
||
func (s *singleWorkerPool) callbackDispatch(ctx context.Context, msg Message) (any, string) {
|
||
return s.pool.callbackDispatch(ctx, msg)
|
||
}
|
||
|
||
func (s *singleWorkerPool) defaultTimeout() time.Duration {
|
||
return s.pool.defaultTimeout()
|
||
}
|