domain.go

// Copyright (C) 2014-2018 Goodrain Co., Ltd.
// RAINBOND, Application Management Platform

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version. For any non-GPL usage of Rainbond,
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
// must be obtained first.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

package cmd
import (
	"github.com/urfave/cli"
	"fmt"
	"bytes"
	"os/exec"
	"github.com/goodrain/rainbond/grctl/clients"
	"github.com/Sirupsen/logrus"
	"strings"
	"github.com/goodrain/rainbond/node/api/model"
)

func NewCmdDomain() cli.Command {
	c:=cli.Command{
		Name: "domain",
		Flags: []cli.Flag{
			cli.StringFlag{
				Name:  "ip",
				Usage: "ip address",
			},
			cli.StringFlag{
				Name:  "domain",
				Usage: "domain",
			},
		},
		Usage: "",
		Action: func(c *cli.Context) error {
			ip:=c.String("ip")
			if len(ip)==0 {
				fmt.Println("ip must not null")
				return nil
			}
			domain:=c.String("domain")
			cmd := exec.Command("bash", "/opt/rainbond/bin/.domain.sh",ip,domain)
			outbuf:=bytes.NewBuffer(nil)
			cmd.Stdout=outbuf
			cmd.Run()
			out:=outbuf.String()
			fmt.Println(out)
			return nil
		},
	}
	return c
}
func NewCmdCheckTask() cli.Command {
	c:=cli.Command{
		Name: "checkTask",
		Flags: []cli.Flag{
			cli.StringFlag{
				Name:  "uuid",
				Usage: "uuid",
			},
		},
		Usage: "",
		Action: func(c *cli.Context) error {
			uuid:=c.String("uuid")
			if len(uuid)==0 {
				fmt.Println("uuid must not null")
				return nil
			}
			tasks,err:=clients.NodeClient.Tasks().List()
			if err != nil {
				logrus.Errorf("error get task list,details %s",err.Error())
				return err
			}
			var result []*ExecedTask
			for _,v:=range tasks{
				taskStatus,ok:=v.Status[uuid]
				if ok{
					status:=strings.ToLower(taskStatus.Status)
					if status=="complete" ||status=="start"{
						var taskentity =&ExecedTask{}
						taskentity.ID=v.ID
						taskentity.Status=taskStatus.Status
						taskentity.Depends=[]string{}
						dealDepend(taskentity,v)
						dealNext(taskentity,tasks)
						result=append(result, taskentity)
						continue
					}

				}else {
					_,scheduled:=v.Scheduler.Status[uuid]
					if scheduled {
						var taskentity =&ExecedTask{}
						taskentity.ID=v.ID
						taskentity.Depends=[]string{}
						dealDepend(taskentity,v)
						dealNext(taskentity,tasks)
						allDepDone:=true

						for _,dep:=range taskentity.Depends {
							task,_:=clients.NodeClient.Tasks().Get(dep)

							_,depOK:=task.Status[uuid]
							if !depOK {
								allDepDone=false
								break
							}
						}
						if allDepDone {
							taskentity.Status="start"
							result=append(result, taskentity)
						}

					}
				}
			}
			for _,v:=range result {
				fmt.Printf("task %s is %s,depends is %v\n",v.ID,v.Status,v.Depends)
			}
			return nil
		},
	}
	return c
}
func dealDepend(result *ExecedTask,task *model.Task) {
	if task.Temp.Depends != nil {
		for _,v:=range task.Temp.Depends{
			result.Depends=append(result.Depends,v.DependTaskID)
		}
	}
}
func dealNext(task *ExecedTask, tasks []*model.Task) {
	for _,v:=range tasks {
		if v.Temp.Depends != nil {
			for _,dep:=range v.Temp.Depends{
				if dep.DependTaskID == task.ID {
					task.Next=append(task.Next,v.ID)
				}
			}
		}
	}
}
type ExecedTask struct {
	ID string
	Status string
	Depends []string
	Next []string
}