// 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 . package exector import ( "fmt" "time" "github.com/goodrain/rainbond/builder" "github.com/Sirupsen/logrus" "github.com/goodrain/rainbond/event" "github.com/tidwall/gjson" //"github.com/docker/docker/api/types" //"github.com/docker/docker/client" "github.com/docker/docker/client" "github.com/goodrain/rainbond/builder/sources" "github.com/goodrain/rainbond/db" ) //ImageBuildItem ImageBuildItem type ImageBuildItem struct { Namespace string `json:"namespace"` TenantName string `json:"tenant_name"` ServiceAlias string `json:"service_alias"` Image string `json:"image"` DestImage string `json:"dest_image"` Logger event.Logger `json:"logger"` EventID string `json:"event_id"` DockerClient *client.Client TenantID string ServiceID string DeployVersion string HubUser string HubPassword string Action string } //NewImageBuildItem 创建实体 func NewImageBuildItem(in []byte) *ImageBuildItem { eventID := gjson.GetBytes(in, "event_id").String() logger := event.GetManager().GetLogger(eventID) return &ImageBuildItem{ Namespace: gjson.GetBytes(in, "namespace").String(), TenantName: gjson.GetBytes(in, "tenant_name").String(), ServiceAlias: gjson.GetBytes(in, "service_alias").String(), ServiceID: gjson.GetBytes(in, "service_id").String(), Image: gjson.GetBytes(in, "image").String(), DeployVersion: gjson.GetBytes(in, "deploy_version").String(), Action: gjson.GetBytes(in, "action").String(), HubUser: gjson.GetBytes(in, "user").String(), HubPassword: gjson.GetBytes(in, "password").String(), Logger: logger, EventID: eventID, } } //Run Run func (i *ImageBuildItem) Run(timeout time.Duration) error { _, err := sources.ImagePull(i.DockerClient, i.Image, i.HubUser, i.HubPassword, i.Logger, 10) if err != nil { logrus.Errorf("pull image %s error: %s", i.Image, err.Error()) i.Logger.Error(fmt.Sprintf("获取指定镜像: %s失败", i.Image), map[string]string{"step": "builder-exector", "status": "failure"}) return err } localImageURL := i.ImageNameHandler(i.Image) if err := sources.ImageTag(i.DockerClient, i.Image, localImageURL, i.Logger, 1); err != nil { logrus.Errorf("change image tag error: %s", err.Error()) i.Logger.Error(fmt.Sprintf("修改镜像tag: %s -> %s 失败", i.Image, localImageURL), map[string]string{"step": "builder-exector", "status": "failure"}) return err } err = sources.ImagePush(i.DockerClient, localImageURL, builder.REGISTRYUSER, builder.REGISTRYPASS, i.Logger, 30) if err != nil { logrus.Errorf("push image into registry error: %s", err.Error()) i.Logger.Error("推送镜像至镜像仓库失败", map[string]string{"step": "builder-exector", "status": "failure"}) return err } if err := i.StorageVersionInfo(localImageURL); err != nil { logrus.Errorf("storage version info error, ignor it: %s", err.Error()) i.Logger.Error("更新应用版本信息失败", map[string]string{"step": "builder-exector", "status": "failure"}) return err } return nil } //ImageNameHandler 根据平台配置处理镜像名称 func (i *ImageBuildItem) ImageNameHandler(source string) string { imageModel := sources.ImageNameHandle(source) localImageURL := fmt.Sprintf("%s/%s:%s", builder.REGISTRYDOMAIN, imageModel.Name, i.DeployVersion) return localImageURL } //StorageVersionInfo 存储version信息 func (i *ImageBuildItem) StorageVersionInfo(imageURL string) error { version, err := db.GetManager().VersionInfoDao().GetVersionByDeployVersion(i.DeployVersion, i.ServiceID) if err != nil { return err } version.DeliveredType = "image" version.DeliveredPath = imageURL version.ImageName = imageURL version.RepoURL = i.Image version.FinalStatus = "success" if err := db.GetManager().VersionInfoDao().UpdateModel(version); err != nil { return err } return nil } //UpdateVersionInfo 更新任务执行结果 func (i *ImageBuildItem) UpdateVersionInfo(status string) error { version, err := db.GetManager().VersionInfoDao().GetVersionByEventID(i.EventID) if err != nil { return err } version.FinalStatus = status version.RepoURL = i.Image if err := db.GetManager().VersionInfoDao().UpdateModel(version); err != nil { return err } return nil }