mirror of
https://github.com/octoleo/syncthing.git
synced 2024-11-06 05:17:49 +00:00
e52be3d83e
So there were some issues here. The main problem was that model.Close(deviceID) was overloaded to mean "the connection was closed by the protocol layer" and "i want to close this connection". That meant it could get called twice - once *to* close the connection and then once more when the connection *was* closed. After this refactor there is instead a Closed(conn) method that is the callback. I didn't need to change the parameter in the end, but I think it's clearer what it means when it takes the connection that was closed instead of a device ID. To close a connection, the new close(deviceID) method is used instead, which only closes the underlying connection and leaves the cleanup to the Closed() callback. I also changed how we do connection switching. Instead of the connection service calling close and then adding the connection, it just adds the new connection. The model knows that it already has a connection and makes sure to close and clean out that one before adding the new connection. To make sure to sequence this properly I added a new map of channels that get created on connection add and closed by Closed(), so that AddConnection() can do the close and wait for the cleanup to happen before proceeding. GitHub-Pull-Request: https://github.com/syncthing/syncthing/pull/3490
61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
// Copyright (C) 2014 The Protocol Authors.
|
|
|
|
package protocol
|
|
|
|
import "time"
|
|
|
|
type TestModel struct {
|
|
data []byte
|
|
folder string
|
|
name string
|
|
offset int64
|
|
size int
|
|
hash []byte
|
|
fromTemporary bool
|
|
closedCh chan struct{}
|
|
closedErr error
|
|
}
|
|
|
|
func newTestModel() *TestModel {
|
|
return &TestModel{
|
|
closedCh: make(chan struct{}),
|
|
}
|
|
}
|
|
|
|
func (t *TestModel) Index(deviceID DeviceID, folder string, files []FileInfo) {
|
|
}
|
|
|
|
func (t *TestModel) IndexUpdate(deviceID DeviceID, folder string, files []FileInfo) {
|
|
}
|
|
|
|
func (t *TestModel) Request(deviceID DeviceID, folder, name string, offset int64, hash []byte, fromTemporary bool, buf []byte) error {
|
|
t.folder = folder
|
|
t.name = name
|
|
t.offset = offset
|
|
t.size = len(buf)
|
|
t.hash = hash
|
|
t.fromTemporary = fromTemporary
|
|
copy(buf, t.data)
|
|
return nil
|
|
}
|
|
|
|
func (t *TestModel) Closed(conn Connection, err error) {
|
|
t.closedErr = err
|
|
close(t.closedCh)
|
|
}
|
|
|
|
func (t *TestModel) ClusterConfig(deviceID DeviceID, config ClusterConfig) {
|
|
}
|
|
|
|
func (t *TestModel) DownloadProgress(DeviceID, string, []FileDownloadProgressUpdate) {
|
|
}
|
|
|
|
func (t *TestModel) closedError() error {
|
|
select {
|
|
case <-t.closedCh:
|
|
return t.closedErr
|
|
case <-time.After(1 * time.Second):
|
|
return nil // Timeout
|
|
}
|
|
}
|