diff --git a/README.md b/README.md index a67139bf..4f3602cf 100644 --- a/README.md +++ b/README.md @@ -94,10 +94,9 @@ jobs.pool = 1000 # 站点名称 site.name = 定时任务管理器 -#通知方式 0=邮件,1=信息 +#通知方式 0=邮件,1=信息,2=钉钉,3=微信 notify.type = 0 - # 数据库配置 db.host = 127.0.0.1 db.user = root @@ -115,10 +114,17 @@ email.user = ci@xxx.cn email.password = "xxxxxx" email.pool = 10 - # 短信通知方式配置 msg.url = http://chanxiyou.com/api/tools/send_sms msg.pool = 10 + +# 钉钉通知配置 +dingtalk.url = "https://oapi.dingtalk.com/robot/send?access_token=%s" +dingtalk.pool = 10 + +# 微信通知方式配置 +wechat.url = http://xx.com/api/tools/send_wechat +wechat.pool = 10 ``` 编译安装-可能会遇到的问题 @@ -128,6 +134,8 @@ jobs/job.go:19:2: cannot find package "golang.org/x/crypto/ssh" in any of: 需要 git clone https://github.com/golang/crypto.git 并拷贝到 $GOPATH/src/golang.org/x/ 下就OK +或 +git clone https://github.com/golang/crypto.git $GOPATH/src/golang.org/x/crypto # Docker 本地编译好的2进制文件放在根目录下执行下面的命令即可拥有 diff --git a/conf/app.conf b/conf/app.conf index bf71c81a..e5b3d434 100644 --- a/conf/app.conf +++ b/conf/app.conf @@ -11,10 +11,9 @@ jobs.pool = 1000 # 站点名称 site.name = 定时任务管理器 -#通知方式 0=邮件,1=信息,2=钉钉 +#通知方式 0=邮件,1=信息,2=钉钉,3=微信 notify.type = 0 - # 数据库配置 db.host = 127.0.0.1 db.user = root @@ -32,12 +31,14 @@ email.user = ci@xxx.cn email.password = "xxxxxx" email.pool = 10 - # 短信通知方式配置 msg.url = http://xx.com/api/tools/send_sms msg.pool = 10 - # 钉钉通知配置 dingtalk.url = "https://oapi.dingtalk.com/robot/send?access_token=%s" dingtalk.pool = 10 + +# 微信通知方式配置 +wechat.url = http://xx.com/api/tools/send_wechat +wechat.pool = 10 \ No newline at end of file diff --git a/controllers/admin.go b/controllers/admin.go index 968ccf7c..e502aa7a 100644 --- a/controllers/admin.go +++ b/controllers/admin.go @@ -59,6 +59,7 @@ func (self *AdminController) Edit() { row["phone"] = Admin.Phone row["email"] = Admin.Email row["dingtalk"] = Admin.Dingtalk + row["wechat"] = Admin.Wechat row["role_ids"] = Admin.RoleIds self.Data["admin"] = row @@ -95,6 +96,7 @@ func (self *AdminController) AjaxSave() { Admin.Phone = strings.TrimSpace(self.GetString("phone")) Admin.Email = strings.TrimSpace(self.GetString("email")) Admin.Dingtalk = strings.TrimSpace(self.GetString("dingtalk")) + Admin.Wechat = strings.TrimSpace(self.GetString("wechat")) Admin.RoleIds = strings.TrimSpace(self.GetString("roleids")) Admin.UpdateTime = time.Now().Unix() Admin.UpdateId = self.userId @@ -128,6 +130,7 @@ func (self *AdminController) AjaxSave() { Admin.Phone = strings.TrimSpace(self.GetString("phone")) Admin.Email = strings.TrimSpace(self.GetString("email")) Admin.Dingtalk = strings.TrimSpace(self.GetString("dingtalk")) + Admin.Wechat = strings.TrimSpace(self.GetString("wechat")) Admin.RoleIds = strings.TrimSpace(self.GetString("roleids")) Admin.UpdateTime = time.Now().Unix() Admin.UpdateId = self.userId @@ -207,6 +210,7 @@ func (self *AdminController) Table() { row["phone"] = v.Phone row["email"] = v.Email row["dingtalk"] = v.Dingtalk + row["wechat"] = v.Wechat row["role_ids"] = v.RoleIds row["create_time"] = beego.Date(time.Unix(v.CreateTime, 0), "Y-m-d H:i:s") row["update_time"] = beego.Date(time.Unix(v.UpdateTime, 0), "Y-m-d H:i:s") diff --git a/controllers/notify_tpl.go b/controllers/notify_tpl.go index b442a0e5..fbf1e05b 100644 --- a/controllers/notify_tpl.go +++ b/controllers/notify_tpl.go @@ -59,7 +59,7 @@ func (self *NotifyTplController) AjaxSave() { notifyTpl.Type = models.NotifyTplTypeDefault notifyTpl.Status, _ = self.GetInt("status") - if notifyTpl.TplType == 1 { + if notifyTpl.TplType == 1 || notifyTpl.TplType == 3 { m := make(map[string]string) err := json.Unmarshal([]byte(notifyTpl.Content), &m) if err != nil { @@ -85,7 +85,7 @@ func (self *NotifyTplController) AjaxSave() { notifyTpl.Content = strings.TrimSpace(self.GetString("content")) notifyTpl.Status, _ = self.GetInt("status") - if notifyTpl.TplType == 1 { + if notifyTpl.TplType == 1 || notifyTpl.TplType == 3 { m := make(map[string]string) err := json.Unmarshal([]byte(notifyTpl.Content), &m) if err != nil { @@ -137,6 +137,7 @@ func (self *NotifyTplController) Table() { "邮件", "信息", "钉钉", + "微信", } self.pageSize = limit diff --git a/controllers/user.go b/controllers/user.go index 2fdfff5b..d1a3a535 100644 --- a/controllers/user.go +++ b/controllers/user.go @@ -30,6 +30,7 @@ func (self *UserController) Edit() { row["phone"] = Admin.Phone row["email"] = Admin.Email row["dingtalk"] = Admin.Dingtalk + row["wechat"] = Admin.Wechat self.Data["admin"] = row self.display() } @@ -46,6 +47,7 @@ func (self *UserController) AjaxSave() { Admin.Phone = strings.TrimSpace(self.GetString("phone")) Admin.Email = strings.TrimSpace(self.GetString("email")) Admin.Dingtalk = strings.TrimSpace(self.GetString("dingtalk")) + Admin.Wechat = strings.TrimSpace(self.GetString("wechat")) resetPwd := self.GetString("reset_pwd") if resetPwd == "1" { diff --git a/jobs/job.go b/jobs/job.go index 98a0119e..b5d4ead6 100644 --- a/jobs/job.go +++ b/jobs/job.go @@ -280,6 +280,7 @@ func (j *Job) Run() { adminInfo := AllAdminInfo(j.task.NotifyUserIds) phone := make(map[string]string, 0) dingtalk := make(map[string]string, 0) + wechat := make(map[string]string, 0) toEmail := "" for _, v := range adminInfo { if v.Phone != "0" && v.Phone != "" { @@ -291,6 +292,9 @@ func (j *Job) Run() { if v.Dingtalk != "0" && v.Dingtalk != "" { dingtalk[v.Dingtalk] = v.Dingtalk } + if v.Wechat != "0" && v.Wechat != "" { + wechat[v.Wechat] = v.Wechat + } } toEmail = strings.TrimRight(toEmail, ";") @@ -302,8 +306,8 @@ func (j *Job) Run() { status := log.Status + 2 - var title = "" - var content = "" + title, content := "", "" + notifyTpl, err := models.NotifyTplGetById(j.task.NotifyTplId) if err != nil { notifyTpl, err := models.NotifyTplGetByTplType(j.task.NotifyType, models.NotifyTplTypeSystem) @@ -319,30 +323,28 @@ func (j *Job) Run() { if title != "" { title = strings.Replace(title, "{{TaskId}}", strconv.Itoa(j.task.Id), -1) title = strings.Replace(title, "{{TaskName}}", j.task.TaskName, -1) - title = strings.Replace(title, "{{CreateTime}}", beego.Date(time.Unix(log.CreateTime, 0), "Y-m-d H:i:s"), -1) + title = strings.Replace(title, "{{ExecuteTime}}", beego.Date(time.Unix(log.CreateTime, 0), "Y-m-d H:i:s"), -1) title = strings.Replace(title, "{{ProcessTime}}", strconv.FormatFloat(float64(log.ProcessTime)/1000, 'f', 6, 64), -1) - title = strings.Replace(title, "{{Status}}", TextStatus[status], -1) - title = strings.Replace(title, "{{TaskOut}}", log.Error, -1) + title = strings.Replace(title, "{{ExecuteStatus}}", TextStatus[status], -1) + title = strings.Replace(title, "{{TaskOutput}}", log.Error, -1) } if content != "" { content = strings.Replace(content, "{{TaskId}}", strconv.Itoa(j.task.Id), -1) content = strings.Replace(content, "{{TaskName}}", j.task.TaskName, -1) - content = strings.Replace(content, "{{CreateTime}}", beego.Date(time.Unix(log.CreateTime, 0), "Y-m-d H:i:s"), -1) + content = strings.Replace(content, "{{ExecuteTime}}", beego.Date(time.Unix(log.CreateTime, 0), "Y-m-d H:i:s"), -1) content = strings.Replace(content, "{{ProcessTime}}", strconv.FormatFloat(float64(log.ProcessTime)/1000, 'f', 6, 64), -1) - content = strings.Replace(content, "{{Status}}", TextStatus[status], -1) - content = strings.Replace(content, "{{TaskOut}}", log.Error, -1) + content = strings.Replace(content, "{{ExecuteStatus}}", TextStatus[status], -1) + content = strings.Replace(content, "{{TaskOutput}}", log.Error, -1) } if j.task.NotifyType == 0 && toEmail != "" { //邮件 mailtype := "html" - ok := notify.SendToChan(toEmail, title, content, mailtype) if !ok { fmt.Println("发送邮件错误", toEmail) } - } else if j.task.NotifyType == 1 && len(phone) > 0 { //信息 param := make(map[string]string) @@ -352,12 +354,30 @@ func (j *Job) Run() { return } - notify.SendSmsToChan(phone, param) + ok := notify.SendSmsToChan(phone, param) + if !ok { + fmt.Println("发送信息错误", phone) + } } else if j.task.NotifyType == 2 && len(dingtalk) > 0 { //钉钉 - notify.SendDingtalkToChan(dingtalk, content) - } + ok := notify.SendDingtalkToChan(dingtalk, content) + if !ok { + fmt.Println("发送钉钉错误", dingtalk) + } + } else if j.task.NotifyType == 3 && len(wechat) > 0 { + //信息 + param := make(map[string]string) + err := json.Unmarshal([]byte(content), ¶m) + if err != nil { + fmt.Println("发送微信错误", err) + return + } + ok := notify.SendWechatToChan(phone, param) + if !ok { + fmt.Println("发送微信错误", phone) + } + } } } @@ -375,6 +395,7 @@ type adminInfo struct { Email string Phone string Dingtalk string + Wechat string RealName string } @@ -400,6 +421,7 @@ func AllAdminInfo(adminIds string) []*adminInfo { Email: v.Email, Phone: v.Phone, Dingtalk: v.Dingtalk, + Wechat: v.Wechat, RealName: v.RealName, } adminInfos = append(adminInfos, &ai) diff --git a/models/admin.go b/models/admin.go index 0f0138a0..0dc630e0 100644 --- a/models/admin.go +++ b/models/admin.go @@ -20,6 +20,7 @@ type Admin struct { Phone string Email string Dingtalk string + Wechat string Salt string LastLogin int64 LastIp string diff --git a/notify/sms.go b/notify/sms.go index 16682e63..df238b25 100644 --- a/notify/sms.go +++ b/notify/sms.go @@ -16,7 +16,7 @@ import ( "github.com/pkg/errors" ) -type AjaxReturn struct { +type SmsAjaxReturn struct { Status int `json:"status"` Message string `json:"message"` Data interface{} `json:"data"` @@ -78,7 +78,7 @@ func (s *Sms) SendSms() error { return err } - ajaxData := AjaxReturn{} + ajaxData := SmsAjaxReturn{} jsonErr := json.Unmarshal([]byte(res), &ajaxData) if jsonErr != nil { diff --git a/notify/wechat.go b/notify/wechat.go new file mode 100644 index 00000000..a1d45a89 --- /dev/null +++ b/notify/wechat.go @@ -0,0 +1,89 @@ +package notify + +import ( + "github.com/astaxie/beego" + "github.com/george518/PPGo_Job/libs" + "log" + "time" + "encoding/json" + "github.com/pkg/errors" +) + +type WechatAjaxReturn struct { + Status int `json:"status"` + Message string `json:"message"` + Data interface{} `json:"data"` +} + +type Wechat struct { + Accounts map[string]string + Param map[string]string +} + +var WechatChan chan *Wechat +var WechatUrl string + +func init() { + WechatUrl = beego.AppConfig.String("wechat.url") + poolSize, _ := beego.AppConfig.Int("wechat.pool") + + //创建通道 + WechatChan = make(chan *Wechat, poolSize) + + go func() { + for { + select { + case m, ok := <-WechatChan: + if !ok { + return + } + if err := m.SendWechat(); err != nil { + beego.Error("SendWechat:", err.Error()) + } + } + } + }() + +} + +func SendWechatToChan(accounts map[string]string, param map[string]string) bool { + wechat := &Wechat{ + Accounts: accounts, + Param: param, + } + + select { + case WechatChan <- wechat: + return true + case <-time.After(time.Second * 3): + return false + } +} + +func (s *Wechat) SendWechat() error { + + for _, v := range s.Accounts { + s.Param["account"] = v + res, err := libs.HttpGet(WechatUrl, s.Param) + + if err != nil { + log.Println(err) + return err + } + + ajaxData := WechatAjaxReturn{} + jsonErr := json.Unmarshal([]byte(res), &ajaxData) + + if jsonErr != nil { + return jsonErr + } + + if ajaxData.Status != 200 { + return errors.Errorf("msg %s", ajaxData.Message) + } + + return nil + + } + return nil +} diff --git a/ppgo_job2.sql b/ppgo_job2.sql index 0bbb118b..2761e968 100644 --- a/ppgo_job2.sql +++ b/ppgo_job2.sql @@ -15,30 +15,6 @@ SET NAMES utf8; SET FOREIGN_KEY_CHECKS = 0; --- ---------------------------- --- Table structure for `default` --- ---------------------------- -DROP TABLE IF EXISTS `default`; -CREATE TABLE `default` ( - `id` int(11) unsigned NOT NULL AUTO_INCREMENT, - `user_name` varchar(20) NOT NULL DEFAULT '' COMMENT '用户名', - `email` varchar(50) NOT NULL DEFAULT '' COMMENT '邮箱', - `password` char(32) NOT NULL DEFAULT '' COMMENT '密码', - `salt` char(10) NOT NULL DEFAULT '' COMMENT '密码盐', - `last_login` int(11) NOT NULL DEFAULT '0' COMMENT '最后登录时间', - `last_ip` char(15) NOT NULL DEFAULT '' COMMENT '最后登录IP', - `status` tinyint(4) NOT NULL DEFAULT '0' COMMENT '状态,0正常 -1禁用', - PRIMARY KEY (`id`), - UNIQUE KEY `idx_user_name` (`user_name`) -) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8; - --- ---------------------------- --- Records of `default` --- ---------------------------- -BEGIN; -INSERT INTO `default` VALUES ('1', 'admin', 'admin@example.com', '7fef6171469e80d32c0559f88b377245', '', '0', '', '0'); -COMMIT; - -- ---------------------------- -- Table structure for `pp_task` -- ---------------------------- @@ -297,34 +273,6 @@ BEGIN; INSERT INTO `pp_uc_role_auth` VALUES ('1', '1'), ('1', '15'), ('1', '16'), ('1', '30'), ('1', '31'), ('1', '36'), ('1', '37'), ('1', '39'), ('1', '41'), ('1', '46'), ('1', '47'), ('1', '48'), ('1', '49'), ('1', '54'), ('1', '55'), ('1', '56'), ('2', '0'), ('2', '17'), ('2', '18'), ('2', '19'), ('2', '20'), ('2', '21'), ('2', '22'), ('2', '23'), ('2', '24'), ('2', '25'), ('2', '26'), ('2', '27'), ('2', '28'), ('2', '29'), ('2', '32'), ('2', '33'), ('2', '34'), ('2', '35'), ('2', '38'), ('2', '40'), ('2', '42'), ('2', '43'), ('2', '44'), ('2', '45'), ('2', '50'), ('2', '51'), ('2', '52'), ('2', '53'); COMMIT; --- ---------------------------- --- Table structure for `pp_user` --- ---------------------------- -DROP TABLE IF EXISTS `pp_user`; -CREATE TABLE `pp_user` ( - `id` int(11) unsigned NOT NULL AUTO_INCREMENT, - `user_name` varchar(20) NOT NULL DEFAULT '' COMMENT '用户名', - `email` varchar(50) NOT NULL DEFAULT '' COMMENT '邮箱', - `password` char(32) NOT NULL DEFAULT '' COMMENT '密码', - `salt` char(10) NOT NULL DEFAULT '' COMMENT '密码盐', - `last_login` int(11) NOT NULL DEFAULT '0' COMMENT '最后登录时间', - `last_ip` char(15) NOT NULL DEFAULT '' COMMENT '最后登录IP', - `status` tinyint(4) NOT NULL DEFAULT '0' COMMENT '状态,0正常 -1禁用', - PRIMARY KEY (`id`), - UNIQUE KEY `idx_user_name` (`user_name`) -) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8; - --- ---------------------------- --- Records of `pp_user` --- ---------------------------- -BEGIN; -INSERT INTO `pp_user` VALUES ('1', 'admin', 'haodaquan@shoplinq.cn', 'abfcf6dcedfb4b5b1505d41a8b4c77e8', 'aYk4Q1P83v', '1528124357', '[', '0'); -COMMIT; - -BEGIN; -ALTER TABLE `pp_task` CHANGE `notify_type` `notify_type` TINYINT(1) UNSIGNED NOT NULL DEFAULT '0' COMMENT '0-邮件通知,1-信息通知,2-钉钉通知,'; -COMMIT; - BEGIN; ALTER TABLE `pp_uc_admin` ADD `dingtalk` VARCHAR(64) NULL COMMENT '钉钉' AFTER `email`; COMMIT; @@ -349,7 +297,7 @@ CREATE TABLE `pp_notify_tpl` ( `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '模板id', `type` enum('system','default') CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT 'default', `tpl_name` varchar(20) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '模板名称', - `tpl_type` tinyint(1) NOT NULL COMMENT '模板类型 0:邮件;1:信息;2:钉钉;', + `tpl_type` tinyint(1) NOT NULL COMMENT '模板类型 0:邮件;1:信息;2:钉钉;3:微信;', `title` varchar(64) DEFAULT NULL COMMENT '标题', `content` text CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '模板内容', `status` tinyint(1) NOT NULL DEFAULT '1' COMMENT '状态 0:禁用;1:启用;', @@ -358,15 +306,24 @@ CREATE TABLE `pp_notify_tpl` ( `update_time` int(11) NOT NULL DEFAULT '0' COMMENT '最后一次编辑时间', `update_id` int(11) NOT NULL DEFAULT '0' COMMENT '最后一次编辑者ID', PRIMARY KEY (`id`) -) ENGINE=InnoDB AUTO_INCREMENT=4 DEFAULT CHARSET=utf8 COMMENT='通知模板'; +) ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8 COMMENT='通知模板'; -- -- 转存表中的数据 `pp_notify_tpl` -- BEGIN; -INSERT INTO `pp_notify_tpl` VALUES(1, 'system', '默认邮箱通知模板', 0, '定时任务异常:{{TaskName}}', 'Hello,定时任务出问题了:\r\n
任务执行详情:
\r\n\r\n任务 ID:{{TaskId}}
\r\n任务名称:{{TaskName}}
\r\n执行时间:{{CreateTime}}
\r\n执行耗时:{{ProcessTime}}秒
\r\n执行状态:{{Status}}\r\n
任务执行输出
\r\n\r\n{{TaskOut}}\r\n
', 1, 1550255030, 1, 1550338305, 1); -INSERT INTO `pp_notify_tpl` VALUES(2, 'system', '默认短信通知模板', 1, '', '{\r\n \"task_id\": \"{{TaskId}}\",\r\n \"task_name\": \"{{TaskName}}\",\r\n \"status\": \"{{Status}}\"\r\n}', 1, 1550255030, 1, 1550338215, 1); -INSERT INTO `pp_notify_tpl` VALUES(3, 'system', '默认钉钉通知模板', 2, '', '任务执行异常详情:\r\n任务 ID:{{TaskId}}\r\n任务名称:{{TaskName}}\r\n执行时间:{{CreateTime}}\r\n执行耗时:{{ProcessTime}}秒\r\n执行状态:{{Status}}\r\n任务执行输出:\r\n{{TaskOut}}', 1, 1550255030, 1, 1550338191, 1); +INSERT INTO `pp_notify_tpl` VALUES(1, 'system', '默认邮箱通知模板', 0, '定时任务异常:{{TaskName}}', 'Hello,定时任务出问题了:\r\n任务执行详情:
\r\n\r\n任务 ID:{{TaskId}}
\r\n任务名称:{{TaskName}}
\r\n执行时间:{{ExecuteTime}}
\r\n执行耗时:{{ProcessTime}}秒
\r\n执行状态:{{ExecuteStatus}}\r\n
任务执行输出
\r\n\r\n{{TaskOutput}}\r\n
', 1, 1550255030, 1, 1550338305, 1); +INSERT INTO `pp_notify_tpl` VALUES(2, 'system', '默认短信通知模板', 1, '', '{\r\n \"task_id\": \"{{TaskId}}\",\r\n \"task_name\": \"{{TaskName}}\",\r\n \"execute_status\": \"{{ExecuteStatus}}\"\r\n}', 1, 1550255030, 1, 1550338215, 1); +INSERT INTO `pp_notify_tpl` VALUES(3, 'system', '默认钉钉通知模板', 2, '', '任务执行异常详情:\r\n任务 ID:{{TaskId}}\r\n任务名称:{{TaskName}}\r\n执行时间:{{ExecuteTime}}\r\n执行耗时:{{ProcessTime}}秒\r\n执行状态:{{ExecuteStatus}}\r\n任务执行输出:\r\n{{TaskOutput}}', 1, 1550255030, 1, 1550338880, 1); +INSERT INTO `pp_notify_tpl` VALUES(4, 'system', '默认微信通知模板', 3, '', '{\r\n \"task_id\": \"{{TaskId}}\",\r\n \"task_name\": \"{{TaskName}}\",\r\n \"execute_status\": \"{{ExecuteStatus}}\"\r\n}', 1, 1550347183, 1, 1550347201, 1); +COMMIT; + +BEGIN; +ALTER TABLE `pp_task` CHANGE `notify_type` `notify_type` TINYINT(1) UNSIGNED NOT NULL DEFAULT '0' COMMENT '0-邮件通知,1-信息通知,2-钉钉通知,3-微信通知,'; +COMMIT; + +BEGIN; +ALTER TABLE `pp_uc_admin` ADD `wechat` VARCHAR(64) NULL COMMENT '微信' AFTER `dingtalk`; COMMIT; SET FOREIGN_KEY_CHECKS = 1; diff --git a/views/admin/add.html b/views/admin/add.html index 633e4ee0..5af1064d 100644 --- a/views/admin/add.html +++ b/views/admin/add.html @@ -42,7 +42,13 @@