mirror of
https://github.com/kopia/kopia.git
synced 2026-01-23 22:07:54 -05:00
99eeb3c063
Supported are:
```
$ kopia throttle set \
--download-bytes-per-second=N | unlimited
--upload-bytes-per-second=N | unlimited
--read-requests-per-second=N | unlimited
--write-requests-per-second=N | unlimited
--list-requests-per-second=N | unlimited
--concurrent-reads=N | unlimited
--concurrent-writes=N | unlimited
```
To change parameters of a running server use:
```
$ kopia server throttle set \
--address=<server-url> \
--server-control-password=<password> \
--download-bytes-per-second=N | unlimited
--upload-bytes-per-second=N | unlimited
--read-requests-per-second=N | unlimited
--write-requests-per-second=N | unlimited
--list-requests-per-second=N | unlimited
--concurrent-reads=N | unlimited
--concurrent-writes=N | unlimited
```
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package cli
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/kopia/kopia/internal/apiclient"
|
|
"github.com/kopia/kopia/internal/serverapi"
|
|
"github.com/kopia/kopia/repo/blob/throttling"
|
|
)
|
|
|
|
type commandServerThrottleSet struct {
|
|
sf serverClientFlags
|
|
|
|
cts commonThrottleSet
|
|
}
|
|
|
|
func (c *commandServerThrottleSet) setup(svc appServices, parent commandParent) {
|
|
cmd := parent.Command("set", "Set throttling parameters for a running server")
|
|
c.sf.setup(cmd)
|
|
c.cts.setup(cmd)
|
|
|
|
cmd.Action(svc.serverAction(&c.sf, c.run))
|
|
}
|
|
|
|
func (c *commandServerThrottleSet) run(ctx context.Context, cli *apiclient.KopiaAPIClient) error {
|
|
var limits throttling.Limits
|
|
|
|
if err := cli.Get(ctx, "control/throttle", nil, &limits); err != nil {
|
|
return errors.Wrap(err, "unable to get current throttle")
|
|
}
|
|
|
|
var changeCount int
|
|
|
|
if err := c.cts.apply(ctx, &limits, &changeCount); err != nil {
|
|
return err
|
|
}
|
|
|
|
if changeCount == 0 {
|
|
log(ctx).Infof("No changes made.")
|
|
return nil
|
|
}
|
|
|
|
if err := cli.Put(ctx, "control/throttle", &limits, &serverapi.Empty{}); err != nil {
|
|
return errors.Wrap(err, "unable to change throttle")
|
|
}
|
|
|
|
return nil
|
|
}
|