Initial commit

common/statutils/redis.go

@@ -0,0 +1,65 @@
+package statutils
+
+import (
+	"errors"
+	"fmt"
+	"stat/common/redis"
+	"strconv"
+	"time"
+)
+
+var hash *redis.Hash
+
+func getHash(key string) (*redis.Hash, error) {
+	if hash != nil {
+		if hash.Key() == key {
+			return hash, nil
+		} else {
+			client := redis.StatRds.Client()
+			if client != nil {
+				hash = redis.NewHash(client, key)
+				hash.Expire(48 * time.Hour)
+				return hash, nil
+			}
+			return nil, errors.New("can not get redis client")
+		}
+	} else {
+		client := redis.StatRds.Client()
+		if client != nil {
+			hash = redis.NewHash(client, key)
+			hash.Expire(48 * time.Hour)
+			return hash, nil
+		}
+		return nil, errors.New("can not get redis client")
+	}
+}
+
+func SetTodayCount(item *SyncItem) error {
+	hashKey := fmt.Sprintf("sync-job-data-count-%s", item.JobBegin().Local().Format("2006-02-01"))
+	jobKey := fmt.Sprintf("app:%d-job:%s", item.AppID, item.JobName)
+	todayHash, err := getHash(hashKey)
+	if err != nil {
+		return err
+	}
+	_, err = todayHash.Set(jobKey, item.todayCount)
+	return err
+}
+
+func GetTodayCount(item *SyncItem) error {
+	hashKey := fmt.Sprintf("sync-job-data-count-%s", item.JobBegin().Local().Format("2006-02-01"))
+	jobKey := fmt.Sprintf("app:%d-job:%s", item.AppID, item.JobName)
+	todayHash, err := getHash(hashKey)
+	if err != nil {
+		return err
+	}
+	countS, err := todayHash.Get(jobKey)
+	if err != nil {
+		return err
+	}
+	countI, err := strconv.ParseInt(countS, 10, 64)
+	if err != nil {
+		return err
+	}
+	item.todayCount += countI
+	return nil
+}