mirror of
https://github.com/syncthing/syncthing.git
synced 2024-11-16 18:41:59 -07:00
34deb82aea
benchmark old ns/op new ns/op delta BenchmarkJobQueueBump 345 154498 +44682.03% BenchmarkJobQueuePushPopDone10k 9437373 3258204 -65.48% benchmark old allocs new allocs delta BenchmarkJobQueueBump 0 0 +0.00% BenchmarkJobQueuePushPopDone10k 10565 22 -99.79% benchmark old bytes new bytes delta BenchmarkJobQueueBump 0 0 +0.00% BenchmarkJobQueuePushPopDone10k 1452498 385869 -73.43%
98 lines
2.2 KiB
Go
98 lines
2.2 KiB
Go
// Copyright (C) 2014 The Syncthing Authors.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify it
|
|
// under the terms of the GNU General Public License as published by the Free
|
|
// Software Foundation, either version 3 of the License, or (at your option)
|
|
// any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful, but WITHOUT
|
|
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
// more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License along
|
|
// with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package model
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/syncthing/syncthing/internal/protocol"
|
|
)
|
|
|
|
type JobQueue struct {
|
|
progress []*protocol.FileInfo
|
|
queued []*protocol.FileInfo
|
|
mut sync.Mutex
|
|
}
|
|
|
|
func NewJobQueue() *JobQueue {
|
|
return &JobQueue{}
|
|
}
|
|
|
|
func (q *JobQueue) Push(file *protocol.FileInfo) {
|
|
q.mut.Lock()
|
|
q.queued = append(q.queued, file)
|
|
q.mut.Unlock()
|
|
}
|
|
|
|
func (q *JobQueue) Pop() *protocol.FileInfo {
|
|
q.mut.Lock()
|
|
defer q.mut.Unlock()
|
|
|
|
if len(q.queued) == 0 {
|
|
return nil
|
|
}
|
|
|
|
var f *protocol.FileInfo
|
|
f, q.queued[0] = q.queued[0], nil
|
|
q.queued = q.queued[1:]
|
|
q.progress = append(q.progress, f)
|
|
|
|
return f
|
|
}
|
|
|
|
func (q *JobQueue) Bump(filename string) {
|
|
q.mut.Lock()
|
|
defer q.mut.Unlock()
|
|
|
|
for i := range q.queued {
|
|
if q.queued[i].Name == filename {
|
|
q.queued[0], q.queued[i] = q.queued[i], q.queued[0]
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (q *JobQueue) Done(file *protocol.FileInfo) {
|
|
q.mut.Lock()
|
|
defer q.mut.Unlock()
|
|
|
|
for i := range q.progress {
|
|
if q.progress[i].Name == file.Name {
|
|
copy(q.progress[i:], q.progress[i+1:])
|
|
q.progress[len(q.progress)-1] = nil
|
|
q.progress = q.progress[:len(q.progress)-1]
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (q *JobQueue) Jobs() ([]protocol.FileInfoTruncated, []protocol.FileInfoTruncated) {
|
|
q.mut.Lock()
|
|
defer q.mut.Unlock()
|
|
|
|
progress := make([]protocol.FileInfoTruncated, len(q.progress))
|
|
for i := range q.progress {
|
|
progress[i] = q.progress[i].ToTruncated()
|
|
}
|
|
|
|
queued := make([]protocol.FileInfoTruncated, len(q.queued))
|
|
for i := range q.queued {
|
|
queued[i] = q.queued[i].ToTruncated()
|
|
}
|
|
|
|
return progress, queued
|
|
}
|