mirror of
https://github.com/kopia/kopia.git
synced 2026-01-06 05:27:59 -05:00
8515d050e5
* feat(infra): improved support for in-process testing * support for killing of a running server using simulated Ctrl-C * support for overriding os.Stdin * migrated many tests from the exe runner to in-process runner * added required indirection when defining Envar() so we can later override it in tests * refactored CLI runners by moving environment overrides to CLITestEnv
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(svc, 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
|
|
}
|