优化微信服务

service/wechatminiprogram/cgi-bin.token.monitor.go

@@ -9,24 +9,30 @@ var (
 	QdTypeRdb = "redis"
 )
 
-func (app *App) AuthGetAccessTokenMonitor(qdType string) error {
+func (app *App) AuthGetAccessTokenMonitor(qdType string) (string, error) {
-	result := app.GetCallBackIp()
+	switch qdType {
-	if len(result.Result.IpList) <= 0 {
+	case QdTypeDb:
-		switch qdType {
+		if app.Db == nil {
-		case QdTypeDb:
+			return "", errors.New("驱动没有初始化")
-			if app.Db == nil {
-				return errors.New("驱动没有初始化")
-			}
-			app.GetAccessTokenDb()
-		case QdTypeRdb:
-			if app.RDb == nil {
-				return errors.New("驱动没有初始化")
-			}
-			app.GetAccessTokenRDb()
-			return nil
-		default:
-			return errors.New("驱动类型不在范围内")
 		}
+		app.AccessToken = app.GetAccessTokenDb()
+		result := app.GetCallBackIp()
+		if len(result.Result.IpList) <= 0 {
+			return app.GetAccessTokenDb(), nil
+		}
+		return app.AccessToken, nil
+	case QdTypeRdb:
+		if app.RDb == nil {
+			return "", errors.New("驱动没有初始化")
+		}
+		app.AccessToken = app.GetAccessTokenRDb()
+		result := app.GetCallBackIp()
+		if len(result.Result.IpList) <= 0 {
+			return app.GetAccessTokenRDb(), nil
+		}
+		return app.AccessToken, nil
+	default:
+		return "", errors.New("驱动类型不在范围内")
 	}
-	return nil
+	return "", nil
 }

service/wechatoffice/cgi-bin.token.monitor.go

@@ -9,24 +9,30 @@ var (
 	QdTypeRdb = "redis"
 )
 
-func (app *App) AuthGetAccessTokenMonitor(qdType string) error {
+func (app *App) AuthGetAccessTokenMonitor(qdType string) (string, error) {
-	result := app.GetCallBackIp()
+	switch qdType {
-	if len(result.Result.IpList) <= 0 {
+	case QdTypeDb:
-		switch qdType {
+		if app.Db == nil {
-		case QdTypeDb:
+			return "", errors.New("驱动没有初始化")
-			if app.Db == nil {
-				return errors.New("驱动没有初始化")
-			}
-			app.GetAccessTokenDb()
-		case QdTypeRdb:
-			if app.RDb == nil {
-				return errors.New("驱动没有初始化")
-			}
-			app.GetAccessTokenRDb()
-			return nil
-		default:
-			return errors.New("驱动类型不在范围内")
 		}
+		app.AccessToken = app.GetAccessTokenDb()
+		result := app.GetCallBackIp()
+		if len(result.Result.IpList) <= 0 {
+			return app.GetAccessTokenDb(), nil
+		}
+		return app.AccessToken, nil
+	case QdTypeRdb:
+		if app.RDb == nil {
+			return "", errors.New("驱动没有初始化")
+		}
+		app.AccessToken = app.GetAccessTokenRDb()
+		result := app.GetCallBackIp()
+		if len(result.Result.IpList) <= 0 {
+			return app.GetAccessTokenRDb(), nil
+		}
+		return app.AccessToken, nil
+	default:
+		return "", errors.New("驱动类型不在范围内")
 	}
-	return nil
+	return "", nil
 }

service/wechatoffice/getticket.monitor.go

@@ -3,15 +3,21 @@ package wechatoffice
 import "errors"
 
 // AuthGetJsapiTicketMonitor 监控api_ticket
-func (app *App) AuthGetJsapiTicketMonitor(qdType string) error {
+func (app *App) AuthGetJsapiTicketMonitor(qdType string) (string, error) {
 	switch qdType {
 	case QdTypeDb:
-		app.GetJsapiTicketDb()
+		if app.Db == nil {
-		return nil
+			return "", errors.New("驱动没有初始化")
+		}
+		app.AccessToken = app.GetAccessTokenDb()
+		return app.GetJsapiTicketDb(), nil
 	case QdTypeRdb:
-		app.GetJsapiTicketRDb()
+		if app.RDb == nil {
-		return nil
+			return "", errors.New("驱动没有初始化")
+		}
+		app.AccessToken = app.GetAccessTokenRDb()
+		return app.GetJsapiTicketRDb(), nil
 	default:
-		return errors.New("驱动类型不在范围内")
+		return "", errors.New("驱动类型不在范围内")
 	}
 }

service/wechatunion/cgi-bin.token.monitor.go

@@ -9,24 +9,30 @@ var (
 	QdTypeRdb = "redis"
 )
 
-func (app *App) AuthGetAccessTokenMonitor(qdType string) error {
+func (app *App) AuthGetAccessTokenMonitor(qdType string) (string, error) {
-	result := app.GetCallBackIp()
+	switch qdType {
-	if len(result.Result.IpList) <= 0 {
+	case QdTypeDb:
-		switch qdType {
+		if app.Db == nil {
-		case QdTypeDb:
+			return "", errors.New("驱动没有初始化")
-			if app.Db == nil {
-				return errors.New("驱动没有初始化")
-			}
-			app.GetAccessTokenDb()
-		case QdTypeRdb:
-			if app.RDb == nil {
-				return errors.New("驱动没有初始化")
-			}
-			app.GetAccessTokenRDb()
-			return nil
-		default:
-			return errors.New("驱动类型不在范围内")
 		}
+		app.AccessToken = app.GetAccessTokenDb()
+		result := app.GetCallBackIp()
+		if len(result.Result.IpList) <= 0 {
+			return app.GetAccessTokenDb(), nil
+		}
+		return app.AccessToken, nil
+	case QdTypeRdb:
+		if app.RDb == nil {
+			return "", errors.New("驱动没有初始化")
+		}
+		app.AccessToken = app.GetAccessTokenRDb()
+		result := app.GetCallBackIp()
+		if len(result.Result.IpList) <= 0 {
+			return app.GetAccessTokenRDb(), nil
+		}
+		return app.AccessToken, nil
+	default:
+		return "", errors.New("驱动类型不在范围内")
 	}
-	return nil
+	return "", nil
 }