基于Prometheus的集成式运维告警中心设计源码

package controllers // Issue: https://github.com/feiyu563/PrometheusAlert/issues/181 import ( "encoding/json" "strconv" "github.com/astaxie/beego" "github.com/astaxie/beego/logs" ) // GitlabController is a beego controller for gitlab webhook events type GitlabController struct { beego.Controller } // GitlabProject is a project section in gitlab event type GitlabProject struct { Name string `json:"name"` Homepage string `json:"homepage"` } // GitlabRepository is a repository section in gitlab event type GitlabRepository struct { Name string `json:"name"` Homepage string `json:"homepage"` } // GitlabUser is a user section in gitlab event // event: Issue, Note, Merge Request, Wiki, Job, Deployment, Feature Flag type GitlabUser struct { Name string `json:"name"` Username string `json:"username"` Email string `json:"email"` } // GitlabAssignee is a assignee section in gitlab event type GitlabAssignee struct { Username string `json:"username"` } // GitlabCommit is a commit section in gitlab event type GitlabCommit struct { Message string `json:"message"` Title string `json:"title"` } // GitlabWiki is a wiki section in gitlab event type GitlabWiki struct { WebUrl string `json:"web_url"` } // GitlabObjectAttributes is a object_attribute section in gitlab event // event: Issue, Merge Request, Pipeline, Comment, Wiki type GitlabObjectAttributes struct { // Issue, Merge Request, Wiki Action string `json:"action"` Title string `json:"title"` // Issue, Merge Request State string `json:"state"` Description string `json:"description"` // Issue, Comment, Wiki Url string `json:"url"` // Merge Request TargetBranch string `json:"target_branch"` SourceBranch string `json:"source_branch"` MergeStatus string `json:"merge_status"` // Pipeline Ref string `json:"ref"` Sha string `json:"sha"` Source string `json:"source"` Status string `json:"status"` // Comment Note string `json:"note"` NoteableType string `json:"noteable_type"` // Wiki Message string `json:"message"` // Feature Flag Name string `json:"name"` Active bool `json:"active"` } // GitlabRunner is a runner section in gitlab event type GitlabRunner struct { Active bool `json:"active"` RunnerType string `json:"runner_type"` IsShared bool `json:"is_shared"` Description string `json:"description"` } // GitlabEvent contains all of gitalb events type GitlabEvent struct { ObjectKind string `json:"object_kind"` Ref string `json:"ref"` CheckoutSha string `json:"checkout_sha"` Message string `json:"message"` Username string `json:"user_name"` UserUsername string `json:"user_username"` Sha string `json:"sha"` BuildName string `json:"build_name"` BuildStage string `json:"build_stage"` BuildStatus string `json:"build_status"` Status string `json:"status"` DeploymentUrl string `json:"deployable_url"` Environment string `json:"environment"` Name string `json:"name"` Url string `json:"url"` Description string `json:"description"` Tag string `json:"tag"` Action string `json:"action"` Commits []GitlabCommit `json:"commits"` User GitlabUser `json:"user"` Project GitlabProject `json:"project"` Repository GitlabRepository `json:"repository"` ObjectAttributes GitlabObjectAttributes `json:"object_attributes"` Assignees []GitlabAssignee `json:"assignees"` Wiki GitlabWiki `json:"wiki"` Runner GitlabRunner `json:"runner"` } // GitlabWeixin sends gitlab webhook events to wx robot func (c *GitlabController) GitlabWeixin() { event := GitlabEvent{} eventType := c.Ctx.Request.Header.Get("X-Gitlab-Event") wxurl := c.GetString("wxurl") logsign := "[" + LogsSign() + "]" logs.Info(logsign, string(c.Ctx.Input.RequestBody)) json.Unmarshal(c.Ctx.Input.RequestBody, &event) c.Data["json"] = sendGitlabEvent(3, event, eventType, logsign, wxurl) logs.Info(logsign, c.Data["json"]) c.ServeJSON() } // GitlabDingding sends gitlab webhook events to dingtalk robot func (c *GitlabController) GitlabDingding() { event := GitlabEvent{} eventType := c.Ctx.Request.Header.Get("X-Gitlab-Event") ddurl := c.GetString("ddurl") logsign := "[" + LogsSign() + "]" logs.Info(logsign, string(c.Ctx.Input.RequestBody)) json.Unmarshal(c.Ctx.Input.RequestBody, &event) c.Data["json"] = sendGitlabEvent(2, event, eventType, logsign, ddurl) logs.Info(logsign, c.Data["json"]) c.ServeJSON() } // GitlabFeishu sends gitlab webhook events to feishu v2 robot func (c *GitlabController) GitlabFeishu() { event := GitlabEvent{} eventType := c.Ctx.Request.Header.Get("X-Gitlab-Event") fsurl := c.GetString("fsurl") logsign := "[" + LogsSign() + "]" logs.Info(logsign, string(c.Ctx.Input.RequestBody)) json.Unmarshal(c.Ctx.Input.RequestBody, &event) c.Data["json"] = sendGitlabEvent(4, event, eventType, logsign, fsurl) logs.Info(logsign, c.Data["json"]) c.ServeJSON() } func genWXtext(event GitlabEvent, eventType string) string { var WXtext, WXbasetext, WXothertext string // 有些payload中不同时包含project和repository信息，因此需要判断下 var name, homepage string if event.Project.Name != "" { name = event.Project.Name homepage = event.Project.Homepage } else { name = event.Repository.Name homepage = event.Repository.Homepage } WXbasetext = "[Gitlab事件通知](" + homepage + ")\n" + "> `事件类型`: " + eventType + "\n" + "> `仓库链接`: [" + name + "](" + homepage + ")\n" switch eventType { case "Push Hook": WXothertext = "> `提交用户`: " + event.Username + "(@" + event.UserUsername + ")\n" + "> `当前Ref`: " + event.Ref + "\n" + "> `当前提交`: " + event.CheckoutSha + "\n" if len(event.Commits) != 0 { WXothertext = WXothertext + "> `提交信息`: \n" + "\n" + event.Commits[len(event.Commits)-1].Message } case "Tag Push Hook": WXothertext = "> `提交用户`: " + event.Username + "(@" + event.UserUsername + ")\n" + "> `当前Ref`: " + event.Ref + "\n" + "> `当前提交`: " + event.CheckoutSha + "\n" if len(event.Commits) != 0 { WXothertext = WXothertext + "> `提交信息`: \n" + "\n" + event.Commits[0].Message } case "Merge Request Hook": WXothertext = "> `提交用户`: " + event.User.Name + "(@" + event.User.Username + ")\n" + "> `源分支`: " + event.ObjectAttributes.SourceBranch + "\n" + "> `目标分支`: " + event.ObjectAttributes.TargetBranch + "\n" + "> `合并请求链接`: [" + event.ObjectAttributes.Title + "](" + event.ObjectAttributes.Url + ")\n" + "> `合并请求状态`: " + event.ObjectAttributes.Action + "\n" if len(event.Assignees) != 0 { WXothertext = WXothertext + "> `分配给`: @" + event.Assignees[0].Username + "\n" } // 描述内容可能有多行，放最后 WXothertext = WXothertext + "> `合并请求描述`:\n" + "\n" + event.ObjectAttributes.Description case "Issue Hook": WXothertext = "> `提交用户`: " + event.User.Name + "(@" + event.User.Username + ")\n" + "> `议题链接`: [" + event.ObjectAttributes.Title + "](" + event.ObjectAttributes.Url + ")\n" + "> `议题状态`: " + event.ObjectAttributes.Action + "\n" if len(event.Assignees) != 0 { WXothertext = WXothertext + "> `分配给`: @" + event.Assignees[0].Username + "\n" } WXothertext = WXothertext + "> `议题描述`:\n" + "\n" + event.ObjectAttributes.Description case "Pipeline Hook": WXothertext = "> `提交用户`: " + event.User.Name + "(@" + event.User.Us