cmdb-k8s

Author: CosmicBDry

README

@@ -1 +1 @@
-云管系统，对接腾讯云、阿里云
+cmdb资产管理系统，对接kubernetes集群、腾讯云、阿里云、gitlab、prometheus

server/buildservice/build.go

@@ -0,0 +1,91 @@
+package buildservice
+
+import (
+	"fmt"
+	"path/filepath"
+	"sync"
+	"time"
+
+	"github.com/CosmicBDry/gocmdb/server/models"
+	"github.com/CosmicBDry/gocmdb/server/rwFile"
+	"github.com/CosmicBDry/gocmdb/server/systemCommand"
+
+	"github.com/astaxie/beego"
+)
+
+type BuildService struct {
+	Mutex     sync.RWMutex   //读写锁
+	ProjectId map[int64]bool //判断项目是否再构建中
+}
+
+func NewBuildService() *BuildService {
+
+	return &BuildService{
+		Mutex:     sync.RWMutex{},
+		ProjectId: make(map[int64]bool),
+	}
+}
+
+var DefaultBuildService = NewBuildService()
+
+func (b *BuildService) Build(gitproject *models.GitProject) error {
+	b.Mutex.RLock() //添加一个读锁,每次只允许一个进程读取b.ProjectId映射
+	if _, ok := b.ProjectId[gitproject.ProjectId]; ok {
+		b.Mutex.RUnlock() //读完之后将锁释放，使得其他进程可读
+		beego.Warning(gitproject.SShUrl + ": 项目正在构建中...,请不要重复构建!")
+		return fmt.Errorf("Is Building")
+	}
+	b.Mutex.RUnlock() ////读完之后将锁释放
+	b.ProjectId[gitproject.ProjectId] = true
+
+	//构建脚本：------------------------------------------------------------------------------->
+	WorkDir := beego.AppConfig.DefaultString("builder::BuildDir", "/tmp/")
+	ProjectDir := fmt.Sprintf("%sProject_%s/", WorkDir, gitproject.Name)
+	PackageDir := beego.AppConfig.DefaultString("builder::PackageDir", "/tmp/")
+	Now := time.Now().Format("2006-01-02_15-04-05")
+	PackageCmd := fmt.Sprintf("cd %s%s/%s/target && tar czf %s%s/%s.tar.gz ./* && wait && sleep 5\n", ProjectDir, gitproject.Branch, gitproject.Name, PackageDir, gitproject.Name, Now)
+	if gitproject.PackageFile != "" {
+		PackageCmd = fmt.Sprintf("cd %s%s/%s && cp -a {%s,} ./target/ && wait && sleep 5 &&cd ./target && tar czf %s%s/%s.tar.gz ./* && wait && sleep 5\n", ProjectDir, gitproject.Branch, gitproject.Name, gitproject.PackageFile, PackageDir, gitproject.Name, Now)
+		fmt.Println("----->: ", PackageCmd)
+	}
+	cmd := []string{
+		fmt.Sprintf("rm -rf %s && wait && mkdir -p %s%s && echo `%s` >> %s%s/build.log && wait\n", ProjectDir, ProjectDir, gitproject.Branch, "date +%Y-%m-%d_%H:%M:%S", ProjectDir, gitproject.Branch),
+		fmt.Sprintf("mkdir -p %s%s\n", PackageDir, gitproject.Name),
+		fmt.Sprintf("cd %s%s &&git clone -b %s %s %s && wait && sleep 5\n", ProjectDir, gitproject.Branch, gitproject.Branch, gitproject.SShUrl, gitproject.Name),
+		fmt.Sprintf("cd %s%s/%s && go build -o ./target/%s && wait && sleep 5\n", ProjectDir, gitproject.Branch, gitproject.Name, gitproject.Name),
+		PackageCmd,
+	}
+
+	for _, v := range cmd {
+		var result string
+		var err error
+		path := fmt.Sprintf("%s%s/build.log", ProjectDir, gitproject.Branch)
+		if result, err = systemCommand.RunCmd(v); err != nil {
+			beego.Error(gitproject.SShUrl + ": 项目代码构建失败,详情请查看build.log构建记录")
+			if errs := rwFile.WriteFile(result+err.Error()+"\n", path); errs != nil {
+				beego.Error(path + " :代码构建记录写入失败," + errs.Error())
+			}
+			delete(b.ProjectId, gitproject.ProjectId)
+			return err
+		}
+
+		if result == "" {
+			result = "执行成功: " + v
+		}
+
+		if err = rwFile.WriteFile(result, path); err != nil {
+			beego.Error(path + " :代码构建记录写入失败," + err.Error())
+			continue
+		}
+	}
+	beego.Informational(gitproject.SShUrl + ": 项目构建成功!")
+	delete(b.ProjectId, gitproject.ProjectId)
+	return nil
+}
+
+func GetBuildLog(gitproject *models.GitProject) string {
+	WorkDir := beego.AppConfig.DefaultString("builder::BuildDir", "/tmp/")
+	ProjectDir := fmt.Sprintf("%sProject_%s/", WorkDir, gitproject.Name)
+	Path := filepath.Join(ProjectDir, gitproject.Branch, "/", "build.log")
+	return rwFile.ReadFile(Path)
+}

server/cloudSync.go

@@ -0,0 +1,11 @@
+package main
+
+import (
+	"github.com/CosmicBDry/gocmdb/server/cobraCmd"
+)
+
+func main() {
+
+	cobraCmd.RootCmd.Execute()
+
+}

server/cmdb.go

@@ -0,0 +1,70 @@
+package cobraCmd
+
+import (
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/CosmicBDry/gocmdb/server/cloud"
+	"github.com/CosmicBDry/gocmdb/server/models"
+	"github.com/astaxie/beego"
+	"github.com/astaxie/beego/orm"
+	"github.com/spf13/cobra"
+)
+
+var (
+	cloudsync bool
+)
+
+var SyncCmd = &cobra.Command{
+	Use:   "cloudsync",
+	Short: "lauch a cloudsync process ",
+	Long:  "lauch a cloudsync process ",
+	RunE: func(cmd *cobra.Command, args []string) error {
+
+		orm.RegisterDriver("mysql", orm.DRMySQL)
+		orm.RegisterDataBase("default", "mysql", beego.AppConfig.String("dsn"))
+
+		if db, err := orm.GetDB(); err != nil || db.Ping() != nil {
+			beego.Error("数据库连接失败")
+			os.Exit(-1)
+		}
+
+		beego.SetLogger("file", `{"level":7,"filename":"logs/cloudsync.log","maxdays":15,"maxlines":1000}`)
+
+		//通过遍历time.Tick(n * time.second)函数实现每间隔n秒后执行以下for循环中的操作
+		for now := range time.Tick(5 * time.Second) {
+			fmt.Println(now)
+			_, _, Platforms := models.DefautlCloudPlatformManager.QueryList(0, 0, 0, "", "", "", "")
+			for _, platform := range Platforms {
+				if platform.Status == models.StatusLock {
+					continue
+				}
+				//fmt.Println(platform)
+				sdk := cloud.DefaultCloudManager.Plugins[platform.Types]
+				sdk.Init(platform.Addr, platform.Region, platform.AccessKey, platform.SecretKey)
+				instances := sdk.GetInstances()
+				fmt.Printf("%#v\n", instances)
+				if len(instances) == 0 {
+
+					models.DefaultVirtualMachineManager.SyncVmStatus(&platform, now) //同步虚拟机状态
+					models.DefautlCloudPlatformManager.CloudSyncTime(&platform, now) //云平台最后一次同步时间
+				}
+				for _, instance := range instances {
+					if err := models.DefaultVirtualMachineManager.SyncVmByCloudPlatform(&platform, instance); err == nil {
+						models.DefaultVirtualMachineManager.SyncVmStatus(&platform, now) //同步虚拟机状态
+						models.DefautlCloudPlatformManager.CloudSyncTime(&platform, now) //云平台最后一次同步时间
+					}
+				}
+			}
+		}
+		return nil
+	},
+}
+
+func init() {
+	RootCmd.AddCommand(SyncCmd)
+
+	SyncCmd.Flags().BoolVarP(&cloudsync, "cloudsync", "s", false, "sync cloud machine")
+
+}

server/cobraCmd/cloudSync.go

@@ -0,0 +1,18 @@
+package cobraCmd
+
+//子命令的初始化包都可以放到此处，如下
+import (
+	_ "github.com/CosmicBDry/gocmdb/server/cloud/plugins"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/spf13/cobra"
+)
+
+var RootCmd = &cobra.Command{
+	Use:   "cmdb [web]",
+	Short: "Asset management system[CMDB]",
+	Long:  "Asset management system[CMDB]",
+}
+
+func Execute() {
+	RootCmd.Execute()
+}

server/cobraCmd/root.go

@@ -0,0 +1,88 @@
+package cobraCmd
+
+import (
+	"os"
+	"time"
+
+	"github.com/CosmicBDry/gocmdb/server/models"
+	"github.com/CosmicBDry/gocmdb/server/routers"
+	"github.com/CosmicBDry/gocmdb/server/utils"
+	"github.com/astaxie/beego"
+	"github.com/astaxie/beego/orm"
+
+	"github.com/spf13/cobra"
+)
+
+var (
+	initial, syncdb, force, verbose, resetPassword bool
+)
+
+var WebCmd = &cobra.Command{
+	Use:   "web",
+	Short: "lauch web program",
+	Long:  "lauch web program",
+	RunE: func(cmd *cobra.Command, args []string) error {
+
+		orm.RegisterDriver("mysql", orm.DRMySQL)
+		orm.RegisterDataBase("default", "mysql", beego.AppConfig.String("dsn"))
+
+		if db, err := orm.GetDB(); err != nil || db.Ping() != nil {
+			beego.Error("数据库连接失败")
+			os.Exit(-1)
+		}
+		routers.Register()
+		beego.SetLogger("file", `{"level":7,"filename":"logs/web.log","maxdays":15,"maxlines":10000}`)
+		switch {
+		case initial:
+			orm.RunSyncdb("default", force, verbose)
+			Birth := time.Now()
+			admin := &models.User{Name: "admin", IsSuperman: true, Birthday: &Birth}
+			ormer := orm.NewOrm()
+			if err := ormer.Read(admin, "Name"); err == orm.ErrNoRows {
+				Password := utils.RandString(20)
+				admin.SetPassword(Password)
+				if _, err := ormer.Insert(admin); err == nil {
+					beego.Informational("Database table and user 'admin' have been initialized , admin password is: ", Password)
+				} else {
+
+					beego.Informational("admin initialization failed!")
+
+				}
+
+			} else {
+
+				beego.Informational("Database table has been initialized,but admin is already exist,admin does not need to be initialized!")
+			}
+		case syncdb:
+			orm.RunSyncdb("default", force, verbose)
+			beego.Informational("sync table success!")
+		case resetPassword:
+			admin := &models.User{Id: 1}
+			password := utils.RandString(20)
+			admin.SetPassword(password)
+			ormer := orm.NewOrm()
+			if rows, err := ormer.Update(admin, "Password"); rows != 0 && err == nil {
+				beego.Informational("User 'admin' reset success,admin password is : ", password)
+			} else {
+				beego.Error("User 'admin' Password reset failed,Ensure that the user 'admin' exists or please check the database connection!")
+
+			}
+
+		default:
+			beego.Run()
+		}
+
+		return nil
+	},
+}
+
+func init() {
+
+	RootCmd.AddCommand(WebCmd)
+	WebCmd.Flags().BoolVarP(&initial, "init", "I", false, "sync table,according to your needs,Adding '-force' option to reset table and admin password")
+	WebCmd.Flags().BoolVarP(&syncdb, "sync", "S", false, "sync table")
+	WebCmd.Flags().BoolVarP(&force, "force", "F", false, "force sync db(drop table)")
+	WebCmd.Flags().BoolVarP(&verbose, "verbose", "V", false, "display detailed information of sql")
+	WebCmd.Flags().BoolVarP(&resetPassword, "resetPassword", "R", false, "only reset admin password!")
+
+}