mirror of
				https://gitee.com/gitea/gitea
				synced 2025-11-04 08:30:25 +08:00 
			
		
		
		
	Replace #22751 1. only support the default branch in the repository setting. 2. autoload schedule data from the schedule table after starting the service. 3. support specific syntax like `@yearly`, `@monthly`, `@weekly`, `@daily`, `@hourly` ## How to use See the [GitHub Actions document](https://docs.github.com/en/actions/using-workflows/events-that-trigger-workflows#schedule) for getting more detailed information. ```yaml on: schedule: - cron: '30 5 * * 1,3' - cron: '30 5 * * 2,4' jobs: test_schedule: runs-on: ubuntu-latest steps: - name: Not on Monday or Wednesday if: github.event.schedule != '30 5 * * 1,3' run: echo "This step will be skipped on Monday and Wednesday" - name: Every time run: echo "This step will always run" ``` Signed-off-by: Bo-Yi.Wu <appleboy.tw@gmail.com> --------- Co-authored-by: Jason Song <i@wolfogre.com> Co-authored-by: techknowlogick <techknowlogick@gitea.io> Co-authored-by: wxiaoguang <wxiaoguang@gmail.com> Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
		
			
				
	
	
		
			121 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			121 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2023 The Gitea Authors. All rights reserved.
 | 
						|
// SPDX-License-Identifier: MIT
 | 
						|
 | 
						|
package actions
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"time"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/models/db"
 | 
						|
	repo_model "code.gitea.io/gitea/models/repo"
 | 
						|
	user_model "code.gitea.io/gitea/models/user"
 | 
						|
	"code.gitea.io/gitea/modules/timeutil"
 | 
						|
	webhook_module "code.gitea.io/gitea/modules/webhook"
 | 
						|
 | 
						|
	"github.com/robfig/cron/v3"
 | 
						|
)
 | 
						|
 | 
						|
// ActionSchedule represents a schedule of a workflow file
 | 
						|
type ActionSchedule struct {
 | 
						|
	ID            int64
 | 
						|
	Title         string
 | 
						|
	Specs         []string
 | 
						|
	RepoID        int64                  `xorm:"index"`
 | 
						|
	Repo          *repo_model.Repository `xorm:"-"`
 | 
						|
	OwnerID       int64                  `xorm:"index"`
 | 
						|
	WorkflowID    string
 | 
						|
	TriggerUserID int64
 | 
						|
	TriggerUser   *user_model.User `xorm:"-"`
 | 
						|
	Ref           string
 | 
						|
	CommitSHA     string
 | 
						|
	Event         webhook_module.HookEventType
 | 
						|
	EventPayload  string `xorm:"LONGTEXT"`
 | 
						|
	Content       []byte
 | 
						|
	Created       timeutil.TimeStamp `xorm:"created"`
 | 
						|
	Updated       timeutil.TimeStamp `xorm:"updated"`
 | 
						|
}
 | 
						|
 | 
						|
func init() {
 | 
						|
	db.RegisterModel(new(ActionSchedule))
 | 
						|
}
 | 
						|
 | 
						|
// GetSchedulesMapByIDs returns the schedules by given id slice.
 | 
						|
func GetSchedulesMapByIDs(ids []int64) (map[int64]*ActionSchedule, error) {
 | 
						|
	schedules := make(map[int64]*ActionSchedule, len(ids))
 | 
						|
	return schedules, db.GetEngine(db.DefaultContext).In("id", ids).Find(&schedules)
 | 
						|
}
 | 
						|
 | 
						|
// GetReposMapByIDs returns the repos by given id slice.
 | 
						|
func GetReposMapByIDs(ids []int64) (map[int64]*repo_model.Repository, error) {
 | 
						|
	repos := make(map[int64]*repo_model.Repository, len(ids))
 | 
						|
	return repos, db.GetEngine(db.DefaultContext).In("id", ids).Find(&repos)
 | 
						|
}
 | 
						|
 | 
						|
var cronParser = cron.NewParser(cron.Minute | cron.Hour | cron.Dom | cron.Month | cron.Dow | cron.Descriptor)
 | 
						|
 | 
						|
// CreateScheduleTask creates new schedule task.
 | 
						|
func CreateScheduleTask(ctx context.Context, rows []*ActionSchedule) error {
 | 
						|
	// Return early if there are no rows to insert
 | 
						|
	if len(rows) == 0 {
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
 | 
						|
	// Begin transaction
 | 
						|
	ctx, committer, err := db.TxContext(ctx)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	defer committer.Close()
 | 
						|
 | 
						|
	// Loop through each schedule row
 | 
						|
	for _, row := range rows {
 | 
						|
		// Create new schedule row
 | 
						|
		if err = db.Insert(ctx, row); err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
 | 
						|
		// Loop through each schedule spec and create a new spec row
 | 
						|
		now := time.Now()
 | 
						|
 | 
						|
		for _, spec := range row.Specs {
 | 
						|
			// Parse the spec and check for errors
 | 
						|
			schedule, err := cronParser.Parse(spec)
 | 
						|
			if err != nil {
 | 
						|
				continue // skip to the next spec if there's an error
 | 
						|
			}
 | 
						|
 | 
						|
			// Insert the new schedule spec row
 | 
						|
			if err = db.Insert(ctx, &ActionScheduleSpec{
 | 
						|
				RepoID:     row.RepoID,
 | 
						|
				ScheduleID: row.ID,
 | 
						|
				Spec:       spec,
 | 
						|
				Next:       timeutil.TimeStamp(schedule.Next(now).Unix()),
 | 
						|
			}); err != nil {
 | 
						|
				return err
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	// Commit transaction
 | 
						|
	return committer.Commit()
 | 
						|
}
 | 
						|
 | 
						|
func DeleteScheduleTaskByRepo(ctx context.Context, id int64) error {
 | 
						|
	ctx, committer, err := db.TxContext(ctx)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	defer committer.Close()
 | 
						|
 | 
						|
	if _, err := db.GetEngine(ctx).Delete(&ActionSchedule{RepoID: id}); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	if _, err := db.GetEngine(ctx).Delete(&ActionScheduleSpec{RepoID: id}); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	return committer.Commit()
 | 
						|
}
 |