reset_usage_job.go

// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
// Licensed under the GNU Affero General Public License (AGPL).
// See License.AGPL.txt in the project root for license information.

package scheduler

import (
	"context"
	"fmt"
	"time"

	"github.com/gitpod-io/gitpod/common-go/log"
	v1 "github.com/gitpod-io/gitpod/usage-api/v1"
	"github.com/robfig/cron"
)

func NewResetUsageJob(schedule time.Duration, clientsConstructor ClientsConstructor) (JobSpec, error) {
	job := &ResetUsageJobSpec{
		clientsConstructor: clientsConstructor,
	}

	parsed, err := cron.Parse(fmt.Sprintf("@every %s", schedule.String()))
	if err != nil {
		return JobSpec{}, fmt.Errorf("failed to parse period into schedule: %w", err)
	}

	return JobSpec{
		Job:                 job,
		ID:                  "reset_usage",
		Schedule:            parsed,
		InitialLockDuration: schedule,
	}, nil

}

type ResetUsageJobSpec struct {
	clientsConstructor ClientsConstructor
}

func (j *ResetUsageJobSpec) Run() (err error) {
	ctx := context.Background()

	log.Info("Running reset usage job.")
	usageClient, _, err := j.clientsConstructor()
	if err != nil {
		return fmt.Errorf("Failed to construct reset usage job clients: %w", err)
	}

	_, err = usageClient.ResetUsage(ctx, &v1.ResetUsageRequest{})
	if err != nil {
		return fmt.Errorf("failed to reset usage: %w", err)
	}

	return nil
}