add sms support

 package main
 package main
 
 
-import _ "openapi/pkg/log"
-
 import (
 import (
+	"fmt"
+
 	"github.com/astaxie/beego"
 	"github.com/astaxie/beego"
+	"gitlab.fjmaimaimai.com/mmm-go/gocomm/config"
 	"gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/log"
 	"gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/log"
-
+	"gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/redis"
+	"openapi/pkg/constant"
 	_ "openapi/pkg/infrastructure/bgorm"
 	_ "openapi/pkg/infrastructure/bgorm"
+	_ "openapi/pkg/log"
 	_ "openapi/pkg/port/beego"
 	_ "openapi/pkg/port/beego"
 )
 )
 
 
@@ -16,5 +19,13 @@ func main() {
 	}()
 	}()
 	log.Info("server on start!")
 	log.Info("server on start!")
 	//constant.DebugConfig()
 	//constant.DebugConfig()
+	err := redis.Init(config.Redis{
+		Addr:     fmt.Sprintf("%v:%v", constant.REDIS_HOST, constant.REDIS_PORT),
+		Password: constant.REDIS_AUTH,
+		MaxIdle:  50,
+	})
+	if err != nil {
+		log.Error(err)
+	}
 	beego.Run()
 	beego.Run()
 }
 }

+package service
+
+import (
+	"fmt"
+	"gitlab.fjmaimaimai.com/mmm-go/gocomm/common"
+	"gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/log"
+	"gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/redis"
+	protocol "openapi/pkg/domain"
+	"openapi/pkg/infrastructure/sms"
+	"openapi/pkg/infrastructure/sms/yunpian"
+	"strings"
+)
+
+// SmsCode 发送短信验证码
+func SmsCode(header *protocol.RequestHeader, request *protocol.SmsCodeRequest) (rsp interface{}, err error) {
+	code := common.RandomStringWithChars(6, protocol.Nums)
+	var smsSvr sms.ISmsService = &yunpian.YPSmsService{}
+	content := fmt.Sprintf("【买买买信息科技】%v(手机验证码，请完成验证)，如非本人操作，请忽略本短信", code)
+	// save log to db
+	log.Debug(fmt.Sprintf("【sms】 smscode phone:%v code:%v", request.Phone, code))
+	// save to redis
+	if err = redis.Set(smsRedisKey(request.Phone), code, sms.DefaultSmsCodeTimeOut); err != nil {
+		logError(err)
+		err = protocol.NewCustomMessage(1001, "请求超时，请检查手机号是否正确或网络连接状态")
+		return nil, err
+	}
+	if err = smsSvr.Send(&sms.Options{Phone: request.Phone, SendType: sms.SendSingle, Content: content}); err != nil {
+		logError(err)
+		err = protocol.NewCustomMessage(1001, "请求超时，请检查手机号是否正确或网络连接状态")
+		return nil, err
+	}
+	return
+}
+
+// CheckSmsCode 检查短信验证码
+func CheckSmsCode(header *protocol.RequestHeader, request *protocol.CheckSmsCodeRequest) (rsp interface{}, err error) {
+	var value string
+	if value, err = redis.Get(smsRedisKey(request.Phone)); err != nil || len(value) == 0 {
+		logError(err)
+		err = protocol.NewCustomMessage(1004, "验证码已失效")
+		return nil, err
+	}
+	if !strings.EqualFold(value, request.Code) {
+		err = protocol.NewCustomMessage(1004, "验证码错误，重新输入")
+		return nil, err
+	}
+	redis.Del(smsRedisKey(request.Phone))
+	return
+}
+
+// SendSmsNotice 发送短信通知
+func SendSmsNotice(header *protocol.RequestHeader, request *protocol.SendSmsNoticeRequest) (rsp interface{}, err error) {
+	var smsSvr sms.ISmsService = &yunpian.YPSmsService{}
+	options := sms.NewOptions(
+		sms.WithPhone(request.Phone),
+		sms.WithSendType(sms.SendSingleByTpl),
+		sms.WithTpl(request.TplId, request.TplValues),
+	)
+	err = smsSvr.Send(options)
+	return
+}
+
+func logError(err error) {
+	log.Error(fmt.Sprintf("【sms】 %v", err.Error()))
+}
+
+func smsRedisKey(phone string) string {
+	return strings.Join([]string{"sms", "smscode", phone}, ":")
+}

@@ -8,11 +8,13 @@ import (
 var (
 var (
 	LogFilePath string = "F:/log/app.log"
 	LogFilePath string = "F:/log/app.log"
 	LogLevel    string = "error"
 	LogLevel    string = "error"
+	RunMode     string = "dev"
 )
 )
 
 
 func init() {
 func init() {
 	LogLevel = config.StringDefault("log_level", LogLevel)
 	LogLevel = config.StringDefault("log_level", LogLevel)
 	LogFilePath = config.StringDefault("aliyun_logs_access", LogFilePath)
 	LogFilePath = config.StringDefault("aliyun_logs_access", LogFilePath)
+	RunMode = config.StringDefault("RUN_MODE", RunMode)
 }
 }
 
 
 func DebugConfig() {
 func DebugConfig() {

+package constant
+
+import "os"
+
+var REDIS_HOST = "127.0.0.1"
+var REDIS_PORT = "6379"
+var REDIS_AUTH = ""
+
+func init() {
+	if os.Getenv("REDIS_HOST") != "" {
+		REDIS_HOST = os.Getenv("REDIS_HOST")
+		REDIS_AUTH = os.Getenv("REDIS_AUTH")
+	}
+	if os.Getenv("REDIS_PORT") != "" {
+		REDIS_PORT = os.Getenv("REDIS_PORT")
+	}
+	if _, ok := os.LookupEnv("REDIS_AUTH"); ok {
+		REDIS_AUTH = os.Getenv("REDIS_AUTH")
+	}
+}

+package constant
+
+var (
+	YunPianAppKey      = "0bf6fb10a11a68a95dee80901eb545b5"
+	YunPianSDKHost     = "https://sms.yunpian.com/v2/"
+	SmsSendSingle      = "sms/single_send.json"
+	SmsSendSingleByTpl = "sms/tpl_single_send.json"
+)

@@ -5,4 +5,8 @@ var errmessge ErrorMap = map[int]string{
 	1:    "系统异常",
 	1:    "系统异常",
 	2:    "参数错误",
 	2:    "参数错误",
 	113:  "签名验证失败",
 	113:  "签名验证失败",
+	1001: "请求超时，请检查手机号是否正确或网络连接状态",
+	1002: "请输入正确的手机号码",
+	1003: "验证码错误，重新输入",
+	1004: "验证码已失效",
 }
 }

+package domain
+
+var Nums = "0123456789"
+
+/*发送短信验证码*/
+type SmsCodeRequest struct {
+	Project string `json:"project"`
+	Phone   string `json:"phone" valid:"required"`
+}
+
+type CheckSmsCodeRequest struct {
+	Phone string `json:"phone" valid:"required"`
+	Code  string `json:"code" valid:"required"`
+}
+
+type SendSmsNoticeRequest struct {
+	// 手机号
+	Phone string `json:"phone" valid:"required"`
+	// 模板id
+	TplId int `json:"tplId" valid:"required"`
+	// 模板参数
+	TplValues map[string]interface{} `json:"tplValues"`
+}

+package sms
+
+const (
+	SendSingle      SendType = 1
+	SendList        SendType = 2
+	SendSingleByTpl SendType = 10
+)
+
+const DefaultSmsCodeTimeOut int64 = 60 * 2
+
+type (
+	SendType    int
+	ISmsService interface {
+		Send(option *Options) error
+	}
+	Options struct {
+		Phone     string
+		Content   string
+		SendType  SendType
+		TplId     int
+		TplValues map[string]interface{}
+	}
+	option func(o *Options)
+)
+
+func NewOptions(options ...option) *Options {
+	o := &Options{}
+	for i := range options {
+		options[i](o)
+	}
+	return o
+}
+func WithPhone(phone string) option {
+	return func(o *Options) {
+		o.Phone = phone
+	}
+}
+func WithSendType(sendType SendType) option {
+	return func(o *Options) {
+		o.SendType = sendType
+	}
+}
+func WithTpl(tplId int, tplValues map[string]interface{}) option {
+	return func(o *Options) {
+		o.TplId = tplId
+		o.TplValues = tplValues
+	}
+}

+package yunpian
+
+import (
+	"bytes"
+	"fmt"
+	"github.com/astaxie/beego/httplib"
+	"gitlab.fjmaimaimai.com/mmm-go/gocomm/common"
+	"gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/log"
+	"net/url"
+	"openapi/pkg/constant"
+	"openapi/pkg/infrastructure/sms"
+	"strconv"
+)
+
+type (
+	YPSmsService  struct{}
+	YPSmsResponse struct {
+		Code   int     `json:"code"`   //0代表发送成功，其他code代表出错，详细见"返回值说明"页面
+		Msg    string  `json:"msg"`    //例如""发送成功""，或者相应错误信息
+		Count  int     `json:"count"`  //发送成功短信的计费条数(计费条数：70个字一条，超出70个字时按每67字一条计费)
+		Mobile string  `json:"string"` //发送手机号
+		Fee    float64 `json:"fee"`    //扣费金额，单位：元，类型：双精度浮点型/double
+		Sid    int64   `json:"sid"`    //短信id，64位整型
+	}
+)
+
+func (svr *YPSmsService) Send(option *sms.Options) error {
+	var err error
+	if option.SendType == sms.SendSingle {
+		_, err = sendSingle(option)
+	} else if option.SendType == sms.SendSingleByTpl {
+		_, err = sendSingleByTpl(option)
+	}
+	return err
+}
+
+func sendSingle(option *sms.Options) (map[string]interface{}, error) {
+	var (
+		resp *YPSmsResponse
+		err  error
+		url  = constant.YunPianSDKHost + constant.SmsSendSingle
+	)
+	post := httplib.Post(url)
+	post.Param("apikey", constant.YunPianAppKey)
+	post.Param("mobile", option.Phone)
+	post.Param("text", option.Content)
+	if err = post.ToJSON(&resp); err != nil {
+		return nil, err
+	}
+	err = resolveResponse(resp, err)
+	return nil, err
+}
+
+func sendSingleByTpl(option *sms.Options) (map[string]interface{}, error) {
+	var (
+		resp *YPSmsResponse
+		err  error
+		url  = constant.YunPianSDKHost + constant.SmsSendSingleByTpl
+	)
+	post := httplib.Post(url)
+	post.Param("apikey", constant.YunPianAppKey)
+	post.Param("mobile", option.Phone)
+	post.Param("tpl_id", strconv.Itoa(option.TplId))
+	if len(option.TplValues) > 0 {
+		post.Param("tpl_value", tplValue(option.TplValues))
+		log.Debug(tplValue(option.TplValues))
+	}
+	if err = post.ToJSON(&resp); err != nil {
+		return nil, err
+	}
+	err = resolveResponse(resp, err)
+	return nil, err
+}
+
+func resolveResponse(resp *YPSmsResponse, err error) error {
+	if resp.Code != 0 {
+		log.Error("【sms】 send sms code:", resp.Code, " error msg:", resp.Msg)
+		err = common.NewErrorWithMsg(1, resp.Msg)
+		return err
+	}
+	return err
+}
+
+func tplValue(tplValues map[string]interface{}) string {
+	var value bytes.Buffer
+	for k, v := range tplValues {
+		value.WriteString(url.PathEscape(fmt.Sprintf("#%v#=%v&", k, v)))
+	}
+	return string(value.Bytes()[:value.Len()-1])
+}

@@ -167,3 +167,10 @@ func GetPageInfo(pageIndex, pageSize int) (offset, size int) {
 	size = pageSize
 	size = pageSize
 	return
 	return
 }
 }
+
+func IsProductEnv(env string) bool {
+	if env == "prod" {
+		return true
+	}
+	return false
+}

+package v1
+
+import (
+	"bytes"
+	"encoding/json"
+	"github.com/astaxie/beego/validation"
+	"gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/log"
+	sms "openapi/pkg/application/sms/service"
+	"openapi/pkg/constant"
+	protocol "openapi/pkg/domain"
+	"openapi/pkg/infrastructure/utils"
+	"openapi/pkg/port/beego/controllers"
+)
+
+type SmsController struct {
+	controllers.BaseController
+}
+
+//短信验证码 SmsCode
+// @router /smsCode [post]
+func (this *SmsController) SmsCode() {
+	var msg *protocol.ResponseMessage
+	defer func() {
+		this.Resp(msg)
+	}()
+	var request *protocol.SmsCodeRequest
+	decoder := json.NewDecoder(bytes.NewBuffer(this.ByteBody))
+	decoder.UseNumber()
+	if err := decoder.Decode(&request); err != nil {
+		log.Error(err)
+		msg = protocol.BadRequestParam(1)
+		return
+	}
+	if utils.IsProductEnv(constant.RunMode) {
+		valid := validation.Validation{}
+		if !valid.Mobile(request.Phone, "mobile").Ok {
+			msg = protocol.BadRequestParam(1001)
+			return
+		}
+	}
+	header := controllers.GetRequestHeader(this.Ctx)
+	msg = protocol.NewReturnResponse(sms.SmsCode(header, request))
+}
+
+//检查短信验证码 CheckSmsCode
+// @router /checkSmsCode [post]
+func (this *SmsController) CheckSmsCode() {
+	var msg *protocol.ResponseMessage
+	defer func() {
+		this.Resp(msg)
+	}()
+	var request *protocol.CheckSmsCodeRequest
+	decoder := json.NewDecoder(bytes.NewBuffer(this.ByteBody))
+	decoder.UseNumber()
+	if err := decoder.Decode(&request); err != nil {
+		log.Error(err)
+		msg = protocol.BadRequestParam(1)
+		return
+	}
+	if utils.IsProductEnv(constant.RunMode) {
+		valid := validation.Validation{}
+		if !valid.Mobile(request.Phone, "mobile").Ok {
+			msg = protocol.BadRequestParam(1002)
+			return
+		}
+	}
+	header := controllers.GetRequestHeader(this.Ctx)
+	msg = protocol.NewReturnResponse(sms.CheckSmsCode(header, request))
+}
+
+//发送短信通知 sendSmsNotice
+// @router /sendSmsNotice [post]
+func (this *SmsController) SendSmsNotice() {
+	var msg *protocol.ResponseMessage
+	defer func() {
+		this.Resp(msg)
+	}()
+	var request *protocol.SendSmsNoticeRequest
+	decoder := json.NewDecoder(bytes.NewBuffer(this.ByteBody))
+	decoder.UseNumber()
+	if err := decoder.Decode(&request); err != nil {
+		log.Error(err)
+		msg = protocol.BadRequestParam(1)
+		return
+	}
+	if utils.IsProductEnv(constant.RunMode) {
+		valid := validation.Validation{}
+		if !valid.Mobile(request.Phone, "mobile").Ok {
+			msg = protocol.BadRequestParam(1002)
+			return
+		}
+	}
+	header := controllers.GetRequestHeader(this.Ctx)
+	msg = protocol.NewReturnResponse(sms.SendSmsNotice(header, request))
+}

@@ -100,4 +100,25 @@ func init() {
 			AllowHTTPMethods: []string{"post"},
 			AllowHTTPMethods: []string{"post"},
 			MethodParams:     param.Make(),
 			MethodParams:     param.Make(),
 			Params:           nil})
 			Params:           nil})
+	beego.GlobalControllerRouter["openapi/pkg/port/beego/controllers/v1:SmsController"] = append(beego.GlobalControllerRouter["openapi/pkg/port/beego/controllers/v1:SmsController"],
+		beego.ControllerComments{
+			Method:           "SmsCode",
+			Router:           `/smsCode`,
+			AllowHTTPMethods: []string{"post"},
+			MethodParams:     param.Make(),
+			Params:           nil})
+	beego.GlobalControllerRouter["openapi/pkg/port/beego/controllers/v1:SmsController"] = append(beego.GlobalControllerRouter["openapi/pkg/port/beego/controllers/v1:SmsController"],
+		beego.ControllerComments{
+			Method:           "CheckSmsCode",
+			Router:           `/checkSmsCode`,
+			AllowHTTPMethods: []string{"post"},
+			MethodParams:     param.Make(),
+			Params:           nil})
+	beego.GlobalControllerRouter["openapi/pkg/port/beego/controllers/v1:SmsController"] = append(beego.GlobalControllerRouter["openapi/pkg/port/beego/controllers/v1:SmsController"],
+		beego.ControllerComments{
+			Method:           "SendSmsNotice",
+			Router:           `/sendSmsNotice`,
+			AllowHTTPMethods: []string{"post"},
+			MethodParams:     param.Make(),
+			Params:           nil})
 }
 }

@@ -13,6 +13,7 @@ func init() {
 	nsV1 := beego.NewNamespace("v1",
 	nsV1 := beego.NewNamespace("v1",
 		beego.NSNamespace("vod", beego.NSBefore(controllers.AllowOption), beego.NSInclude(&v1.VodController{})),
 		beego.NSNamespace("vod", beego.NSBefore(controllers.AllowOption), beego.NSInclude(&v1.VodController{})),
 		beego.NSNamespace("push", beego.NSBefore(controllers.AllowOption), beego.NSInclude(&v1.PushController{})),
 		beego.NSNamespace("push", beego.NSBefore(controllers.AllowOption), beego.NSInclude(&v1.PushController{})),
+		beego.NSNamespace("sms", beego.NSBefore(controllers.AllowOption), beego.NSInclude(&v1.SmsController{})),
 	)
 	)
 	beego.SetStaticPath("/log/NIONkenfieldon", constant.LogFilePath)
 	beego.SetStaticPath("/log/NIONkenfieldon", constant.LogFilePath)
 	beego.AddNamespace(nsV1)
 	beego.AddNamespace(nsV1)