fusenapi/model/gmodel/fsguestmodel.go

package gmodel

import (
	"context"
	"fusenapi/utils/auth"
	"time"

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

type FsGuest struct {
	GuestId        int64   `gorm:"primary_key" json:"guest_id"` // 游客ID
	AuthKey        *string `gorm:"" json:"auth_key"`            // jwt token
	Status         *int64  `gorm:"" json:"status"`              // 1正常 0不正常
	IsDel          *int64  `gorm:"" json:"is_del"`              // 是否删除 1删除
	CreatedAt      *int64  `gorm:"" json:"created_at"`          // 添加时间
	UpdatedAt      *int64  `gorm:"" json:"updated_at"`          // 更新时间
	IsOpenRender   *int64  `gorm:"" json:"is_open_render"`      // 是否打开个性化渲染（1：开启，0：关闭）
	IsThousandFace *int64  `gorm:"" json:"is_thousand_face"`    // 是否已经存在千人千面（1：存在，0：不存在）
	IsLowRendering *int64  `gorm:"" json:"is_low_rendering"`    // 是否开启低渲染模型渲染
	IsRemoveBg     *int64  `gorm:"" json:"is_remove_bg"`        // 用户上传logo是否去除背景
}
type FsGuestModel struct{ db *gorm.DB }

func NewFsGuestModel(db *gorm.DB) *FsGuestModel { return &FsGuestModel{db} }

func (m *FsGuestModel) GenerateGuestID(ctx context.Context, AccessSecret *string) (authKey string, err error) {

	err = m.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
		now := time.Now().Unix()
		var record = &FsGuest{}
		tx.Create(record)

		authKey, err = auth.GenerateJwtToken(AccessSecret, now, 31536000, 0, record.GuestId)
		if err != nil {
			logx.Error(err)
			tx.Rollback()
			return err
		}
		record.AuthKey = &authKey
		record.CreatedAt = &now
		tx.Updates(record)
		return nil
	})

	if err != nil {
		return "", err
	}

	return authKey, nil
}