func (this *gowatch) drainExec() {
log.Println("command:", this.cmd)
var msg string
for {
startTime := time.Now()
cmd := this.cmd
if len(cmd) == 0 {
cmd = []string{"echo", "no command specified"}
}
cprintf(CGREEN, "exec start")
c := StartCmd(cmd[0], cmd[1:]...)
// Start to run command
err := c.Start()
if err != nil {
cprintf("35", err.Error())
}
// Wait until killed or finished
select {
case msg = <-this.sig:
cprintf(CYELLOW, "program terminated, signal(%s)", this.KillSignal)
if err := KillCmd(c, this.KillSignal); err != nil {
log.Errorf("group kill: %v", err)
}
if msg == "EXIT" {
os.Exit(1)
}
goto SKIP_WAITING
case err = <-Go(c.Wait):
if err != nil {
cprintf(CPURPLE, "program exited: %v", err)
}
}
log.Infof("finish in %s", time.Since(startTime))
// Whether to restart right now
if this.AutoRestart {
goto SKIP_WAITING
}
cprintf("33", "-- wait signal --")
if msg = <-this.sig; msg == "EXIT" {
os.Exit(1)
}
SKIP_WAITING:
if this.RestartInterval > 0 {
log.Infof("restart after %s", this.RestartInterval)
}
time.Sleep(this.RestartInterval)
}
}
func (r *Rpc) UpdatePubAddr(args *PublishInfo, reply *bool) error {
log.Infof("pub addr %v", *args)
*reply = true
err := models.UpdatePubAddr(args.Mid, args.ZipBallUrl)
return err
}
// generate qiniu token
func qntoken(key string) string {
scope := defaultBulket + ":" + key
log.Infof("qiniu scrope: %s", scope)
policy := rs.PutPolicy{
Expires: uint32(time.Now().Unix() + 3600),
Scope: scope,
}
return policy.Token(nil)
}
// keep task status always avaliable
func drainTask() {
for {
log.Infof("drain task start after 25min")
time.Sleep(25 * time.Minute)
if repos, err := models.GetAllRepos(1000, 0); err == nil {
for _, r := range repos {
routers.TriggerBuildRepositoryById(r.Id)
}
}
}
}
func CreateNewBuilding(rid int64, branch string, os, arch string, action string) (err error) {
repo, err := GetRepositoryById(rid)
if err != nil {
return
}
task := &Task{
Rid: rid,
Action: action,
TagBranch: branch,
PushType: "branch",
PushValue: branch,
Os: os,
Arch: arch,
CgoEnable: repo.IsCgo,
}
var cvsinfo *base.CVSInfo
log.Infof("add task for repo: %v", repo.Uri)
if cvsinfo, err = base.ParseCvsURI(repo.Uri); err != nil {
return
}
if cvsinfo.Provider == base.PVD_GITHUB {
info, _, er := GHClient.Repositories.GetBranch(cvsinfo.Owner, cvsinfo.RepoName, branch)
if er != nil {
err = er
return
}
log.Infof("get information from github:%v", info)
task.PushType = "commit"
task.PushValue = *info.Commit.SHA
if info.Commit.Message != nil {
task.CommitMessage = *info.Commit.Message
}
}
_, err = CreateTask(task)
return
}
func Action(c *cli.Context) {
var goos, goarch = c.String("os"), c.String("arch")
//var depth = c.Int("depth")
var output = c.String("output")
var gom = c.String("gom")
var nobuild = c.Bool("nobuild")
var adds = c.StringSlice("add")
var rmflag = c.Bool("rm")
if c.Bool("debug") {
log.SetOutputLevel(log.Ldebug)
}
if c.Bool("init") {
data, _ := goyaml.Marshal(DefaultPcfg)
fmt.Print(string(data))
return
}
log.Debugf("os: %s, arch: %s", goos, goarch)
var err error
defer func() {
if err != nil {
log.Fatal(err)
}
}()
sess := sh.NewSession()
sess.SetEnv("GOOS", goos)
sess.SetEnv("GOARCH", goarch)
sess.ShowCMD = true
gomarr := strings.Fields(gom)
if len(gomarr) >= 1 {
sess.Alias("go", gomarr[0], gomarr[1:]...)
}
// parse yaml
pcfg, err := ReadPkgConfig(RCFILE)
if err != nil {
return
}
pwd, _ := os.Getwd()
gobin := filepath.Join(pwd, sanitizedName(pcfg.Settings.TargetDir))
sess.SetEnv("GOBIN", gobin)
log.Debugf("set env GOBIN=%s", gobin)
log.Debug("config:", pcfg)
// pcfg.Filesets.Includes = append(pcfg.Filesets.Includes, adds...)
var skips []*regexp.Regexp
for _, str := range pcfg.Filesets.Excludes {
skips = append(skips, regexp.MustCompile("^"+str+"$"))
}
var needs []*regexp.Regexp
for _, str := range pcfg.Filesets.Includes {
needs = append(needs, regexp.MustCompile("^"+str+"$"))
}
//log.Infof("archive file to: %s", output)
var z Archiver
hasExt := func(ext string) bool { return strings.HasSuffix(output, ext) }
switch {
case hasExt(".zip"):
fmt.Println("zip format")
z, err = CreateZip(output)
case hasExt(".tar"):
fmt.Println("tar format")
z, err = CreateTar(output)
case hasExt(".tgz"):
fallthrough
case hasExt(".tar.gz"):
fmt.Println("tar.gz format")
z, err = CreateTgz(output)
default:
fmt.Println("unsupport file archive format")
os.Exit(1)
}
if err != nil {
return
}
var files = []string{}
var buildFiles = pcfg.Settings.Outfiles
// build source
if !nobuild {
targetDir := sanitizedName(pcfg.Settings.TargetDir)
if !sh.Test("dir", targetDir) {
os.MkdirAll(targetDir, 0755)
}
symdir := filepath.Join(targetDir, goos+"_"+goarch)
if err = os.Symlink(gobin, symdir); err != nil {
log.Fatalf("os symlink error: %s", err)
}
defer os.Remove(symdir)
// opts := []string{"install", "-v"}
// opts = append(opts, strings.Fields(pcfg.Settings.Addopts)...) // TODO: here need to use shell args parse lib
if pcfg.Settings.Build == "" {
pcfg.Settings.Build = DEFAULT_BUILD
}
if err = sess.Command("bash", "-c", pcfg.Settings.Build).Run(); err != nil {
// if err = sess.Command("go", opts).Run(); err != nil {
return
}
os.Remove(symdir) // I have to do it twice
if len(buildFiles) == 0 {
cwd, _ := os.Getwd()
program := filepath.Base(cwd)
buildFiles = append(buildFiles, program)
}
for _, filename := range buildFiles {
if goos == "windows" {
filename += ".exe"
}
files = append(files, filename)
}
}
log.Debug("archive files")
depth := pcfg.Filesets.Depth
for _, filename := range pcfg.Filesets.Includes {
fs, err := findFiles(filename, depth, skips)
if err != nil {
return
}
files = append(files, fs...)
}
// adds - parse by cli
files = append(files, adds...)
uniqset := make(map[string]bool, len(files))
for _, file := range files {
file = sanitizedName(file)
if uniqset[file] {
continue
}
uniqset[file] = true
log.Infof("zip add file: %v", file)
if err = z.Add(file); err != nil {
return
}
}
log.Info("finish archive file")
err = z.Close()
if rmflag {
for _, filename := range buildFiles {
if goos == "windows" {
filename += ".exe"
}
os.Remove(filename)
}
}
}
func Action(c *cli.Context) {
var goos, goarch = c.String("os"), c.String("arch")
var depth = c.Int("depth")
var output = c.String("output")
var gom = c.String("gom")
var nobuild = c.Bool("nobuild")
var adds = c.StringSlice("add")
if goos == "" {
goos = runtime.GOOS
}
if goarch == "" {
goarch = runtime.GOARCH
}
var err error
defer func() {
if err != nil {
log.Fatal(err)
}
}()
sess := sh.NewSession()
sess.SetEnv("GOOS", goos)
sess.SetEnv("GOARCH", goarch)
sess.ShowCMD = true
gomarr := strings.Fields(gom)
if len(gomarr) >= 1 {
sess.Alias("go", gomarr[0], gomarr[1:]...)
}
// parse yaml
pcfg, err := config.ReadPkgConfig(config.RCFILE)
if err != nil {
return
}
// var pcfg = new(config.PackageConfig)
// if sh.Test("file", config.RCFILE) {
// data, er := ioutil.ReadFile(config.RCFILE)
// if er != nil {
// err = er
// return
// }
// if err = goyaml.Unmarshal(data, pcfg); err != nil {
// return
// }
// } else {
// pcfg = config.DefaultPcfg
// }
pwd, _ := os.Getwd()
gobin := filepath.Join(pwd, sanitizedName(pcfg.Settings.TargetDir))
sess.SetEnv("GOBIN", gobin)
log.Debugf("set env GOBIN=%s", gobin)
log.Debug("config:", pcfg)
pcfg.Filesets.Includes = append(pcfg.Filesets.Includes, adds...)
var skips []*regexp.Regexp
for _, str := range pcfg.Filesets.Excludes {
skips = append(skips, regexp.MustCompile("^"+str+"$"))
}
log.Infof("archive file to: %s", output)
var z Archiver
hasExt := func(ext string) bool { return strings.HasSuffix(output, ext) }
switch {
case hasExt(".zip"):
fmt.Println("zip format")
z, err = CreateZip(output)
case hasExt(".tar"):
fmt.Println("tar format")
z, err = CreateTar(output)
case hasExt(".tgz"):
fallthrough
case hasExt(".tar.gz"):
fmt.Println("tar.gz format")
z, err = CreateTgz(output)
default:
fmt.Println("unsupport file archive format")
os.Exit(1)
}
if err != nil {
return
}
var files []string
// build source
if !nobuild {
targetDir := sanitizedName(pcfg.Settings.TargetDir)
if !sh.Test("dir", targetDir) {
os.MkdirAll(targetDir, 0755)
}
symdir := filepath.Join(targetDir, goos+"_"+goarch)
if err = os.Symlink(gobin, symdir); err != nil {
log.Fatalf("os symlink error: %s", err)
}
defer os.Remove(symdir)
opts := []string{"install", "-v"}
opts = append(opts, strings.Fields(pcfg.Settings.Addopts)...) // TODO: here need to use shell args parse lib
if err = sess.Command("go", opts).Run(); err != nil {
return
}
os.Remove(symdir) // I have to do it twice
cwd, _ := os.Getwd()
program := filepath.Base(cwd)
if goos == "windows" {
program += ".exe"
}
if sh.Test("file", program) {
files = append(files, program)
}
}
log.Debug("archive files")
for _, filename := range pcfg.Filesets.Includes {
fs, err := findFiles(filename, depth, skips)
if err != nil {
return
}
files = append(files, fs...)
}
for _, file := range files {
log.Infof("zip add file: %v", file)
if err = z.Add(file); err != nil {
return
}
}
log.Info("finish archive file")
err = z.Close()
}
