package logic

import (
	"context"
	"database/sql"
	"gitee.com/xiyu-gozero/comm/timex"
	"gitee.com/xiyu-gozero/point-rpc/model"
	"gitee.com/xiyu-gozero/point-rpc/rpc/internal/svc"
	"gitee.com/xiyu-gozero/point-rpc/rpc/point"

	"github.com/zeromicro/go-zero/core/logx"
)

type SavePointLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewSavePointLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SavePointLogic {
	return &SavePointLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *SavePointLogic) SavePoint(in *point.SavePointReq) (*point.PointResp, error) {
	// todo: add your logic here and delete this line
	pe := model.Point{
		Id:      in.Id,
		Balance: in.Balance,
		Canuse:  in.Canuse,
		Locking: in.Locking,
		UserId:  in.UserId,
	}
	// add or update
	var err error
	var res sql.Result
	if in.Id == 0 {
		res, err = l.svcCtx.PointModel.Insert(l.ctx, &pe)
	} else {
		err = l.svcCtx.PointModel.Update(l.ctx, &pe)
	}
	if err != nil {
		return nil, err
	}
	if res != nil {
		id, err := res.LastInsertId()
		if err != nil {
			return nil, err
		}
		pe.Id = id
	}
	return &point.PointResp{
		Id:         pe.Id,
		Balance:    pe.Balance,
		Canuse:     pe.Canuse,
		Locking:    pe.Locking,
		UserId:     pe.UserId,
		CreateTime: timex.DateTimeFx(pe.CreateTime),
		UpdateTime: timex.DateTimeFx(pe.UpdateTime),
	}, nil
}