Create & Init Project...

app/job/main/sms/dao/mengwang/BUILD

@@ -0,0 +1,48 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["client_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/job/main/sms/conf:go_default_library",
+        "//app/service/main/sms/model:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["client.go"],
+    importpath = "go-common/app/job/main/sms/dao/mengwang",
+    tags = ["automanaged"],
+    deps = [
+        "//app/job/main/sms/conf:go_default_library",
+        "//app/job/main/sms/model:go_default_library",
+        "//app/service/main/sms/model: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/sms/dao/mengwang/client.go

@@ -0,0 +1,209 @@
+package mengwang
+
+import (
+	"bytes"
+	"context"
+	"crypto/md5"
+	"encoding/hex"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/url"
+	"strconv"
+	"time"
+
+	"go-common/app/job/main/sms/conf"
+	"go-common/app/job/main/sms/model"
+	smsmdl "go-common/app/service/main/sms/model"
+	"go-common/library/log"
+	bm "go-common/library/net/http/blademaster"
+)
+
+// Client .
+type Client struct {
+	c      conf.Provider
+	client *bm.Client
+}
+
+type response struct {
+	Result int   `json:"result"`
+	MsgID  int64 `json:"msgid"`
+}
+
+// GetPid gets MengWang type ID.
+func (v *Client) GetPid() int32 {
+	return smsmdl.ProviderMengWang
+}
+
+// NewClient new MengWang.
+func NewClient(c *conf.Config) *Client {
+	return &Client{
+		c:      *c.Provider,
+		client: bm.NewClient(c.HTTPClient),
+	}
+}
+
+func (v *Client) post(ctx context.Context, url string, params interface{}, res *response) (err error) {
+	var bs []byte
+	if bs, err = json.Marshal(params); err != nil {
+		log.Error("json.Marshal param(%v) error(%v)", params, err)
+		return
+	}
+	req, err := http.NewRequest(http.MethodPost, url, bytes.NewBuffer([]byte(bs)))
+	if err != nil {
+		log.Error("http.NewRequest(%s) param(%v) error(%v)", url, params, err)
+		return
+	}
+	req.Header.Set("Content-Type", "application/json")
+	if err = v.client.Do(ctx, req, &res); err != nil {
+		log.Error("client.Do(%s,%v) error(%v)", url, params, err)
+		return
+	}
+	log.Info("url(%s) body(%v) resp(%+v)", url, params, res)
+	return
+}
+
+// SendSms sends MengWang sms.
+func (v *Client) SendSms(ctx context.Context, r *smsmdl.ModelSend) (msgid string, err error) {
+	params := make(map[string]string)
+	pwd, ts := genPwd(v.c.MengWangSmsUser, v.c.MengWangSmsPwd)
+	params["userid"] = v.c.MengWangSmsUser
+	params["pwd"] = pwd
+	params["timestamp"] = ts
+	params["mobile"] = r.Mobile
+	params["content"] = url.QueryEscape(r.Content)
+	res := new(response)
+	if err = v.post(ctx, v.c.MengWangSmsURL, params, res); err != nil {
+		log.Error("mengwang SendSms param(%v) error(%v)", params, err)
+		return
+	}
+	if res.Result != 0 {
+		err = fmt.Errorf("mengwang SendSms param(%v) error(%v)", params, res.Result)
+		return
+	}
+	msgid = strconv.FormatInt(res.MsgID, 10)
+	return
+}
+
+// SendActSms sends MengWang act sms.
+func (v *Client) SendActSms(ctx context.Context, r *smsmdl.ModelSend) (msgid string, err error) {
+	params := make(map[string]string)
+	pwd, ts := genPwd(v.c.MengWangActUser, v.c.MengWangActPwd)
+	r.Content = r.Content + model.SmsSuffix
+	params["userid"] = v.c.MengWangActUser
+	params["pwd"] = pwd
+	params["timestamp"] = ts
+	params["mobile"] = r.Mobile
+	params["content"] = url.QueryEscape(r.Content)
+	res := new(response)
+	if err = v.post(ctx, v.c.MengWangActURL, params, res); err != nil {
+		log.Error("mengwang SendActSms param(%v) error(%v)", params, err)
+		return
+	}
+	if res.Result != 0 {
+		err = fmt.Errorf("mengwang SendActSms param(%v) error(%v)", params, res.Result)
+		return
+	}
+	msgid = strconv.FormatInt(res.MsgID, 10)
+	return
+}
+
+// SendBatchActSms sends multi MengWang act sms.
+func (v *Client) SendBatchActSms(ctx context.Context, r *smsmdl.ModelSend) (msgid string, err error) {
+	params := make(map[string]string)
+	pwd, ts := genPwd(v.c.MengWangActUser, v.c.MengWangActPwd)
+	params["userid"] = v.c.MengWangActUser
+	params["pwd"] = pwd
+	params["timestamp"] = ts
+	params["mobile"] = r.Mobile
+	params["content"] = url.QueryEscape(r.Content + model.SmsSuffix)
+	res := new(response)
+	if err = v.post(ctx, v.c.MengWangBatchURL, params, res); err != nil {
+		log.Error("mengwang SendBatchActSms param(%v) error(%v)", params, err)
+		return
+	}
+	if res.Result != 0 {
+		err = fmt.Errorf("mengwang SendBatchActSms param(%v) error(%v)", params, res.Result)
+		return
+	}
+	msgid = strconv.FormatInt(res.MsgID, 10)
+	return
+}
+
+// SendInternationalSms sends MengWang international sms.
+func (v *Client) SendInternationalSms(ctx context.Context, r *smsmdl.ModelSend) (msgid string, err error) {
+	params := make(map[string]string)
+	pwd, ts := genPwd(v.c.MengWangInternationUser, v.c.MengWangInternationPwd)
+	params["userid"] = v.c.MengWangInternationUser
+	params["pwd"] = pwd
+	params["timestamp"] = ts
+	params["mobile"] = "00" + r.Country + r.Mobile
+	params["content"] = url.QueryEscape(r.Content)
+	res := new(response)
+	if err = v.post(ctx, v.c.MengWangInternationURL, params, res); err != nil {
+		log.Error("mengwang SendInternationalSms param(%v) error(%v)", params, err)
+		return
+	}
+	if res.Result != 0 {
+		err = fmt.Errorf("mengwang SendInternationalSms param(%v) error(%v)", params, res.Result)
+		return
+	}
+	msgid = strconv.FormatInt(res.MsgID, 10)
+	return
+}
+
+// Callback .
+type Callback struct {
+	MsgID      int64  `json:"msgid"`
+	Num        int    `json:"pknum"`
+	Total      int    `json:"pktotal"`
+	Mobile     string `json:"mobile"`
+	SendTime   string `json:"stime"`
+	ReportTime string `json:"rtime"`
+	Status     string `json:"errcode"`
+	Desc       string `json:"errdesc"`
+}
+
+type callbackResponse struct {
+	Result    int         `json:"result"`
+	Callbacks []*Callback `json:"rpts"`
+}
+
+// Callback .
+func (v *Client) Callback(ctx context.Context, user, pwd, url string, count int) (callbacks []*Callback, err error) {
+	pwd, ts := genPwd(user, pwd)
+	params := make(map[string]string)
+	params["userid"] = user
+	params["pwd"] = pwd
+	params["timestamp"] = ts
+	params["retsize"] = strconv.Itoa(count)
+	bs, err := json.Marshal(params)
+	if err != nil {
+		log.Error("json.Marshal param(%v) error(%v)", params, err)
+		return
+	}
+	req, err := http.NewRequest(http.MethodPost, url, bytes.NewBuffer([]byte(bs)))
+	if err != nil {
+		log.Error("http.NewRequest(%s) param(%v) error(%v)", url, params, err)
+		return
+	}
+	req.Header.Set("Content-Type", "application/json")
+	res := new(callbackResponse)
+	if err = v.client.Do(ctx, req, &res); err != nil {
+		log.Error("client.Do(%s,%v) error(%v)", url, params, err)
+		return
+	}
+	if res.Result != 0 {
+		err = fmt.Errorf("mengwang callback param(%v) res(%+v)", params, res)
+		return
+	}
+	callbacks = res.Callbacks
+	return
+}
+
+func genPwd(user, pwd string) (spwd, ts string) {
+	ft := time.Now().Format("0102150405")
+	str := fmt.Sprintf("%s%s%s%s", user, "00000000", pwd, ft)
+	mh := md5.Sum([]byte(str))
+	return hex.EncodeToString(mh[:]), ft
+}

app/job/main/sms/dao/mengwang/client_test.go

@@ -0,0 +1,71 @@
+package mengwang
+
+import (
+	"context"
+	"flag"
+	"path/filepath"
+	"testing"
+
+	"go-common/app/job/main/sms/conf"
+	"go-common/app/service/main/sms/model"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var (
+	mw  *Client
+	sl  = &model.ModelSend{Mobile: "", Content: "卍 测试短信，验证码为5467当日有效 https://search.bilibili.com/all?keyword=你好"} // 17621660828
+	isl = &model.ModelSend{Country: "852", Mobile: "", Content: "卍 您的账号正在哔哩哔哩2017动画角色人气大赏活动中进行领票操作，验证码为5467当日有效"} // 00852 69529378 梦网的香港测试号
+)
+
+func init() {
+	dir, _ := filepath.Abs("../../cmd/sms-job-test.toml")
+	flag.Set("conf", dir)
+	conf.Init()
+	mw = NewClient(conf.Conf)
+}
+
+func TestSendSms(t *testing.T) {
+	Convey("mengwang send sms", t, func() {
+		msgid, err := mw.SendSms(context.TODO(), sl)
+		So(err, ShouldBeNil)
+		t.Logf("msgid(%s)", msgid)
+	})
+}
+
+func TestSendActSms(t *testing.T) {
+	Convey("mengwang send act sms", t, func() {
+		msgid, err := mw.SendActSms(context.TODO(), sl)
+		So(err, ShouldBeNil)
+		t.Logf("msgid(%s)", msgid)
+	})
+}
+
+func TestSendBatchSms(t *testing.T) {
+	Convey("mengwang send batch sms", t, func() {
+		msl := sl
+		msl.Mobile = "" // 17621660828,17621660828
+		msgid, err := mw.SendActSms(context.TODO(), msl)
+		So(err, ShouldBeNil)
+		t.Logf("msgid(%s)", msgid)
+	})
+}
+
+func TestSendInternationalSms(t *testing.T) {
+	Convey("mengwang send international sms", t, func() {
+		msgid, err := mw.SendInternationalSms(context.TODO(), isl)
+		So(err, ShouldBeNil)
+		t.Logf("msgid(%s)", msgid)
+	})
+}
+
+func TestCallback(t *testing.T) {
+	Convey("mengwang callback", t, func() {
+		// callbacks, err := mw.Callback(context.Background(), conf.Conf.Pconf.MengWangSmsUser, conf.Conf.Pconf.MengWangSmsPwd, conf.Conf.Pconf.MengWangSmsCallbackURL, 5)
+		// So(err, ShouldBeNil)
+		// t.Logf("callbacks(%d)", len(callbacks))
+		// for _, v := range callbacks {
+		// 	t.Logf("callback(%+v)", v)
+		// }
+	})
+}