支持test mode

adslib/config.go

@@ -26,6 +26,7 @@ type SvrConf struct {
 	LogPath            string                                `toml:"log_path"`
 	RedisHost          string                                `toml:"redis_host"`
 	RedisPassword      string                                `toml:"redis_password"`
+	TestMode           bool                                  `toml:"test_mode"`
 }
 type FlogControlItem struct {
 	Percent  int `toml:"percent"`

adslib/device/device.go

@@ -6,6 +6,7 @@ import (
 	"github.com/gomodule/redigo/redis"
 	log "github.com/sirupsen/logrus"
 	"math/rand"
+	"miads/adslib"
 	"miads/adslib/ads_redis"
 	"strconv"
 	"strings"
@@ -24,6 +25,10 @@ func SetAdsTagLog(adv string, reqSource string, key string, cityCode int) {
 }
 
 func setTagLog(key string) {
+	if adslib.GetConf().TestMode {
+		return
+	}
+
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()
 
@@ -35,6 +40,10 @@ func setTagLog(key string) {
 
 // 存储ip上一次请求时间
 func SetIpReqTime(ip string, reqTime int64) {
+	if adslib.GetConf().TestMode {
+		return
+	}
+
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()
 
@@ -62,6 +71,10 @@ func GetIpReqTime(ip string) (int64, error) {
 }
 
 func SetMiDeviceConf(key string, d DeviceConf) error {
+	if adslib.GetConf().TestMode {
+		return
+	}
+
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()
 
@@ -108,6 +121,10 @@ func GetMiDeviceConf(ip string, randNum int, specialDeviceNum int) (*DeviceConf,
 
 // 设置每天的city_code的值
 func SetDailyFakeDeviceConfByCityCode(cityCode int, d DeviceConf, dateInt int) error {
+	if adslib.GetConf().TestMode {
+		return nil
+	}
+
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()
 

adslib/graylog/graylog.go

@@ -9,7 +9,8 @@ import (
 	"net/url"
 	"time"
 )
-var grayLog  *gelf.Gelf
+
+var grayLog *gelf.Gelf
 
 func init() {
 	grayLog = gelf.New(gelf.Config{
@@ -25,18 +26,22 @@ func Log(log interface{}) {
 	logStr, _ := json.Marshal(log)
 	fmt.Printf("graylog: %s\n", logStr)
 	grayLog.Log(string(logStr))
+
+	if adslib.GetConf().TestMode {
+		return
+	}
 }
 
 func LogApi(shortMsg string, desc string, extraDesc string, reqSource string) {
 	grayLogData := struct {
-		ShortMessage string `json:"short_message"`
-		ReqSource string `json:"req_source"`
-		AdsDetail string `json:"ads_detail"`
+		ShortMessage   string `json:"short_message"`
+		ReqSource      string `json:"req_source"`
+		AdsDetail      string `json:"ads_detail"`
 		AdsExtraDetail string `json:"ads_extra_detail"`
 	}{
-		ShortMessage: shortMsg,
-		ReqSource:    reqSource,
-		AdsDetail: desc,
+		ShortMessage:   shortMsg,
+		ReqSource:      reqSource,
+		AdsDetail:      desc,
 		AdsExtraDetail: extraDesc,
 	}
 	Log(grayLogData)
@@ -50,9 +55,9 @@ func ReportGrayLog(req adslib.Request, dsp utils.DspParam, uid int, jsonFlag int
 	gelfData["_device_id"] = req.DeviceId
 	gelfData["_ip"] = dsp.Ip
 	gelfData["_uri"] = req.QueryString
-	gelfData["_province"]=dsp.Province
-	gelfData["_city"]=dsp.City
-	gelfData["_city_code"]=dsp.DspCityCode
+	gelfData["_province"] = dsp.Province
+	gelfData["_city"] = dsp.City
+	gelfData["_city_code"] = dsp.DspCityCode
 
 	var reqData url.Values
 	if jsonFlag == 0 {
@@ -67,7 +72,7 @@ func ReportGrayLog(req adslib.Request, dsp utils.DspParam, uid int, jsonFlag int
 		if k == "ip" {
 			continue
 		}
-		gelfData["_" + k] = v
+		gelfData["_"+k] = v
 	}
 
 	gelfData["_request"] = reqData.Encode()

adslib/redis_data/redis_data.go

@@ -158,6 +158,10 @@ func GetFlowPercentDuration(advertiser string, action string) *adslib.FlogContro
 
 // 设置回调回来的广告动作的次数
 func SetDistributeActionNum(advertiser string, action string, bannerid int) {
+	if adslib.GetConf().TestMode {
+		return
+	}
+
 	if advertiser == "" {
 		return
 	}
@@ -196,6 +200,9 @@ func GetDeviceIpReport(deviceId string, ip string) (bool, error) {
 
 // 设置设备+ip上报的时间点
 func SetDeviceIpReport(deviceid string, ip string) {
+	if adslib.GetConf().TestMode {
+		return
+	}
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()
 
@@ -205,6 +212,9 @@ func SetDeviceIpReport(deviceid string, ip string) {
 
 // 设置总请求到广告的次数
 func SetAdsRequestNum(advertiser string, bannerId int) {
+	if adslib.GetConf().TestMode {
+		return
+	}
 	if advertiser == "" {
 		return
 	}
@@ -220,6 +230,9 @@ func SetAdsRequestNum(advertiser string, bannerId int) {
 }
 
 func SetAdsRealRequestNum(advertiser string) {
+	if adslib.GetConf().TestMode {
+		return
+	}
 	if advertiser == "" {
 		return
 	}
@@ -260,6 +273,9 @@ func GetMinScriptOrderByAdv(advertiser string) (int64, error) {
 
 // 设置来源的曝光次数
 func SetReqSourceView(adv string, reqSource string, source string) {
+	if adslib.GetConf().TestMode {
+		return
+	}
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()
 
@@ -309,6 +325,10 @@ func GetFinishedDispatchCount(orderId int64, dispatchType string, curTime time.T
 // 增加已经投放的量
 // dispatchType: "click", "show"
 func IncrFinishedDispatchCount(orderId int64, dispatchType string, incrCnt int, curTime time.Time) (int, error) {
+	if adslib.GetConf().TestMode {
+		return 0, nil
+	}
+
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()
 
@@ -347,6 +367,10 @@ func GetPreMinuteFinishedDispatchCount(orderId int64, dispatchType string, curTi
 // 增加当前分钟已经投放的量
 // dispatchType: "click", "show"
 func IncrPreMinuteFinishedDispatchCount(orderId int64, dispatchType string, incrCnt int, curTime time.Time) (int, error) {
+	if adslib.GetConf().TestMode {
+		return 0, nil
+	}
+
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()
 
@@ -363,6 +387,9 @@ func IncrPreMinuteFinishedDispatchCount(orderId int64, dispatchType string, incr
 // 设置总计划投放数量
 // dispatchType: "click", "show"
 func SetPlanDispatchCount(orderId int64, dispatchType string, cnt int, curTime time.Time) {
+	if adslib.GetConf().TestMode {
+		return
+	}
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()
 
@@ -373,6 +400,9 @@ func SetPlanDispatchCount(orderId int64, dispatchType string, cnt int, curTime t
 // 设置订单计划投放数量
 // dispatchType: "click", "show"
 func SetOrderPlanDispatchCount(orderId int64, dispatchType string, cnt int, curTime time.Time) {
+	if adslib.GetConf().TestMode {
+		return
+	}
 	conn := ads_redis.RedisConn.Get()
 	defer conn.Close()