Create & Init Project...

app/job/main/archive/dao/monitor/BUILD

@@ -0,0 +1,51 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = [
+        "dao_test.go",
+        "monitor_test.go",
+    ],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/job/main/archive/conf:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "dao.go",
+        "monitor.go",
+    ],
+    importpath = "go-common/app/job/main/archive/dao/monitor",
+    tags = ["automanaged"],
+    deps = [
+        "//app/job/main/archive/conf:go_default_library",
+        "//library/log:go_default_library",
+        "//library/net/http/blademaster:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/job/main/archive/dao/monitor/dao.go

@@ -0,0 +1,21 @@
+package monitor
+
+import (
+	"go-common/app/job/main/archive/conf"
+	bm "go-common/library/net/http/blademaster"
+)
+
+// Dao is redis dao.
+type Dao struct {
+	c      *conf.Config
+	client *bm.Client
+}
+
+// New is new redis dao.
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		c:      c,
+		client: bm.NewClient(c.HTTPClient),
+	}
+	return d
+}

app/job/main/archive/dao/monitor/dao_test.go

@@ -0,0 +1,19 @@
+package monitor
+
+import (
+	"flag"
+	"path/filepath"
+
+	"go-common/app/job/main/archive/conf"
+)
+
+var (
+	d *Dao
+)
+
+func init() {
+	dir, _ := filepath.Abs("../../cmd/archive-job-test.toml")
+	flag.Set("conf", dir)
+	conf.Init()
+	d = New(conf.Conf)
+}

app/job/main/archive/dao/monitor/monitor.go

@@ -0,0 +1,55 @@
+package monitor
+
+import (
+	"context"
+	"crypto/md5"
+	"encoding/hex"
+	"encoding/json"
+	"net/http"
+	"net/url"
+	"strconv"
+	"strings"
+	"time"
+
+	"go-common/library/log"
+)
+
+type wxParams struct {
+	Username  string `json:"username"`
+	Content   string `json:"content"`
+	Token     string `json:"token"`
+	Timestamp int64  `json:"timestamp"`
+	Sign      string `json:"signature"`
+}
+type resp struct {
+	Status int64  `json:"status"`
+	Msg    string `json:"msg"`
+}
+
+// Send send message to phone
+func (d *Dao) Send(c context.Context, users, msg, token, secret string) (err error) {
+	params := url.Values{}
+	params.Set("username", users)
+	params.Set("content", msg)
+	params.Set("token", token)
+	params.Set("timestamp", strconv.FormatInt(time.Now().Unix(), 10))
+	mh := md5.Sum([]byte(params.Encode() + secret))
+	params.Set("signature", hex.EncodeToString(mh[:]))
+	p := &wxParams{
+		Username: params.Get("username"),
+		Content:  params.Get("content"),
+		Token:    params.Get("token"),
+		Sign:     params.Get("signature"),
+	}
+	p.Timestamp, _ = strconv.ParseInt(params.Get("timestamp"), 10, 64)
+	bs, _ := json.Marshal(p)
+	payload := strings.NewReader(string(bs))
+	req, _ := http.NewRequest("POST", "http://bap.bilibili.co/api/v1/message/add", payload)
+	req.Header.Add("content-type", "application/json; charset=utf-8")
+	v := &resp{}
+	if err = d.client.Do(context.TODO(), req, v); err != nil {
+		log.Error("s.client.Do error(%v)", err)
+		return
+	}
+	return
+}

app/job/main/archive/dao/monitor/monitor_test.go

@@ -0,0 +1,17 @@
+package monitor
+
+import (
+	"context"
+	"testing"
+
+	"go-common/app/job/main/archive/conf"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func Test_Send(t *testing.T) {
+	Convey("Send", t, func() {
+		err := d.Send(context.TODO(), conf.Conf.WeChantUsers, "报警短信test", conf.Conf.WeChatToken, conf.Conf.WeChatSecret)
+		So(err, ShouldBeNil)
+	})
+}