CouponBackendGo/internal/logic/couponusers/couponUsers.go

package couponusers

import (
	"CouponBackendGo/api/v1/couponusers"
	"CouponBackendGo/internal/dao"
	"CouponBackendGo/internal/service"
	"context"
	"database/sql"
	"encoding/json"
	"errors"
	"fmt"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/xuri/excelize/v2"
	"mime/multipart"
	"strconv"
	"strings"
)

type sCouponUsers struct{}

func init() {
	service.RegisterCouponUsers(&sCouponUsers{})
}

// 根据优惠券id查看拥有优惠券的用户
func (s *sCouponUsers) GetCouponUsersByCondition(ctx context.Context, couponId, jwcode int, name string) (users []couponusers.GetCouponUsersRes, err error) {
	//从redis中获取数据但不返回
	value, _ := g.Redis().Get(ctx, fmt.Sprintf("%d CouponUsers", couponId))
	if value.String() != "" {
		err = json.Unmarshal(value.Bytes(), &users)
		if err != nil {
			return nil, errors.New("redis中数据解析失败")
		}
	} else {

		/*查询所有数据*/
		//在coupon_users中查询出对应的jwcode
		err = dao.CouponUsers.Ctx(ctx).Fields("jwcode").
			Where("coupon_id = ", couponId).Scan(&users)
		if err != nil {
			return nil, errors.New("根据卡券id搜索jwcode失败")
		}
		//根据jwcode查询用户信息
		for i, userInfo := range users {
			err = dao.MemberInfo.Ctx(ctx).Fields("jwcode", "name", "deptName", "shopName").
				Where("jwcode = ?", userInfo.Jwcode).Scan(&users[i])
			if err != nil {
				return nil, errors.New("根据jwcode搜索用户信息失败")
			}
		}
		/*查询所有数据*/

		//将数据存入redis
		UsersMarshal, _ := json.Marshal(users)
		_, err = g.Redis().Set(ctx, fmt.Sprintf("%d CouponUsers", couponId), UsersMarshal)
		if err != nil {
			return nil, errors.New("将数据存入redis失败")
		}

	}

	if jwcode != 0 && name == "" {
		var result []couponusers.GetCouponUsersRes
		for _, user := range users {
			if user.Jwcode == jwcode {
				result = append(result, user)
			}
		}
		return result, err
	} else if jwcode == 0 && name != "" {
		var result []couponusers.GetCouponUsersRes
		for _, user := range users {
			if strings.Contains(user.Name, name) {
				result = append(result, user)
			}
		}
		return result, err
	} else if jwcode != 0 && name != "" {
		var result []couponusers.GetCouponUsersRes
		for _, user := range users {
			if user.Jwcode == jwcode && strings.Contains(user.Name, name) {
				result = append(result, user)
			}
		}
		return result, err
	}

	return users, err
}

// 根据jwcode,优惠券id删除用户
func (s *sCouponUsers) DeleteCouponUserByJwcode(ctx context.Context, couponId, jwcode int) (result sql.Result, err error) {
	if jwcode == 0 {
		return nil, errors.New("jwcode不能为空")
	}
	if couponId == 0 {
		return nil, errors.New("couponId不能为空")
	}
	result, err = dao.CouponUsers.Ctx(ctx).Where("coupon_id = ?", couponId).Where("jwcode = ?", jwcode).Delete()
	if err != nil {
		return nil, errors.New("删除用户失败")
	}
	return result, err
}

// 通过excel导入精网号
func (s *sCouponUsers) InsertJwcodeByExcel(file multipart.File) (jwcodes []int, err error) {
	//打开文件并返回一个excelize.File对象,用于读取和操作Excel文件的内容
	f, err := excelize.OpenReader(file)
	if err != nil {
		return nil, errors.New("打开文件失败")
	}
	// 关闭文件
	defer func(f *excelize.File) {
		err := f.Close()
		if err != nil {
			// 处理关闭文件时的错误
			return
		}
	}(f)

	//读取所有工作表名称
	GetSheetMap := f.GetSheetMap()
	fmt.Println(GetSheetMap)
	if len(GetSheetMap) == 0 {
		return nil, errors.New("没有工作表")
	}
	// 读取第一个工作表
	sheetName := GetSheetMap[1]
	//rows, err := f.GetRows(f.GetSheetName(1))
	rows, err := f.GetRows(sheetName)
	if err != nil {
		return nil, err
	}

	// 将每行的第一列转换为int，并添加到切片中
	for i, row := range rows {
		//跳过第一行
		if i == 0 {
			continue
		}
		// 假设jwcode在每行的第一列
		/*参数校验*/
		//参数校验,检查每行是否有足够数据
		if len(row) == 0 {
			continue // 跳过空行
		}
		//参数校验,检查jwcode是否为非空字符串
		jwcodeStr := row[0]
		if jwcodeStr == "" {
			continue // 跳过空行
		}
		//将字符串转换为整数
		jwcode, err := strconv.Atoi(jwcodeStr)
		if err != nil {
			return nil, errors.New("参数转换失败")
		}
		fmt.Println(jwcode)
		jwcodes = append(jwcodes, jwcode)
		/*参数校验*/
	}
	return
}

// 根据精网号发放用户优惠券
func (s *sCouponUsers) InsertCouponUserByJwcode(ctx context.Context) {

	return
}