Initial commit: img-infinite-canvas AI design workbench MVP
Moteva-style AI design workbench replica. Home prompt creates a project; the project page provides an infinite canvas with node drag, zoom/pan, a design chat panel, history replay, image asset upload, and project save/regenerate. - Backend: go-zero, DDD layering, sqlc/pgx, optional PostgreSQL; memory or Redis cache; asynq job queue; MinIO/S3/R2/OSS object storage; sky-valley/pi agent runtime adapter. - Frontend: Next.js App Router + Vite artifact build, TypeScript, i18n, shadcn/ui components, auth (OTP/Turnstile/Google/WeChat). - Deploy: Docker Compose and k3s manifests. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,100 @@
|
||||
package realtime
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type MemoryBus struct {
|
||||
mu sync.RWMutex
|
||||
subscribers map[string]map[*memorySubscription]struct{}
|
||||
}
|
||||
|
||||
type memorySubscription struct {
|
||||
bus *MemoryBus
|
||||
projectID string
|
||||
ch chan Event
|
||||
once sync.Once
|
||||
}
|
||||
|
||||
func NewMemoryBus() *MemoryBus {
|
||||
return &MemoryBus{subscribers: make(map[string]map[*memorySubscription]struct{})}
|
||||
}
|
||||
|
||||
func (b *MemoryBus) Publish(ctx context.Context, event Event) error {
|
||||
if err := ctx.Err(); err != nil {
|
||||
return err
|
||||
}
|
||||
b.mu.RLock()
|
||||
subscribers := make([]*memorySubscription, 0, len(b.subscribers[event.ProjectID]))
|
||||
for sub := range b.subscribers[event.ProjectID] {
|
||||
subscribers = append(subscribers, sub)
|
||||
}
|
||||
b.mu.RUnlock()
|
||||
for _, sub := range subscribers {
|
||||
select {
|
||||
case sub.ch <- event:
|
||||
default:
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (b *MemoryBus) Subscribe(ctx context.Context, projectID string) (Subscription, error) {
|
||||
if err := ctx.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
sub := &memorySubscription{
|
||||
bus: b,
|
||||
projectID: projectID,
|
||||
ch: make(chan Event, 32),
|
||||
}
|
||||
b.mu.Lock()
|
||||
if b.subscribers[projectID] == nil {
|
||||
b.subscribers[projectID] = make(map[*memorySubscription]struct{})
|
||||
}
|
||||
b.subscribers[projectID][sub] = struct{}{}
|
||||
b.mu.Unlock()
|
||||
go func() {
|
||||
<-ctx.Done()
|
||||
_ = sub.Close()
|
||||
}()
|
||||
return sub, nil
|
||||
}
|
||||
|
||||
func (b *MemoryBus) Close() error {
|
||||
b.mu.Lock()
|
||||
defer b.mu.Unlock()
|
||||
for _, subscribers := range b.subscribers {
|
||||
for sub := range subscribers {
|
||||
sub.once.Do(func() {
|
||||
sub.closeLocked()
|
||||
})
|
||||
}
|
||||
}
|
||||
b.subscribers = make(map[string]map[*memorySubscription]struct{})
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *memorySubscription) C() <-chan Event {
|
||||
return s.ch
|
||||
}
|
||||
|
||||
func (s *memorySubscription) Close() error {
|
||||
s.once.Do(func() {
|
||||
s.bus.mu.Lock()
|
||||
defer s.bus.mu.Unlock()
|
||||
s.closeLocked()
|
||||
})
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *memorySubscription) closeLocked() {
|
||||
if subscribers := s.bus.subscribers[s.projectID]; subscribers != nil {
|
||||
delete(subscribers, s)
|
||||
if len(subscribers) == 0 {
|
||||
delete(s.bus.subscribers, s.projectID)
|
||||
}
|
||||
}
|
||||
close(s.ch)
|
||||
}
|
||||
Reference in New Issue
Block a user