mirror of
https://github.com/octoleo/restic.git
synced 2024-12-24 20:00:21 +00:00
109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
// Copyright 2017, Google
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package b2
|
|
|
|
import "fmt"
|
|
|
|
// StatusInfo reports information about a client.
|
|
type StatusInfo struct {
|
|
Writers map[string]*WriterStatus
|
|
Readers map[string]*ReaderStatus
|
|
MethodCalls map[string]int
|
|
}
|
|
|
|
// WriterStatus reports the status for each writer.
|
|
type WriterStatus struct {
|
|
// Progress is a slice of completion ratios. The index of a ratio is its
|
|
// chunk id less one.
|
|
Progress []float64
|
|
}
|
|
|
|
// ReaderStatus reports the status for each reader.
|
|
type ReaderStatus struct {
|
|
// Progress is a slice of completion ratios. The index of a ratio is its
|
|
// chunk id less one.
|
|
Progress []float64
|
|
}
|
|
|
|
// Status returns information about the current state of the client.
|
|
func (c *Client) Status() *StatusInfo {
|
|
c.slock.Lock()
|
|
defer c.slock.Unlock()
|
|
|
|
si := &StatusInfo{
|
|
Writers: make(map[string]*WriterStatus),
|
|
Readers: make(map[string]*ReaderStatus),
|
|
MethodCalls: make(map[string]int),
|
|
}
|
|
|
|
for name, w := range c.sWriters {
|
|
si.Writers[name] = w.status()
|
|
}
|
|
|
|
for name, r := range c.sReaders {
|
|
si.Readers[name] = r.status()
|
|
}
|
|
|
|
for name, n := range c.sMethods {
|
|
si.MethodCalls[name] = n
|
|
}
|
|
|
|
return si
|
|
}
|
|
|
|
func (c *Client) addWriter(w *Writer) {
|
|
c.slock.Lock()
|
|
defer c.slock.Unlock()
|
|
|
|
if c.sWriters == nil {
|
|
c.sWriters = make(map[string]*Writer)
|
|
}
|
|
|
|
c.sWriters[fmt.Sprintf("%s/%s", w.o.b.Name(), w.name)] = w
|
|
}
|
|
|
|
func (c *Client) removeWriter(w *Writer) {
|
|
c.slock.Lock()
|
|
defer c.slock.Unlock()
|
|
|
|
if c.sWriters == nil {
|
|
return
|
|
}
|
|
|
|
delete(c.sWriters, fmt.Sprintf("%s/%s", w.o.b.Name(), w.name))
|
|
}
|
|
|
|
func (c *Client) addReader(r *Reader) {
|
|
c.slock.Lock()
|
|
defer c.slock.Unlock()
|
|
|
|
if c.sReaders == nil {
|
|
c.sReaders = make(map[string]*Reader)
|
|
}
|
|
|
|
c.sReaders[fmt.Sprintf("%s/%s", r.o.b.Name(), r.name)] = r
|
|
}
|
|
|
|
func (c *Client) removeReader(r *Reader) {
|
|
c.slock.Lock()
|
|
defer c.slock.Unlock()
|
|
|
|
if c.sReaders == nil {
|
|
return
|
|
}
|
|
|
|
delete(c.sReaders, fmt.Sprintf("%s/%s", r.o.b.Name(), r.name))
|
|
}
|