woodpecker/runner/builtin/updater.go

package builtin

import (
	"encoding/json"
	"io"
	"io/ioutil"

	"github.com/drone/drone/common"
	"github.com/drone/drone/datastore"
	"github.com/drone/drone/eventbus"
)

type Updater interface {
	SetBuild(*common.Repo, *common.Build) error
	SetTask(*common.Repo, *common.Build, *common.Task) error
	SetLogs(*common.Repo, *common.Build, *common.Task, io.ReadCloser) error
}

// NewUpdater returns an implementation of the Updater interface
// that directly modifies the database and sends messages to the bus.
func NewUpdater(bus eventbus.Bus, store datastore.Datastore) Updater {
	return &updater{bus, store}
}

type updater struct {
	bus   eventbus.Bus
	store datastore.Datastore
}

func (u *updater) SetBuild(r *common.Repo, b *common.Build) error {
	err := u.store.SetBuildState(r.FullName, b)
	if err != nil {
		return err
	}

	// if the build is complete we may need to update
	if b.State != common.StatePending && b.State != common.StateRunning {
		repo, err := u.store.Repo(r.FullName)
		if err == nil {
			if repo.Last == nil || b.Number >= repo.Last.Number {
				repo.Last = b
				u.store.SetRepo(repo)
			}
		}
	}

	msg, err := json.Marshal(b)
	if err != nil {
		return err
	}

	u.bus.Send(&eventbus.Event{
		Name: r.FullName,
		Kind: eventbus.EventRepo,
		Msg:  msg,
	})
	return nil
}

func (u *updater) SetTask(r *common.Repo, b *common.Build, t *common.Task) error {
	err := u.store.SetBuildTask(r.FullName, b.Number, t)
	if err != nil {
		return err
	}

	msg, err := json.Marshal(b)
	if err != nil {
		return err
	}

	u.bus.Send(&eventbus.Event{
		Name: r.FullName,
		Kind: eventbus.EventRepo,
		Msg:  msg,
	})
	return nil
}

func (u *updater) SetLogs(r *common.Repo, b *common.Build, t *common.Task, rc io.ReadCloser) error {
	defer rc.Close()
	out, err := ioutil.ReadAll(rc)
	if err != nil {
		return err
	}
	return u.store.SetLogs(r.FullName, b.Number, t.Number, out)
}