活动短链接生成任务；

controllers/短信模块临时备注

@@ -1,3 +0,0 @@
-修改个人信息
-权限管理中添加管理员
-客户管理

jobcron/activity_short_url_job.go

@@ -0,0 +1,128 @@
+package jobcron
+
+import (
+	"SCRM/service"
+	"SCRM/utils"
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+
+	"github.com/astaxie/beego"
+	"github.com/robfig/cron"
+)
+
+// cron表达式 https://www.cnblogs.com/zuxingyu/p/6023919.html
+var activityShortUrlCronJob *cron.Cron
+
+func init() {
+	activityShortUrlCronJob = cron.New()
+	spec := "0 0 0 * * ?" // 每天晚上12点整检查活动短链接库存充不充足
+	activityShortUrlCronJob.AddFunc(spec, func() {
+		CreateActivityShortURLIfNeeded()
+	})
+}
+
+type CreateShortURLServiceMaxId struct {
+	Value int
+}
+
+func CreateActivityShortURLIfNeeded() {
+	var maxActivityId CreateShortURLServiceMaxId
+	service.PatientReadDB().Raw("SELECT MAX(id) AS value FROM sgj_patient_activity").Scan(&maxActivityId)
+	utils.TraceLog("当前最大的活动 ID 为：%v", maxActivityId)
+	var maxShortURLId CreateShortURLServiceMaxId
+	service.PatientReadDB().Raw("SELECT MAX(id) AS value FROM sgj_patient_activity_wx_share").Scan(&maxShortURLId)
+	utils.TraceLog("当前最大的活动短链接 ID 为：%v", maxShortURLId)
+
+	if maxShortURLId.Value-maxActivityId.Value <= 300 {
+		utils.TraceLog("即将生成更多活动短链接")
+		go CreateMoreActivityShortURL(maxShortURLId.Value)
+	} else {
+		utils.TraceLog("活动短链接充足，无需生成")
+	}
+}
+
+// 新浪短链接服务示例 https://www.cnblogs.com/azure/archive/2012/08/29/WeiboAPI.html
+func CreateMoreActivityShortURL(lastShortURLID int) {
+	var maxUrlCountPerRequest int = 20
+	createCount, _ := beego.AppConfig.Int("create_short_url_count_every_time")
+	pageCount := createCount / maxUrlCountPerRequest
+	tx := service.PatientWriteDB().Begin()
+	for i := 0; i < pageCount; i++ {
+		longURLParams := make([]string, 0, maxUrlCountPerRequest)
+		for longURLIndex := 1; longURLIndex < maxUrlCountPerRequest+1; longURLIndex++ {
+			id := lastShortURLID + i*maxUrlCountPerRequest + longURLIndex
+			param := fmt.Sprintf("url_long=%v", fmt.Sprintf("%v/weixin/activity/%v", beego.AppConfig.String("weixinwebdomain"), id))
+			longURLParams = append(longURLParams, param)
+		}
+
+		urlStr := fmt.Sprintf("http://api.weibo.com/2/short_url/shorten.json?source=2849184197&%v", strings.Join(longURLParams, "&"))
+		u, _ := url.Parse(urlStr)
+		u.RawQuery = u.Query().Encode()
+		// fmt.Println(u.String())
+
+		resp, requestErr := http.Get(u.String())
+		if requestErr != nil {
+			utils.ErrorLog("http.Get error: %v", requestErr)
+			continue
+		}
+		result, readErr := ioutil.ReadAll(resp.Body)
+		resp.Body.Close()
+		if readErr != nil {
+			utils.ErrorLog("ioutil.ReadAll error: %v", readErr)
+			continue
+		}
+		// fmt.Println(string(result))
+		var respJSON map[string]interface{}
+		if err := json.Unmarshal([]byte(string(result)), &respJSON); err != nil {
+			utils.ErrorLog("json.Unmarshal error: %v\n", err)
+			continue
+		}
+		// fmt.Printf("resp: %v\n", respJSON)
+		if respJSON["error"] != nil {
+			utils.ErrorLog("request error: %v\n", respJSON)
+			continue
+		}
+
+		urlJSONs := respJSON["urls"].([]interface{})
+		now := time.Now().Unix()
+		valueStrs := make([]string, 0, len(urlJSONs))
+		values := make([]interface{}, 0, len(urlJSONs)*9)
+
+		for index, url := range urlJSONs {
+			urlJSON := url.(map[string]interface{})
+			shortURL := urlJSON["url_short"].(string)
+			id := lastShortURLID + i*maxUrlCountPerRequest + (index + 1)
+			// CreateActivityWxShareWithShortURL(id, shortURL)
+
+			// (`id`,`activity_id`,`title`,`subtitle`,`image`,`status`,`ctime`,`mtime`,`short_url`)
+			valueStrs = append(valueStrs, "(?, ?, ?, ?, ?, ?, ?, ?, ?)")
+			values = append(values, id)
+			values = append(values, id)
+			values = append(values, "")
+			values = append(values, "")
+			values = append(values, "")
+			values = append(values, 1)
+			values = append(values, now)
+			values = append(values, now)
+			values = append(values, shortURL)
+		}
+
+		sql := fmt.Sprintf("INSERT INTO sgj_patient_activity_wx_share (id, activity_id, title, subtitle, image, status, ctime, mtime, short_url) VALUES %v;", strings.Join(valueStrs, ", "))
+		if err := tx.Exec(sql, values...).Error; err != nil {
+			utils.ErrorLog("插入活动短链接失败: %v", err)
+			tx.Rollback()
+			break
+		}
+	}
+	tx.Commit()
+	utils.TraceLog("完成生成更多活动短链接")
+}
+
+func BeginCreateActivityShortURLCronJob() {
+	activityShortUrlCronJob.Start()
+}

jobcron/qiniu_job.go

@@ -0,0 +1,42 @@
+package jobcron
+
+import (
+	"SCRM/service"
+	"fmt"
+
+	"github.com/astaxie/beego"
+	"github.com/qiniu/api.v7/auth/qbox"
+	"github.com/qiniu/api.v7/storage"
+	"github.com/robfig/cron"
+)
+
+var qnTokenCronJob *cron.Cron
+
+func init() {
+	qnTokenCronJob = cron.New()
+	qnTokenCronJob.AddFunc("@every 1h50m", func() {
+		go RequestQNToken()
+	})
+}
+
+func RequestQNToken() {
+	accessKey := beego.AppConfig.String("qiniu_accesskey")
+	secretKey := beego.AppConfig.String("qiniu_secretkey")
+	bucket := beego.AppConfig.String("qiniu_bucket")
+	putPolicy := storage.PutPolicy{
+		Scope:      bucket,
+		ReturnBody: `{"url":"$(key)","hash":"$(etag)","state":"SUCCESS"}`,
+	}
+	putPolicy.Expires = 7200
+	mac := qbox.NewMac(accessKey, secretKey)
+	upToken := putPolicy.UploadToken(mac)
+	fmt.Println("new qiniu token: ", upToken)
+
+	redisClient := service.RedisClient()
+	defer redisClient.Close()
+	redisClient.Set("qn_token", upToken, 0)
+}
+
+func BeginRerequestQNTokenCronJob() {
+	qnTokenCronJob.Start()
+}

main.go

@@ -5,8 +5,6 @@ import (
 	_ "SCRM/routers"
 	"SCRM/service"
 
-	// "SCRM/jobcron"
-
 	"github.com/astaxie/beego"
 )
 
@@ -22,8 +20,15 @@ func main() {
 		// jobcron.RequestMpWechatAccessToken()
 		// jobcron.ReqJsapiTicket()
 	}()
+
 	jobcron.StartOpenWechatCron()
 	jobcron.BeginSendUnsendSMSCronJob()
 
+	jobcron.CreateActivityShortURLIfNeeded()
+	jobcron.BeginCreateActivityShortURLCronJob()
+
+	go jobcron.RequestQNToken()
+	jobcron.BeginRerequestQNTokenCronJob()
+
 	beego.Run()
 }

service/sms_service/sms_service.go

@@ -65,7 +65,7 @@ func SMSOrgSendWithCustomContent(orgID int64, originContent string, mobiles []st
 	if getFreeLimitErr != nil {
 		return getFreeLimitErr
 	}
-	if willConsumeLimit+freeLimit.UsedCount <= freeLimit.TotalCount {
+	if willConsumeLimit+freeLimit.UsedCount > freeLimit.TotalCount {
 		return SMSErrorFreeLimitInsufficient
 	}