perf(chat,update): пул воркеров per-user вместо сериальной очереди + HEAD-проба обновлений
Some checks failed
CI / test (push) Failing after 1m15s
CI / build-and-package (amd64, linux) (push) Failing after 58s
CI / build-and-package (amd64, windows) (push) Successful in 30s

- chat.Router: ограниченный пул chatWorkers=4 воркеров + FIFO-очереди
  per-user (userState/workerLoop/runUser). Порядок сообщений одного UserID
  сохраняется; разные пользователи обрабатываются параллельно (до 4
  одновременных LLM-вызовов), long-poll Telegram не блокируется чужим
  аналитиком. Backpressure по jobs — только на перегруженного пользователя.
- app.FreeChat: sessions под sync.Mutex (защита от data race при параллельных
  воркерах роутера).
- update: ResolveLatest проверяет наличие бинаря HEAD-пробой без скачивания
  тела (fallback GET Range 0-0 при 405/501), сортировка версий по id убыв.;
  один общий http.Client (keep-alive) вместо нового на каждый запрос.
- тесты: порядок/параллелизм per-user в router, HEAD-без-тела и фоллбэк на
  версию без бинаря в update.
- память Serena: инварианты Router/update, примечания по форматированию на Windows.
This commit is contained in:
ki.sagidullin
2026-08-22 11:44:34 +05:00
parent 2854697415
commit cd0619926e
7 changed files with 403 additions and 40 deletions

View File

@@ -9,8 +9,8 @@ import (
)
const (
uidA UserID = "u-a"
uidB UserID = "u-b"
uidA UserID = "u-a"
uidB UserID = "u-b"
tg Address = "tg://123"
tui Address = "tui://local"
)
@@ -213,3 +213,105 @@ func TestRouter_Ask_PendingNotConsumedFromOtherAddr(t *testing.T) {
t.Fatal("pending должен остаться (ответ из другого адреса)")
}
}
// TestRouter_PerUserOrdering проверяет, что сообщения одного пользователя
// обрабатываются строго в порядке поступления (пул воркеров не перемешивает).
func TestRouter_PerUserOrdering(t *testing.T) {
cb := newFakeOnMsg()
r := NewRouter(cb.h)
tgCh := newFakeChannel(tg)
_ = r.Attach(tgCh)
for _, txt := range []string{"1", "2", "3"} {
tgCh.emit(uidA, tg, txt)
}
if !cb.wait(3) {
t.Fatal("сообщения не обработаны за таймаут")
}
for i, want := range []string{"1", "2", "3"} {
if got := cb.get(i).Msg.Text; got != want {
t.Errorf("порядок обработки нарушен: idx %d = %q, want %q", i, got, want)
}
}
}
// TestRouter_ParallelismAcrossUsers проверяет, что пока обработчик одного
// пользователя заблокирован (долгий LLM-вызов), сообщение другого пользователя
// обрабатывается в другом воркере, а второе сообщение того же пользователя —
// ждёт своей очереди (per-user порядок).
func TestRouter_ParallelismAcrossUsers(t *testing.T) {
r := NewRouter(nil)
tgCh := newFakeChannel(tg)
tuiCh := newFakeChannel(tui)
_ = r.Attach(tgCh)
_ = r.Attach(tuiCh)
blocked := make(chan struct{})
release := make(chan struct{})
var muLocal sync.Mutex
seen := make([]string, 0, 3)
signal := make(chan struct{}, 8)
h := func(inc Incoming) {
if inc.Msg.Text == "block" {
close(blocked)
<-release // держим воркера, пока не отпустим
}
muLocal.Lock()
seen = append(seen, inc.Msg.Text)
muLocal.Unlock()
signal <- struct{}{}
}
r.onUserMsg = h
snapshot := func() []string {
muLocal.Lock()
defer muLocal.Unlock()
return append([]string(nil), seen...)
}
waitFor := func(n int) bool {
deadline := time.After(2 * time.Second)
for len(snapshot()) < n {
select {
case <-signal:
case <-deadline:
return false
}
}
return true
}
// первое сообщение A блокирует своего воркера
tgCh.emit(uidA, tg, "block")
<-blocked
// B обрабатывается параллельно, пока A висит
tuiCh.emit(uidB, tui, "B1")
select {
case <-signal:
case <-time.After(100 * time.Millisecond):
t.Fatal("B не обработан, пока блокирован A — чат-путь снова сериализован")
}
if got := snapshot(); len(got) != 1 || got[0] != "B1" {
t.Fatalf("ожидали обработку B1, got %v", got)
}
// второе сообщение A НЕ обрабатывается, пока занят воркер A (порядок per-user)
tgCh.emit(uidA, tg, "a2")
select {
case <-signal:
t.Fatal("сообщение A обработано ДО освобождения A — нарушен per-user порядок")
case <-time.After(80 * time.Millisecond):
}
// отпускаем A → дообрабатывается a2
close(release)
if !waitFor(3) {
t.Fatal("итоговые сообщения не обработаны")
}
got := snapshot()
if got[2] != "a2" {
t.Errorf("порядок персональной очереди нарушен: pos2 = %q, want a2", got[2])
}
}