youtu_server/game_open_api/model/app_user_model.go

package model

import (
	"context"
	"fmt"
	"github.com/zeromicro/go-zero/core/stores/cache"
	"github.com/zeromicro/go-zero/core/stores/sqlc"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

var _ AppUserModel = (*customAppUserModel)(nil)

type (
	// AppUserModel is an interface to be customized, add more methods here,
	// and implement the added methods in customAppUserModel.
	AppUserModel interface {
		appUserModel
		FindOrCreate(ctx context.Context, data *AppUser) (err error)
	}

	customAppUserModel struct {
		*defaultAppUserModel
	}
)

const cacheEcpmAppUserPrefix = "cache:ecpm:appUser:"

// NewAppUserModel returns a model for the database table.
func NewAppUserModel(conn sqlx.SqlConn, c cache.CacheConf, opts ...cache.Option) AppUserModel {
	return &customAppUserModel{
		defaultAppUserModel: newAppUserModel(conn, c, opts...),
	}
}

func (m *customAppUserModel) FindOrCreate(ctx context.Context, data *AppUser) (err error) {
	ecpmAppUserIdKey := fmt.Sprintf("%sappId:%d:openid:%s", cacheEcpmAppUserPrefix, data.AppAccountId, data.Openid)
	var resp AppUser
	err = m.QueryRowCtx(ctx, &resp, ecpmAppUserIdKey, func(ctx context.Context, conn sqlx.SqlConn, v any) error {
		query := "select * from app_user where `openid` = ? and `app_account_id`= ? limit 1"
		return conn.QueryRowCtx(ctx, v, query, data.Openid, data.AppAccountId)
	})
	switch err {
	case nil:
		*data = resp
		return nil
	case sqlc.ErrNotFound:
		res, err := m.Insert(ctx, data)
		if err != nil {
			return err
		}
		id, err := res.LastInsertId()
		if err != nil {
			return err
		}
		data.Id = uint64(id)
		return nil
	default:
		return err
	}
}