2
2
mirror of https://github.com/octoleo/restic.git synced 2024-06-09 04:12:25 +00:00

Backend: Expose connections parameter

This commit is contained in:
Michael Eischer 2021-08-07 22:20:49 +02:00
parent 07a565e6f7
commit 4f97492d28
10 changed files with 83 additions and 31 deletions

View File

@ -24,6 +24,7 @@ import (
type Backend struct { type Backend struct {
accountName string accountName string
container *storage.Container container *storage.Container
connections uint
sem *backend.Semaphore sem *backend.Semaphore
prefix string prefix string
listMaxItems int listMaxItems int
@ -55,6 +56,7 @@ func open(cfg Config, rt http.RoundTripper) (*Backend, error) {
be := &Backend{ be := &Backend{
container: service.GetContainerReference(cfg.Container), container: service.GetContainerReference(cfg.Container),
accountName: cfg.AccountName, accountName: cfg.AccountName,
connections: cfg.Connections,
sem: sem, sem: sem,
prefix: cfg.Prefix, prefix: cfg.Prefix,
Layout: &backend.DefaultLayout{ Layout: &backend.DefaultLayout{
@ -109,6 +111,10 @@ func (be *Backend) Join(p ...string) string {
return path.Join(p...) return path.Join(p...)
} }
func (be *Backend) Connections() uint {
return be.connections
}
// Location returns this backend's location (the container name). // Location returns this backend's location (the container name).
func (be *Backend) Location() string { func (be *Backend) Location() string {
return be.Join(be.container.Name, be.prefix) return be.Join(be.container.Name, be.prefix)

View File

@ -133,6 +133,10 @@ func (be *b2Backend) SetListMaxItems(i int) {
be.listMaxItems = i be.listMaxItems = i
} }
func (be *b2Backend) Connections() uint {
return be.cfg.Connections
}
// Location returns the location for the backend. // Location returns the location for the backend.
func (be *b2Backend) Location() string { func (be *b2Backend) Location() string {
return be.cfg.Bucket return be.cfg.Bucket

View File

@ -45,6 +45,10 @@ func (be *Backend) Remove(ctx context.Context, h restic.Handle) error {
return nil return nil
} }
func (be *Backend) Connections() uint {
return be.b.Connections()
}
// Location returns the location of the backend. // Location returns the location of the backend.
func (be *Backend) Location() string { func (be *Backend) Location() string {
return "DRY:" + be.b.Location() return "DRY:" + be.b.Location()

View File

@ -34,6 +34,7 @@ import (
type Backend struct { type Backend struct {
gcsClient *storage.Client gcsClient *storage.Client
projectID string projectID string
connections uint
sem *backend.Semaphore sem *backend.Semaphore
bucketName string bucketName string
bucket *storage.BucketHandle bucket *storage.BucketHandle
@ -102,12 +103,13 @@ func open(cfg Config, rt http.RoundTripper) (*Backend, error) {
} }
be := &Backend{ be := &Backend{
gcsClient: gcsClient, gcsClient: gcsClient,
projectID: cfg.ProjectID, projectID: cfg.ProjectID,
sem: sem, connections: cfg.Connections,
bucketName: cfg.Bucket, sem: sem,
bucket: gcsClient.Bucket(cfg.Bucket), bucketName: cfg.Bucket,
prefix: cfg.Prefix, bucket: gcsClient.Bucket(cfg.Bucket),
prefix: cfg.Prefix,
Layout: &backend.DefaultLayout{ Layout: &backend.DefaultLayout{
Path: cfg.Prefix, Path: cfg.Prefix,
Join: path.Join, Join: path.Join,
@ -185,6 +187,10 @@ func (be *Backend) Join(p ...string) string {
return path.Join(p...) return path.Join(p...)
} }
func (be *Backend) Connections() uint {
return be.connections
}
// Location returns this backend's location (the bucket name). // Location returns this backend's location (the bucket name).
func (be *Backend) Location() string { func (be *Backend) Location() string {
return be.Join(be.bucketName, be.prefix) return be.Join(be.bucketName, be.prefix)

View File

@ -229,6 +229,10 @@ func (be *MemoryBackend) List(ctx context.Context, t restic.FileType, fn func(re
return ctx.Err() return ctx.Err()
} }
func (be *MemoryBackend) Connections() uint {
return 2
}
// Location returns the location of the backend (RAM). // Location returns the location of the backend (RAM).
func (be *MemoryBackend) Location() string { func (be *MemoryBackend) Location() string {
return "RAM" return "RAM"

View File

@ -29,9 +29,10 @@ var _ restic.Backend = &Backend{}
// Backend uses the REST protocol to access data stored on a server. // Backend uses the REST protocol to access data stored on a server.
type Backend struct { type Backend struct {
url *url.URL url *url.URL
sem *backend.Semaphore connections uint
client *http.Client sem *backend.Semaphore
client *http.Client
backend.Layout backend.Layout
} }
@ -57,10 +58,11 @@ func Open(cfg Config, rt http.RoundTripper) (*Backend, error) {
} }
be := &Backend{ be := &Backend{
url: cfg.URL, url: cfg.URL,
client: client, client: client,
Layout: &backend.RESTLayout{URL: url, Join: path.Join}, Layout: &backend.RESTLayout{URL: url, Join: path.Join},
sem: sem, connections: cfg.Connections,
sem: sem,
} }
return be, nil return be, nil
@ -105,6 +107,10 @@ func Create(ctx context.Context, cfg Config, rt http.RoundTripper) (*Backend, er
return be, nil return be, nil
} }
func (b *Backend) Connections() uint {
return b.connections
}
// Location returns this backend's location (the server's URL). // Location returns this backend's location (the server's URL).
func (b *Backend) Location() string { func (b *Backend) Location() string {
return b.url.String() return b.url.String()

View File

@ -255,6 +255,10 @@ func (be *Backend) ReadDir(ctx context.Context, dir string) (list []os.FileInfo,
return list, nil return list, nil
} }
func (be *Backend) Connections() uint {
return be.cfg.Connections
}
// Location returns this backend's location (the bucket name). // Location returns this backend's location (the bucket name).
func (be *Backend) Location() string { func (be *Backend) Location() string {
return be.Join(be.cfg.Bucket, be.cfg.Prefix) return be.Join(be.cfg.Bucket, be.cfg.Prefix)

View File

@ -24,10 +24,11 @@ import (
// beSwift is a backend which stores the data on a swift endpoint. // beSwift is a backend which stores the data on a swift endpoint.
type beSwift struct { type beSwift struct {
conn *swift.Connection conn *swift.Connection
sem *backend.Semaphore connections uint
container string // Container name sem *backend.Semaphore
prefix string // Prefix of object names in the container container string // Container name
prefix string // Prefix of object names in the container
backend.Layout backend.Layout
} }
@ -68,9 +69,10 @@ func Open(ctx context.Context, cfg Config, rt http.RoundTripper) (restic.Backend
Transport: rt, Transport: rt,
}, },
sem: sem, connections: cfg.Connections,
container: cfg.Container, sem: sem,
prefix: cfg.Prefix, container: cfg.Container,
prefix: cfg.Prefix,
Layout: &backend.DefaultLayout{ Layout: &backend.DefaultLayout{
Path: cfg.Prefix, Path: cfg.Prefix,
Join: path.Join, Join: path.Join,
@ -113,6 +115,10 @@ func (be *beSwift) createContainer(ctx context.Context, policy string) error {
return be.conn.ContainerCreate(ctx, be.container, h) return be.conn.ContainerCreate(ctx, be.container, h)
} }
func (be *beSwift) Connections() uint {
return be.connections
}
// Location returns this backend's location (the container name). // Location returns this backend's location (the container name).
func (be *beSwift) Location() string { func (be *beSwift) Location() string {
return be.container return be.container

View File

@ -11,17 +11,18 @@ import (
// Backend implements a mock backend. // Backend implements a mock backend.
type Backend struct { type Backend struct {
CloseFn func() error CloseFn func() error
IsNotExistFn func(err error) bool IsNotExistFn func(err error) bool
SaveFn func(ctx context.Context, h restic.Handle, rd restic.RewindReader) error SaveFn func(ctx context.Context, h restic.Handle, rd restic.RewindReader) error
OpenReaderFn func(ctx context.Context, h restic.Handle, length int, offset int64) (io.ReadCloser, error) OpenReaderFn func(ctx context.Context, h restic.Handle, length int, offset int64) (io.ReadCloser, error)
StatFn func(ctx context.Context, h restic.Handle) (restic.FileInfo, error) StatFn func(ctx context.Context, h restic.Handle) (restic.FileInfo, error)
ListFn func(ctx context.Context, t restic.FileType, fn func(restic.FileInfo) error) error ListFn func(ctx context.Context, t restic.FileType, fn func(restic.FileInfo) error) error
RemoveFn func(ctx context.Context, h restic.Handle) error RemoveFn func(ctx context.Context, h restic.Handle) error
TestFn func(ctx context.Context, h restic.Handle) (bool, error) TestFn func(ctx context.Context, h restic.Handle) (bool, error)
DeleteFn func(ctx context.Context) error DeleteFn func(ctx context.Context) error
LocationFn func() string ConnectionsFn func() uint
HasherFn func() hash.Hash LocationFn func() string
HasherFn func() hash.Hash
} }
// NewBackend returns new mock Backend instance // NewBackend returns new mock Backend instance
@ -39,6 +40,14 @@ func (m *Backend) Close() error {
return m.CloseFn() return m.CloseFn()
} }
func (m *Backend) Connections() uint {
if m.ConnectionsFn == nil {
return 2
}
return m.ConnectionsFn()
}
// Location returns a location string. // Location returns a location string.
func (m *Backend) Location() string { func (m *Backend) Location() string {
if m.LocationFn == nil { if m.LocationFn == nil {

View File

@ -18,6 +18,9 @@ type Backend interface {
// repository. // repository.
Location() string Location() string
// Connections returns the maxmimum number of concurrent backend operations.
Connections() uint
// Hasher may return a hash function for calculating a content hash for the backend // Hasher may return a hash function for calculating a content hash for the backend
Hasher() hash.Hash Hasher() hash.Hash