init

at-api/tests/traffic_test.go

@@ -0,0 +1,57 @@
+package test
+
+import (
+	api "anytunnel/at-api"
+	"encoding/json"
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/snail007/go-activerecord/mysql"
+)
+
+func TestTraffic(t *testing.T) {
+	raw := `{"1":{"positive":3231,"negative":5324},"2":{"negative":3997696,"positive":0}}`
+	dataMap := map[string]map[string]uint{}
+	err := json.Unmarshal([]byte(raw), &dataMap)
+	if err != nil {
+		fmt.Println("ERR:", err)
+		return
+	} else {
+		db := api.G.DB()
+		var rs *mysql.ResultSet
+		rs, err := db.Query(db.AR().
+			Select("user_id,tunnel_id").
+			From("tunnel").
+			Where(map[string]interface{}{
+				"tunnel_id": []string{"1", "2"},
+				"is_delete": 0,
+			}))
+		if err != nil {
+			fmt.Println("ERR:", err)
+			return
+		}
+		userIds := rs.MapValues("tunnel_id", "user_id")
+		data := []map[string]interface{}{}
+		for k, v := range dataMap {
+			userID, ok := userIds[k]
+			if !ok {
+				continue
+			}
+			data = append(data, map[string]interface{}{
+				"bytes +":     v["negative"] + v["positive"],
+				"user_id":     userID,
+				"month":       time.Now().Format("200601"),
+				"update_time": time.Now().Unix(),
+			})
+		}
+		//fmt.Println(data)
+		rs, err = db.Exec(db.AR().
+			UpdateBatch("traffic", data, []string{"user_id", "month"}))
+		if err != nil {
+			fmt.Println("ERR:", err)
+			return
+		}
+		fmt.Println("Affected Rows : ", rs.RowsAffected)
+	}
+}