doesnm/quickshare
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(handlers): replace boltdb with sqlite in handlers

Browse source
This commit is contained in:
hexxa 2022-09-03 20:24:45 +08:00 • committed by Hexxa
parent 791848f75c
commit 085a3e4e10
14 changed files with 342 additions and 307 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/iolimiter/iolimiter.go
View file

@ -1,11 +1,11 @@
package iolimiter
import (
"context"
"fmt"
"sync"
"github.com/ihexxa/quickshare/src/db"
"github.com/ihexxa/quickshare/src/db/userstore"
"github.com/ihexxa/quickshare/src/golimiter"
)
@ -20,11 +20,11 @@ type IOLimiter struct {
mtx *sync.Mutex
UploadLimiter *golimiter.Limiter
DownloadLimiter *golimiter.Limiter
users userstore.IUserStore
users db.IUserDB
quotaCache map[uint64]*db.Quota
}
func NewIOLimiter(cap, cyc int, users userstore.IUserStore) *IOLimiter {
func NewIOLimiter(cap, cyc int, users db.IUserDB) *IOLimiter {
return &IOLimiter{
mtx: &sync.Mutex{},
UploadLimiter: golimiter.New(cap, cyc),
@ -40,7 +40,7 @@ func (lm *IOLimiter) CanWrite(id uint64, chunkSize int) (bool, error) {
quota, ok := lm.quotaCache[id]
if !ok {
user, err := lm.users.GetUser(id)
user, err := lm.users.GetUser(context.TODO(), id) // TODO: add context
if err != nil {
return false, err
}
@ -64,7 +64,7 @@ func (lm *IOLimiter) CanRead(id uint64, chunkSize int) (bool, error) {
quota, ok := lm.quotaCache[id]
if !ok {
user, err := lm.users.GetUser(id)
user, err := lm.users.GetUser(context.TODO(), id) // TODO: add context
if err != nil {
return false, err
}