2018-02-19 22:24:10 +00:00
|
|
|
// Copyright 2018 Drone.IO Inc.
|
2018-03-21 13:02:17 +00:00
|
|
|
//
|
2018-02-19 22:24:10 +00:00
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
2018-03-21 13:02:17 +00:00
|
|
|
//
|
2018-02-19 22:24:10 +00:00
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
2018-03-21 13:02:17 +00:00
|
|
|
//
|
2018-02-19 22:24:10 +00:00
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
|
2017-03-05 07:56:08 +00:00
|
|
|
package server
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"context"
|
|
|
|
"encoding/json"
|
2017-04-12 17:56:30 +00:00
|
|
|
"fmt"
|
2017-03-05 07:56:08 +00:00
|
|
|
"log"
|
|
|
|
"strconv"
|
2017-09-08 23:45:17 +00:00
|
|
|
"time"
|
2017-03-05 07:56:08 +00:00
|
|
|
|
2017-06-28 17:21:22 +00:00
|
|
|
oldcontext "golang.org/x/net/context"
|
|
|
|
|
2017-07-19 21:46:03 +00:00
|
|
|
"google.golang.org/grpc/metadata"
|
|
|
|
|
2019-08-27 11:01:29 +00:00
|
|
|
"github.com/laszlocph/woodpecker/cncd/logging"
|
|
|
|
"github.com/laszlocph/woodpecker/cncd/pipeline/pipeline/rpc"
|
|
|
|
"github.com/laszlocph/woodpecker/cncd/pipeline/pipeline/rpc/proto"
|
|
|
|
"github.com/laszlocph/woodpecker/cncd/pubsub"
|
|
|
|
"github.com/laszlocph/woodpecker/cncd/queue"
|
2019-06-28 12:23:52 +00:00
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
|
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
2020-05-18 14:46:13 +00:00
|
|
|
"github.com/sirupsen/logrus"
|
2017-03-05 07:56:08 +00:00
|
|
|
|
2019-08-27 11:01:29 +00:00
|
|
|
"github.com/laszlocph/woodpecker/model"
|
|
|
|
"github.com/laszlocph/woodpecker/remote"
|
|
|
|
"github.com/laszlocph/woodpecker/store"
|
2017-09-08 23:45:17 +00:00
|
|
|
|
|
|
|
"github.com/drone/expr"
|
2017-03-05 07:56:08 +00:00
|
|
|
)
|
|
|
|
|
2017-04-11 17:06:45 +00:00
|
|
|
var Config = struct {
|
|
|
|
Services struct {
|
|
|
|
Pubsub pubsub.Publisher
|
|
|
|
Queue queue.Queue
|
|
|
|
Logs logging.Log
|
|
|
|
Senders model.SenderService
|
|
|
|
Secrets model.SecretService
|
|
|
|
Registries model.RegistryService
|
2017-06-26 19:27:53 +00:00
|
|
|
Environ model.EnvironService
|
2017-04-11 17:06:45 +00:00
|
|
|
}
|
|
|
|
Storage struct {
|
|
|
|
// Users model.UserStore
|
|
|
|
// Repos model.RepoStore
|
|
|
|
// Builds model.BuildStore
|
|
|
|
// Logs model.LogStore
|
2017-05-05 16:59:37 +00:00
|
|
|
Config model.ConfigStore
|
|
|
|
Files model.FileStore
|
|
|
|
Procs model.ProcStore
|
2017-04-11 17:06:45 +00:00
|
|
|
// Registries model.RegistryStore
|
|
|
|
// Secrets model.SecretStore
|
|
|
|
}
|
|
|
|
Server struct {
|
2017-08-17 19:37:52 +00:00
|
|
|
Key string
|
|
|
|
Cert string
|
|
|
|
Host string
|
|
|
|
Port string
|
|
|
|
Pass string
|
2017-09-12 23:57:25 +00:00
|
|
|
RepoConfig string
|
2017-08-17 19:37:52 +00:00
|
|
|
SessionExpires time.Duration
|
2017-04-11 17:06:45 +00:00
|
|
|
// Open bool
|
|
|
|
// Orgs map[string]struct{}
|
|
|
|
// Admins map[string]struct{}
|
|
|
|
}
|
2018-03-21 12:51:54 +00:00
|
|
|
Prometheus struct {
|
|
|
|
AuthToken string
|
|
|
|
}
|
2017-04-11 17:06:45 +00:00
|
|
|
Pipeline struct {
|
2017-06-24 12:16:16 +00:00
|
|
|
Limits model.ResourceLimit
|
2017-04-11 17:06:45 +00:00
|
|
|
Volumes []string
|
|
|
|
Networks []string
|
|
|
|
Privileged []string
|
|
|
|
}
|
|
|
|
}{}
|
|
|
|
|
2017-03-05 07:56:08 +00:00
|
|
|
type RPC struct {
|
2019-06-28 12:23:52 +00:00
|
|
|
remote remote.Remote
|
|
|
|
queue queue.Queue
|
|
|
|
pubsub pubsub.Publisher
|
|
|
|
logger logging.Log
|
|
|
|
store store.Store
|
|
|
|
host string
|
|
|
|
buildTime *prometheus.GaugeVec
|
|
|
|
buildCount *prometheus.CounterVec
|
2017-03-05 07:56:08 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Next implements the rpc.Next function
|
2017-03-12 08:46:59 +00:00
|
|
|
func (s *RPC) Next(c context.Context, filter rpc.Filter) (*rpc.Pipeline, error) {
|
2020-05-18 14:46:13 +00:00
|
|
|
metadata, ok := metadata.FromIncomingContext(c)
|
2017-07-20 16:21:15 +00:00
|
|
|
if ok {
|
|
|
|
hostname, ok := metadata["hostname"]
|
|
|
|
if ok && len(hostname) != 0 {
|
|
|
|
logrus.Debugf("agent connected: %s: polling", hostname[0])
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-09-08 23:45:17 +00:00
|
|
|
fn, err := createFilterFunc(filter)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
2017-03-12 08:46:59 +00:00
|
|
|
}
|
2019-06-16 13:56:32 +00:00
|
|
|
for {
|
|
|
|
task, err := s.queue.Poll(c, fn)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
} else if task == nil {
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
if task.ShouldRun() {
|
|
|
|
pipeline := new(rpc.Pipeline)
|
|
|
|
err = json.Unmarshal(task.Data, pipeline)
|
|
|
|
return pipeline, err
|
|
|
|
} else {
|
|
|
|
s.Done(c, task.ID, rpc.State{})
|
|
|
|
}
|
|
|
|
}
|
2017-03-05 07:56:08 +00:00
|
|
|
}
|
|
|
|
|
2017-03-05 11:05:16 +00:00
|
|
|
// Wait implements the rpc.Wait function
|
|
|
|
func (s *RPC) Wait(c context.Context, id string) error {
|
|
|
|
return s.queue.Wait(c, id)
|
2017-03-05 07:56:08 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Extend implements the rpc.Extend function
|
|
|
|
func (s *RPC) Extend(c context.Context, id string) error {
|
|
|
|
return s.queue.Extend(c, id)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Update implements the rpc.Update function
|
|
|
|
func (s *RPC) Update(c context.Context, id string, state rpc.State) error {
|
2017-04-01 11:17:04 +00:00
|
|
|
procID, err := strconv.ParseInt(id, 10, 64)
|
2017-03-05 07:56:08 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2017-04-03 09:34:37 +00:00
|
|
|
pproc, err := s.store.ProcLoad(procID)
|
2017-03-05 07:56:08 +00:00
|
|
|
if err != nil {
|
2017-04-03 09:34:37 +00:00
|
|
|
log.Printf("error: rpc.update: cannot find pproc with id %d: %s", procID, err)
|
2017-03-05 07:56:08 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2017-04-03 09:34:37 +00:00
|
|
|
build, err := s.store.GetBuild(pproc.BuildID)
|
2017-03-05 07:56:08 +00:00
|
|
|
if err != nil {
|
2017-04-03 09:34:37 +00:00
|
|
|
log.Printf("error: cannot find build with id %d: %s", pproc.BuildID, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
proc, err := s.store.ProcChild(build, pproc.PID, state.Proc)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find proc with name %s: %s", state.Proc, err)
|
2017-03-05 07:56:08 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2020-05-18 14:46:13 +00:00
|
|
|
metadata, ok := metadata.FromIncomingContext(c)
|
2017-07-20 03:07:29 +00:00
|
|
|
if ok {
|
|
|
|
hostname, ok := metadata["hostname"]
|
|
|
|
if ok && len(hostname) != 0 {
|
|
|
|
proc.Machine = hostname[0]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-03-05 07:56:08 +00:00
|
|
|
repo, err := s.store.GetRepo(build.RepoID)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find repo with id %d: %s", build.RepoID, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2019-09-15 05:29:45 +00:00
|
|
|
if proc, err = UpdateProcStatus(s.store, *proc, state, build.Started); err != nil {
|
2017-04-01 11:17:04 +00:00
|
|
|
log.Printf("error: rpc.update: cannot update proc: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
build.Procs, _ = s.store.ProcList(build)
|
2017-04-04 09:30:06 +00:00
|
|
|
build.Procs = model.Tree(build.Procs)
|
2017-04-01 11:17:04 +00:00
|
|
|
message := pubsub.Message{
|
|
|
|
Labels: map[string]string{
|
|
|
|
"repo": repo.FullName,
|
|
|
|
"private": strconv.FormatBool(repo.IsPrivate),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
message.Data, _ = json.Marshal(model.Event{
|
|
|
|
Repo: *repo,
|
|
|
|
Build: *build,
|
|
|
|
})
|
|
|
|
s.pubsub.Publish(c, "topic/events", message)
|
2017-03-05 07:56:08 +00:00
|
|
|
|
2017-04-01 11:17:04 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Upload implements the rpc.Upload function
|
|
|
|
func (s *RPC) Upload(c context.Context, id string, file *rpc.File) error {
|
|
|
|
procID, err := strconv.ParseInt(id, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
2017-03-05 07:56:08 +00:00
|
|
|
}
|
|
|
|
|
2017-04-03 09:34:37 +00:00
|
|
|
pproc, err := s.store.ProcLoad(procID)
|
2017-04-01 11:17:04 +00:00
|
|
|
if err != nil {
|
2017-04-03 09:34:37 +00:00
|
|
|
log.Printf("error: cannot find parent proc with id %d: %s", procID, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
build, err := s.store.GetBuild(pproc.BuildID)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find build with id %d: %s", pproc.BuildID, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
proc, err := s.store.ProcChild(build, pproc.PID, file.Proc)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find child proc with name %s: %s", file.Proc, err)
|
2017-04-01 11:17:04 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2017-04-03 09:34:37 +00:00
|
|
|
if file.Mime == "application/json+logs" {
|
|
|
|
return s.store.LogSave(
|
|
|
|
proc,
|
|
|
|
bytes.NewBuffer(file.Data),
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2017-08-02 20:04:00 +00:00
|
|
|
report := &model.File{
|
2017-04-01 11:17:04 +00:00
|
|
|
BuildID: proc.BuildID,
|
|
|
|
ProcID: proc.ID,
|
2017-07-27 17:06:24 +00:00
|
|
|
PID: proc.PID,
|
2017-04-01 11:17:04 +00:00
|
|
|
Mime: file.Mime,
|
|
|
|
Name: file.Name,
|
|
|
|
Size: file.Size,
|
|
|
|
Time: file.Time,
|
2017-08-02 20:04:00 +00:00
|
|
|
}
|
|
|
|
if d, ok := file.Meta["X-Tests-Passed"]; ok {
|
|
|
|
report.Passed, _ = strconv.Atoi(d)
|
|
|
|
}
|
|
|
|
if d, ok := file.Meta["X-Tests-Failed"]; ok {
|
|
|
|
report.Failed, _ = strconv.Atoi(d)
|
|
|
|
}
|
|
|
|
if d, ok := file.Meta["X-Tests-Skipped"]; ok {
|
|
|
|
report.Skipped, _ = strconv.Atoi(d)
|
|
|
|
}
|
|
|
|
|
|
|
|
if d, ok := file.Meta["X-Checks-Passed"]; ok {
|
|
|
|
report.Passed, _ = strconv.Atoi(d)
|
|
|
|
}
|
|
|
|
if d, ok := file.Meta["X-Checks-Failed"]; ok {
|
|
|
|
report.Failed, _ = strconv.Atoi(d)
|
|
|
|
}
|
|
|
|
|
|
|
|
if d, ok := file.Meta["X-Coverage-Lines"]; ok {
|
|
|
|
report.Passed, _ = strconv.Atoi(d)
|
|
|
|
}
|
|
|
|
if d, ok := file.Meta["X-Coverage-Total"]; ok {
|
|
|
|
if total, _ := strconv.Atoi(d); total != 0 {
|
|
|
|
report.Failed = total - report.Passed
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return Config.Storage.Files.FileCreate(
|
|
|
|
report,
|
2017-04-01 11:17:04 +00:00
|
|
|
bytes.NewBuffer(file.Data),
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Init implements the rpc.Init function
|
|
|
|
func (s *RPC) Init(c context.Context, id string, state rpc.State) error {
|
|
|
|
procID, err := strconv.ParseInt(id, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
proc, err := s.store.ProcLoad(procID)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find proc with id %d: %s", procID, err)
|
|
|
|
return err
|
|
|
|
}
|
2020-05-18 14:46:13 +00:00
|
|
|
metadata, ok := metadata.FromIncomingContext(c)
|
2017-07-19 21:46:03 +00:00
|
|
|
if ok {
|
|
|
|
hostname, ok := metadata["hostname"]
|
|
|
|
if ok && len(hostname) != 0 {
|
|
|
|
proc.Machine = hostname[0]
|
|
|
|
}
|
|
|
|
}
|
2017-04-01 11:17:04 +00:00
|
|
|
|
|
|
|
build, err := s.store.GetBuild(proc.BuildID)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find build with id %d: %s", proc.BuildID, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
repo, err := s.store.GetRepo(build.RepoID)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find repo with id %d: %s", build.RepoID, err)
|
|
|
|
return err
|
|
|
|
}
|
2017-03-05 07:56:08 +00:00
|
|
|
|
|
|
|
if build.Status == model.StatusPending {
|
2019-09-14 12:12:04 +00:00
|
|
|
if build, err = UpdateToStatusRunning(s.store, *build, state.Started); err != nil {
|
2017-04-01 11:17:04 +00:00
|
|
|
log.Printf("error: init: cannot update build_id %d state: %s", build.ID, err)
|
|
|
|
}
|
2017-03-05 07:56:08 +00:00
|
|
|
}
|
|
|
|
|
2017-04-01 11:17:04 +00:00
|
|
|
defer func() {
|
|
|
|
build.Procs, _ = s.store.ProcList(build)
|
|
|
|
message := pubsub.Message{
|
|
|
|
Labels: map[string]string{
|
|
|
|
"repo": repo.FullName,
|
|
|
|
"private": strconv.FormatBool(repo.IsPrivate),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
message.Data, _ = json.Marshal(model.Event{
|
|
|
|
Repo: *repo,
|
|
|
|
Build: *build,
|
|
|
|
})
|
|
|
|
s.pubsub.Publish(c, "topic/events", message)
|
|
|
|
}()
|
2017-03-05 07:56:08 +00:00
|
|
|
|
2019-09-15 05:29:45 +00:00
|
|
|
_, err = UpdateProcToStatusStarted(s.store, *proc, state)
|
|
|
|
return err
|
2017-04-01 11:17:04 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Done implements the rpc.Done function
|
|
|
|
func (s *RPC) Done(c context.Context, id string, state rpc.State) error {
|
|
|
|
procID, err := strconv.ParseInt(id, 10, 64)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2017-03-05 07:56:08 +00:00
|
|
|
|
2017-04-01 11:17:04 +00:00
|
|
|
proc, err := s.store.ProcLoad(procID)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find proc with id %d: %s", procID, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
build, err := s.store.GetBuild(proc.BuildID)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find build with id %d: %s", proc.BuildID, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
repo, err := s.store.GetRepo(build.RepoID)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error: cannot find repo with id %d: %s", build.RepoID, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2019-09-15 05:29:45 +00:00
|
|
|
if proc, err = UpdateProcStatusToDone(s.store, *proc, state); err != nil {
|
|
|
|
log.Printf("error: done: cannot update proc_id %d state: %s", proc.ID, err)
|
|
|
|
}
|
2019-06-16 08:54:31 +00:00
|
|
|
|
2019-06-16 13:56:32 +00:00
|
|
|
var queueErr error
|
|
|
|
if proc.Failing() {
|
|
|
|
queueErr = s.queue.Error(c, id, fmt.Errorf("Proc finished with exitcode %d, %s", state.ExitCode, state.Error))
|
|
|
|
} else {
|
2019-07-22 10:43:59 +00:00
|
|
|
queueErr = s.queue.Done(c, id, proc.State)
|
2019-06-16 13:56:32 +00:00
|
|
|
}
|
|
|
|
if queueErr != nil {
|
2019-06-16 08:54:31 +00:00
|
|
|
log.Printf("error: done: cannot ack proc_id %d: %s", procID, err)
|
|
|
|
}
|
|
|
|
|
|
|
|
procs, _ := s.store.ProcList(build)
|
|
|
|
s.completeChildrenIfParentCompleted(procs, proc)
|
|
|
|
|
|
|
|
if !isThereRunningStage(procs) {
|
2019-09-14 12:12:04 +00:00
|
|
|
if build, err = UpdateStatusToDone(s.store, *build, buildStatus(procs), proc.Stopped); err != nil {
|
2019-06-16 08:54:31 +00:00
|
|
|
log.Printf("error: done: cannot update build_id %d final state: %s", build.ID, err)
|
|
|
|
}
|
|
|
|
|
2019-06-17 08:48:40 +00:00
|
|
|
if !isMultiPipeline(procs) {
|
|
|
|
s.updateRemoteStatus(repo, build, nil)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if isMultiPipeline(procs) {
|
|
|
|
s.updateRemoteStatus(repo, build, proc)
|
2019-06-16 08:54:31 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if err := s.logger.Close(c, id); err != nil {
|
|
|
|
log.Printf("error: done: cannot close build_id %d logger: %s", proc.ID, err)
|
|
|
|
}
|
|
|
|
|
|
|
|
s.notify(c, repo, build, procs)
|
|
|
|
|
2019-06-28 12:23:52 +00:00
|
|
|
if build.Status == model.StatusSuccess || build.Status == model.StatusFailure {
|
|
|
|
s.buildCount.WithLabelValues(repo.FullName, build.Branch, build.Status, "total").Inc()
|
|
|
|
s.buildTime.WithLabelValues(repo.FullName, build.Branch, build.Status, "total").Set(float64(build.Finished - build.Started))
|
|
|
|
}
|
|
|
|
if isMultiPipeline(procs) {
|
|
|
|
s.buildTime.WithLabelValues(repo.FullName, build.Branch, proc.State, proc.Name).Set(float64(proc.Stopped - proc.Started))
|
|
|
|
}
|
|
|
|
|
2019-06-16 08:54:31 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2019-06-17 08:48:40 +00:00
|
|
|
func isMultiPipeline(procs []*model.Proc) bool {
|
|
|
|
countPPIDZero := 0
|
|
|
|
for _, proc := range procs {
|
|
|
|
if proc.PPID == 0 {
|
|
|
|
countPPIDZero++
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return countPPIDZero > 1
|
|
|
|
}
|
|
|
|
|
2019-06-16 08:54:31 +00:00
|
|
|
// Log implements the rpc.Log function
|
|
|
|
func (s *RPC) Log(c context.Context, id string, line *rpc.Line) error {
|
|
|
|
entry := new(logging.Entry)
|
|
|
|
entry.Data, _ = json.Marshal(line)
|
|
|
|
s.logger.Write(c, id, entry)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *RPC) completeChildrenIfParentCompleted(procs []*model.Proc, completedProc *model.Proc) {
|
2017-04-01 11:17:04 +00:00
|
|
|
for _, p := range procs {
|
2019-06-16 08:54:31 +00:00
|
|
|
if p.Running() && p.PPID == completedProc.PID {
|
2019-09-15 05:29:45 +00:00
|
|
|
if _, err := UpdateProcToStatusSkipped(s.store, *p, completedProc.Stopped); err != nil {
|
2017-04-01 11:17:04 +00:00
|
|
|
log.Printf("error: done: cannot update proc_id %d child state: %s", p.ID, err)
|
|
|
|
}
|
2017-03-05 07:56:08 +00:00
|
|
|
}
|
2017-04-04 09:30:06 +00:00
|
|
|
}
|
2019-06-16 08:54:31 +00:00
|
|
|
}
|
2017-04-04 09:30:06 +00:00
|
|
|
|
2019-06-16 08:54:31 +00:00
|
|
|
func isThereRunningStage(procs []*model.Proc) bool {
|
2017-04-04 09:30:06 +00:00
|
|
|
for _, p := range procs {
|
|
|
|
if p.PPID == 0 {
|
|
|
|
if p.Running() {
|
2019-06-16 08:54:31 +00:00
|
|
|
return true
|
2017-04-04 09:30:06 +00:00
|
|
|
}
|
2019-06-16 08:54:31 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
func buildStatus(procs []*model.Proc) string {
|
|
|
|
status := model.StatusSuccess
|
|
|
|
|
|
|
|
for _, p := range procs {
|
|
|
|
if p.PPID == 0 {
|
2017-04-01 11:17:04 +00:00
|
|
|
if p.Failing() {
|
2017-04-04 09:30:06 +00:00
|
|
|
status = p.State
|
2017-04-01 11:17:04 +00:00
|
|
|
}
|
2017-03-05 07:56:08 +00:00
|
|
|
}
|
2017-04-01 11:17:04 +00:00
|
|
|
}
|
2017-04-12 17:56:30 +00:00
|
|
|
|
2019-06-16 08:54:31 +00:00
|
|
|
return status
|
|
|
|
}
|
|
|
|
|
2019-06-17 08:48:40 +00:00
|
|
|
func (s *RPC) updateRemoteStatus(repo *model.Repo, build *model.Build, proc *model.Proc) {
|
2019-06-16 08:54:31 +00:00
|
|
|
user, err := s.store.GetUser(repo.UserID)
|
|
|
|
if err == nil {
|
|
|
|
if refresher, ok := s.remote.(remote.Refresher); ok {
|
|
|
|
ok, _ := refresher.Refresh(user)
|
|
|
|
if ok {
|
|
|
|
s.store.UpdateUser(user)
|
2017-04-12 17:56:30 +00:00
|
|
|
}
|
|
|
|
}
|
2019-06-16 08:54:31 +00:00
|
|
|
uri := fmt.Sprintf("%s/%s/%d", s.host, repo.FullName, build.Number)
|
2019-06-17 08:48:40 +00:00
|
|
|
err = s.remote.Status(user, repo, build, uri, proc)
|
2019-06-16 08:54:31 +00:00
|
|
|
if err != nil {
|
|
|
|
logrus.Errorf("error setting commit status for %s/%d: %v", repo.FullName, build.Number, err)
|
|
|
|
}
|
2017-03-05 07:56:08 +00:00
|
|
|
}
|
2019-06-16 08:54:31 +00:00
|
|
|
}
|
2017-03-05 07:56:08 +00:00
|
|
|
|
2019-06-16 08:54:31 +00:00
|
|
|
func (s *RPC) notify(c context.Context, repo *model.Repo, build *model.Build, procs []*model.Proc) {
|
2017-04-04 09:30:06 +00:00
|
|
|
build.Procs = model.Tree(procs)
|
2017-04-01 11:17:04 +00:00
|
|
|
message := pubsub.Message{
|
|
|
|
Labels: map[string]string{
|
|
|
|
"repo": repo.FullName,
|
|
|
|
"private": strconv.FormatBool(repo.IsPrivate),
|
|
|
|
},
|
|
|
|
}
|
2017-03-05 07:56:08 +00:00
|
|
|
message.Data, _ = json.Marshal(model.Event{
|
|
|
|
Repo: *repo,
|
|
|
|
Build: *build,
|
|
|
|
})
|
|
|
|
s.pubsub.Publish(c, "topic/events", message)
|
|
|
|
}
|
2017-06-22 19:11:05 +00:00
|
|
|
|
2017-09-08 23:45:17 +00:00
|
|
|
func createFilterFunc(filter rpc.Filter) (queue.Filter, error) {
|
|
|
|
var st *expr.Selector
|
|
|
|
var err error
|
|
|
|
|
|
|
|
if filter.Expr != "" {
|
|
|
|
st, err = expr.ParseString(filter.Expr)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return func(task *queue.Task) bool {
|
|
|
|
if st != nil {
|
|
|
|
match, _ := st.Eval(expr.NewRow(task.Labels))
|
|
|
|
return match
|
|
|
|
}
|
|
|
|
|
|
|
|
for k, v := range filter.Labels {
|
|
|
|
if task.Labels[k] != v {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
2017-06-28 17:21:22 +00:00
|
|
|
//
|
|
|
|
//
|
|
|
|
//
|
|
|
|
|
|
|
|
// DroneServer is a grpc server implementation.
|
|
|
|
type DroneServer struct {
|
2019-06-28 12:23:52 +00:00
|
|
|
peer RPC
|
2017-06-28 17:21:22 +00:00
|
|
|
}
|
|
|
|
|
2019-06-28 12:23:52 +00:00
|
|
|
func NewDroneServer(remote remote.Remote, queue queue.Queue, logger logging.Log, pubsub pubsub.Publisher, store store.Store, host string) *DroneServer {
|
|
|
|
buildTime := promauto.NewGaugeVec(prometheus.GaugeOpts{
|
|
|
|
Namespace: "drone",
|
|
|
|
Name: "build_time",
|
|
|
|
Help: "Build time.",
|
|
|
|
}, []string{"repo", "branch", "status", "pipeline"})
|
|
|
|
buildCount := promauto.NewCounterVec(prometheus.CounterOpts{
|
|
|
|
Namespace: "drone",
|
|
|
|
Name: "build_count",
|
|
|
|
Help: "Build count.",
|
|
|
|
}, []string{"repo", "branch", "status", "pipeline"})
|
2017-06-28 17:21:22 +00:00
|
|
|
peer := RPC{
|
2019-06-28 12:23:52 +00:00
|
|
|
remote: remote,
|
|
|
|
store: store,
|
|
|
|
queue: queue,
|
|
|
|
pubsub: pubsub,
|
|
|
|
logger: logger,
|
|
|
|
host: host,
|
|
|
|
buildTime: buildTime,
|
|
|
|
buildCount: buildCount,
|
|
|
|
}
|
|
|
|
return &DroneServer{peer: peer}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DroneServer) Next(c oldcontext.Context, req *proto.NextRequest) (*proto.NextReply, error) {
|
2017-06-28 17:21:22 +00:00
|
|
|
filter := rpc.Filter{
|
|
|
|
Labels: req.GetFilter().GetLabels(),
|
2017-10-17 15:47:53 +00:00
|
|
|
Expr: req.GetFilter().GetExpr(),
|
2017-06-28 17:21:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
res := new(proto.NextReply)
|
2019-06-28 12:23:52 +00:00
|
|
|
pipeline, err := s.peer.Next(c, filter)
|
2017-06-28 17:21:22 +00:00
|
|
|
if err != nil {
|
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
if pipeline == nil {
|
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
|
|
|
|
res.Pipeline = new(proto.Pipeline)
|
|
|
|
res.Pipeline.Id = pipeline.ID
|
|
|
|
res.Pipeline.Timeout = pipeline.Timeout
|
|
|
|
res.Pipeline.Payload, _ = json.Marshal(pipeline.Config)
|
|
|
|
|
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DroneServer) Init(c oldcontext.Context, req *proto.InitRequest) (*proto.Empty, error) {
|
|
|
|
state := rpc.State{
|
|
|
|
Error: req.GetState().GetError(),
|
|
|
|
ExitCode: int(req.GetState().GetExitCode()),
|
|
|
|
Finished: req.GetState().GetFinished(),
|
|
|
|
Started: req.GetState().GetStarted(),
|
|
|
|
Proc: req.GetState().GetName(),
|
|
|
|
Exited: req.GetState().GetExited(),
|
|
|
|
}
|
|
|
|
res := new(proto.Empty)
|
2019-06-28 12:23:52 +00:00
|
|
|
err := s.peer.Init(c, req.GetId(), state)
|
2017-06-28 17:21:22 +00:00
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DroneServer) Update(c oldcontext.Context, req *proto.UpdateRequest) (*proto.Empty, error) {
|
|
|
|
state := rpc.State{
|
|
|
|
Error: req.GetState().GetError(),
|
|
|
|
ExitCode: int(req.GetState().GetExitCode()),
|
|
|
|
Finished: req.GetState().GetFinished(),
|
|
|
|
Started: req.GetState().GetStarted(),
|
|
|
|
Proc: req.GetState().GetName(),
|
|
|
|
Exited: req.GetState().GetExited(),
|
|
|
|
}
|
|
|
|
res := new(proto.Empty)
|
2019-06-28 12:23:52 +00:00
|
|
|
err := s.peer.Update(c, req.GetId(), state)
|
2017-06-28 17:21:22 +00:00
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DroneServer) Upload(c oldcontext.Context, req *proto.UploadRequest) (*proto.Empty, error) {
|
|
|
|
file := &rpc.File{
|
|
|
|
Data: req.GetFile().GetData(),
|
|
|
|
Mime: req.GetFile().GetMime(),
|
|
|
|
Name: req.GetFile().GetName(),
|
|
|
|
Proc: req.GetFile().GetProc(),
|
|
|
|
Size: int(req.GetFile().GetSize()),
|
|
|
|
Time: req.GetFile().GetTime(),
|
2017-08-02 20:04:00 +00:00
|
|
|
Meta: req.GetFile().GetMeta(),
|
2017-06-28 17:21:22 +00:00
|
|
|
}
|
2017-08-02 20:04:00 +00:00
|
|
|
|
2017-06-28 17:21:22 +00:00
|
|
|
res := new(proto.Empty)
|
2019-06-28 12:23:52 +00:00
|
|
|
err := s.peer.Upload(c, req.GetId(), file)
|
2017-06-28 17:21:22 +00:00
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DroneServer) Done(c oldcontext.Context, req *proto.DoneRequest) (*proto.Empty, error) {
|
|
|
|
state := rpc.State{
|
|
|
|
Error: req.GetState().GetError(),
|
|
|
|
ExitCode: int(req.GetState().GetExitCode()),
|
|
|
|
Finished: req.GetState().GetFinished(),
|
|
|
|
Started: req.GetState().GetStarted(),
|
|
|
|
Proc: req.GetState().GetName(),
|
|
|
|
Exited: req.GetState().GetExited(),
|
|
|
|
}
|
|
|
|
res := new(proto.Empty)
|
2019-06-28 12:23:52 +00:00
|
|
|
err := s.peer.Done(c, req.GetId(), state)
|
2017-06-28 17:21:22 +00:00
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DroneServer) Wait(c oldcontext.Context, req *proto.WaitRequest) (*proto.Empty, error) {
|
|
|
|
res := new(proto.Empty)
|
2019-06-28 12:23:52 +00:00
|
|
|
err := s.peer.Wait(c, req.GetId())
|
2017-06-28 17:21:22 +00:00
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DroneServer) Extend(c oldcontext.Context, req *proto.ExtendRequest) (*proto.Empty, error) {
|
|
|
|
res := new(proto.Empty)
|
2019-06-28 12:23:52 +00:00
|
|
|
err := s.peer.Extend(c, req.GetId())
|
2017-06-28 17:21:22 +00:00
|
|
|
return res, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *DroneServer) Log(c oldcontext.Context, req *proto.LogRequest) (*proto.Empty, error) {
|
|
|
|
line := &rpc.Line{
|
|
|
|
Out: req.GetLine().GetOut(),
|
|
|
|
Pos: int(req.GetLine().GetPos()),
|
|
|
|
Time: req.GetLine().GetTime(),
|
|
|
|
Proc: req.GetLine().GetProc(),
|
|
|
|
}
|
|
|
|
res := new(proto.Empty)
|
2019-06-28 12:23:52 +00:00
|
|
|
err := s.peer.Log(c, req.GetId(), line)
|
2017-06-28 17:21:22 +00:00
|
|
|
return res, err
|
|
|
|
}
|