From ee63b9c19d1a41e01a7055e320e99bfffb3a78e6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8B=8F=E5=B0=B9=E5=B2=9A?= <770236076@qq.com>
Date: Thu, 26 Mar 2020 15:34:36 +0800
Subject: [PATCH] =?UTF-8?q?=E9=93=B6=E8=B1=B9cookie=E8=BF=87=E6=9C=9F?=
 =?UTF-8?q?=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 platformapi/yinbaoapi/store_page.go |  1 +
 platformapi/yinbaoapi/yinbaoapi.go  | 30 +++++++++++++++++++++--------
 2 files changed, 23 insertions(+), 8 deletions(-)

diff --git a/platformapi/yinbaoapi/store_page.go b/platformapi/yinbaoapi/store_page.go
index d4f601b7..8f0bc328 100644
--- a/platformapi/yinbaoapi/store_page.go
+++ b/platformapi/yinbaoapi/store_page.go
@@ -18,6 +18,7 @@ const (
 
 func (a *API) AccessStorePage(action string, bizParams map[string]interface{}) (retVal map[string]interface{}, err error) {
 	fullURL := utils.GenerateGetURL(pageUrl, action, nil)
+	a.addPageCount()
 	// result, _ := json.MarshalIndent(bizParams, "", " ")
 	err = platformapi.AccessPlatformAPIWithRetry(a.client,
 		func() *http.Request {
diff --git a/platformapi/yinbaoapi/yinbaoapi.go b/platformapi/yinbaoapi/yinbaoapi.go
index ff3ae30f..5d706ad4 100644
--- a/platformapi/yinbaoapi/yinbaoapi.go
+++ b/platformapi/yinbaoapi/yinbaoapi.go
@@ -34,10 +34,23 @@ const (
 type API struct {
 	platformapi.APICookie
 
-	appKey string
-	appID  string
-	client *http.Client
-	config *platformapi.APIConfig
+	appKey    string
+	appID     string
+	pageCount int
+	client    *http.Client
+	config    *platformapi.APIConfig
+}
+
+func (a *API) setPageCount(pc int) {
+	a.pageCount = pc
+}
+
+func (a *API) GetPageCount() int {
+	return a.pageCount
+}
+
+func (a *API) addPageCount() {
+	a.setPageCount(a.GetPageCount() + 1)
 }
 
 func New(appKey, appID string, config ...*platformapi.APIConfig) *API {
@@ -46,10 +59,11 @@ func New(appKey, appID string, config ...*platformapi.APIConfig) *API {
 		curConfig = *config[0]
 	}
 	return &API{
-		appKey: appKey,
-		appID:  appID,
-		client: &http.Client{Timeout: curConfig.ClientTimeout},
-		config: &curConfig,
+		appKey:    appKey,
+		appID:     appID,
+		pageCount: 1,
+		client:    &http.Client{Timeout: curConfig.ClientTimeout},
+		config:    &curConfig,
 	}
 }