mirror of
https://github.com/syncthing/syncthing.git
synced 2024-11-16 18:41:59 -07:00
2234c45c19
The connections service no longer depends directly on the syncthing model object, but on an interface instead. This makes it drastically easier to write clients that handle the model differently, but still want to benefit from existing and future connections changes in the core. This was motivated by burkemw3's interest in creating a FUSE client that can present a view of the global model, but not have all of the file data locally. The actual decoupling was done by adding a connections.Model interface. This interface is effectively an extension of the protocol.Model interface that also handles connections alongside the modified service.
33 lines
668 B
Go
33 lines
668 B
Go
// Copyright (C) 2014 The Syncthing Authors.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
// You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package connections
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/juju/ratelimit"
|
|
)
|
|
|
|
type LimitedWriter struct {
|
|
writer io.Writer
|
|
bucket *ratelimit.Bucket
|
|
}
|
|
|
|
func NewWriteLimiter(w io.Writer, b *ratelimit.Bucket) *LimitedWriter {
|
|
return &LimitedWriter{
|
|
writer: w,
|
|
bucket: b,
|
|
}
|
|
}
|
|
|
|
func (w *LimitedWriter) Write(buf []byte) (int, error) {
|
|
if w.bucket != nil {
|
|
w.bucket.Wait(int64(len(buf)))
|
|
}
|
|
return w.writer.Write(buf)
|
|
}
|