anyproto/any-sync
1
0
Fork 0
mirror of https://github.com/anyproto/any-sync.git synced 2025-06-10 10:00:49 +09:00
Code Activity

WIP test sync protocol

Browse source
This commit is contained in:
mcrakhman 2024-06-02 14:55:11 +02:00
parent 20b64b3940
commit fb18c54702
No known key found for this signature in database
GPG key ID: DED12CFEF5B8396B
35 changed files with 849 additions and 152 deletions
Download patch file Download diff file Expand all files Collapse all files

16
commonspace/sync/requestpool.go
View file

@ -1,26 +1,32 @@
package sync
import (
"context"
"sync"
)
type RequestPool interface {
TryTake(peerId, objectId string) bool
Release(peerId, objectId string)
QueueRequestAction(peerId, objectId string, action func()) (err error)
QueueRequestAction(peerId, objectId string, action func(ctx context.Context)) (err error)
}
type requestPool struct {
mu sync.Mutex
taken map[string]struct{}
pools map[string]*tryAddQueue
ctx context.Context
cancel context.CancelFunc
isClosed bool
}
func NewRequestPool() RequestPool {
ctx, cancel := context.WithCancel(context.Background())
return &requestPool{
taken: make(map[string]struct{}),
pools: make(map[string]*tryAddQueue),
ctx: ctx,
cancel: cancel,
taken: make(map[string]struct{}),
pools: make(map[string]*tryAddQueue),
}
}
@ -47,7 +53,7 @@ func (rp *requestPool) Release(peerId, objectId string) {
delete(rp.taken, id)
}
func (rp *requestPool) QueueRequestAction(peerId, objectId string, action func()) (err error) {
func (rp *requestPool) QueueRequestAction(peerId, objectId string, action func(ctx context.Context)) (err error) {
rp.mu.Lock()
if rp.isClosed {
rp.mu.Unlock()
@ -70,7 +76,7 @@ func (rp *requestPool) QueueRequestAction(peerId, objectId string, action func()
pool.TryAdd(objectId, wrappedAction, func() {})
return
}
action()
action(rp.ctx)
rp.Release(peerId, objectId)
}
pool.Replace(objectId, wrappedAction, func() {})