youngee_mc_api/app/service/youngee_talent_service/talent_ks_auth.go

package youngee_talent_service

import (
	"context"
	"fmt"
	"github.com/chromedp/chromedp"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/net/ghttp"
	"github.com/lin-jim-leon/kuaishou/open/merchant"
	"log"
	"time"
	"youngmini_server/app/model/youngee_talent_model"
	"youngmini_server/app/utils"
)

func GetQrcode(r *ghttp.Request) *TalentHttpResult {
	//达人id获取
	tid, err := utils.SessionTalentInfo.GetTalentIdFromSession(r)
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: "Get talent id failed"}
	}
	println("here")
	// 记录开始时间
	startTime := time.Now()
	// 创建一个新的上下文
	ctx, cancel := chromedp.NewContext(context.Background())
	// 构建包含 tid 的 urlstr
	urlstr := fmt.Sprintf("https://open.kuaishou.com/oauth2/connect?state=%s&app_id=ks651333097154138217&redirect_uri=https://younggee.com/kuaishouauth&scope=merchant_distribution,merchant_refund,merchant_item,merchant_order,user_info,merchant_servicemarket,merchant_user,merchant_logistics&response_type=code", tid)
	// 执行任务
	//var buf []byte
	var text string
	var ok bool
	//浏览器启动！
	err = chromedp.Run(ctx,
		//打开授权网址
		chromedp.Navigate(urlstr),
		//等待元素可见
		//chromedp.WaitVisible(`#semiTabPanelqr > div > div > div.qr-container > img`, chromedp.ByQuery),
		// 获取元素的src属性值
		//chromedp.AttributeValue(`#semiTabPanelqr > div > div > div.qr-container > img`, "src", &text, &ok), //抖音
		chromedp.AttributeValue(`body > div > div > div > div.qr-code-main > img:nth-child(1)`, "src", &text, &ok), //快手

		//chromedp.AttributeValue(`#app > div > section > header > div > div.head-menu > ul > li:nth-child(1) > img`, "src", &text, &ok),
	)
	if err != nil {
		log.Fatal(err)
	}
	println("text----->", text)
	println("ok----->", ok)
	println("url----->", urlstr)
	fmt.Println("代码运行时间:", time.Since(startTime))
	//r.Response.WriteJson(text)

	// 在300秒后取消上下文
	time.AfterFunc(300*time.Second, func() {
		cancel()
	})

	return &TalentHttpResult{Code: 0, Msg: "success", Data: text}
}

func CheckAccount(r *ghttp.Request) *TalentHttpResult {
	fmt.Println("into check")
	//达人id获取
	tid, err := utils.SessionTalentInfo.GetTalentIdFromSession(r)
	if err != nil {
		return &TalentHttpResult{Code: -1, Msg: "Get talent id failed"}
	}
	userInfo := &youngee_talent_model.KuaishouUserInfo{}
	err = g.DB().Model(userInfo).Where("talent_id", tid).Scan(&userInfo)
	fmt.Println("userInfo*********", userInfo)
	if err != nil {
		// 查询失败(未找到结果)，返回失败结果
		return &TalentHttpResult{Code: -1, Msg: "Query failed", Data: nil}
	}
	// 查询成功，返回成功结果和数据
	return &TalentHttpResult{Code: 0, Msg: "success", Data: userInfo}
}

func QuerySalesFor30Days(r *ghttp.Request) *TalentHttpResult {
	fmt.Println("into querySalesFor30Days")
	ClientKey := "ks651333097154138217"
	//ClientSecret := "dBt0rVRhTpUqcrOYGGpv0A"
	SignSecret := "bf6393dce0a2b669ee348bebb837b0da"
	tid, _ := utils.SessionTalentInfo.GetTalentIdFromSession(r)
	// 创建一个 KuaishouUserInfo 结构体的实例
	userInfo := &youngee_talent_model.KuaishouUserInfo{}

	// 查询数据库中的 access_token 字段
	// 使用 g.DB() 获取数据库连接，并执行查询
	key, err := g.DB().Model(userInfo).Fields("access_token").Where("talent_id = ?", tid).Value()
	AccessToken := key.String()
	if err != nil {
		// 处理错误
		fmt.Println("Error querying access_token:", err)
	}

	// 输出查询到的 access_token 值
	fmt.Println("Access Token:", AccessToken)

	//循环使用接口
	//
	// 获取当前时间的时间戳（毫秒）
	currentTime := time.Now().UnixNano() / int64(time.Millisecond)
	// 30天前的时间戳（毫秒）
	beginTime30DaysAgo := currentTime - 30*24*60*60*1000
	// 定义每段查询的天数
	intervalDays := 7
	intervalMillis := int64(intervalDays * 24 * 60 * 60 * 1000)

	// 初始化 beginTime 和 endTime
	beginTime := beginTime30DaysAgo
	endTime := beginTime + intervalMillis
	saleNum := 0 //30天总销量
	// 循环查询，先处理四个7天的时间段
	for i := 0; i < 4; i++ {
		// 调整 endTime，确保不会超过当前时间
		if endTime > currentTime {
			endTime = currentTime
		}
		saleNum += GetSaleNumByDayInterval(ClientKey, SignSecret, AccessToken, beginTime, endTime)
		// 更新时间段
		beginTime = endTime
		endTime = beginTime + intervalMillis
	}

	// 最后处理剩余的2天时间段
	endTime = currentTime
	saleNum += GetSaleNumByDayInterval(ClientKey, SignSecret, AccessToken, beginTime, endTime)
	// 查询成功，返回成功结果和数据
	return &TalentHttpResult{Code: 0, Msg: "获取30天销售量成功", Data: saleNum}
}

func GetSaleNumByDayInterval(ClientKey string, SignSecret string, AccessToken string, beginTime int64, endTime int64) int {
	pageSize := 100
	totalSaleNum := 0

	// 定义要查询的订单状态
	statuses := []int{30, 50, 60}

	// 遍历所有订单状态并调用 Corderlist 函数
	for _, status := range statuses {
		// 初始化 pcursor
		pcursor := ""

		for {
			// 调用 Corderlist 函数获取订单列表
			response, err := merchant.Corderlist(ClientKey, SignSecret, AccessToken, status, pageSize, beginTime, endTime, pcursor)
			fmt.Println("response********", response)
			if err != nil {
				fmt.Printf("Error calling Corderlist: %v\n", err)
				break
			}

			// 检查响应代码
			if response.Code != "1" {
				fmt.Printf("Corderlist response error: %s\n", response.Msg)
				break
			}

			// 累加订单中商品的数量
			for _, order := range response.Data.OrderViews {
				for _, product := range order.CPSOrderProductViews {
					totalSaleNum += product.Num
				}
			}

			// 检查分页指针 pcursor
			//100个订单以内的情况
			if response.Data.Cursor == "nomore" {
				break
			}
			//大于等于100个订单
			// 更新 pcursor 以获取下一页数据
			pcursor = response.Data.Cursor

			// 处理分页后的数据
			// 如果 pcursor 不是 "nomore"，我们需要累加最后一条订单的数量
			if len(response.Data.OrderViews) > 0 {
				lastOrder := response.Data.OrderViews[len(response.Data.OrderViews)-1]
				if len(lastOrder.CPSOrderProductViews) > 0 {
					lastProduct := lastOrder.CPSOrderProductViews[len(lastOrder.CPSOrderProductViews)-1]
					totalSaleNum += lastProduct.Num
				}
			}
		}
	}
	return totalSaleNum
}