From eded5d0f8a4eb980ff7869c908e9cca05794bd3c Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Fri, 22 Nov 2024 14:15:21 +0800
Subject: [PATCH] 1

---
 server/wheelserver/model/buff.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/server/wheelserver/model/buff.go b/server/wheelserver/model/buff.go
index 911ff5b0..24295283 100644
--- a/server/wheelserver/model/buff.go
+++ b/server/wheelserver/model/buff.go
@@ -2,7 +2,9 @@ package model
 
 import (
 	"f5"
+	"errors"
 	"main/constant"
+	"gorm.io/gorm"
 )
 
 type Buff struct {
@@ -25,3 +27,21 @@ func (this *Buff) Create() error {
 	}
 	return nil
 }
+
+func (this *Buff) UpdateFields(fields []string) error {
+	if result := f5.GetApp().GetOrmDb(constant.WHEEL_DB).Model(this).Select(
+		fields).Updates(this); result.Error != nil {
+		return result.Error
+	}
+	return nil
+}
+
+func (this *Buff) FindByBuffUniId(accountId string, buffUniId int64) (error, bool) {
+	if result := f5.GetApp().GetOrmDb(constant.WHEEL_DB).Table(this.TableName()).Take(
+		this, "account_id = ? and idx = ?", accountId, buffUniId); result.Error != nil &&
+		!errors.Is(result.Error, gorm.ErrRecordNotFound) {
+		return result.Error, false
+	} else {
+		return nil, result.RowsAffected > 0
+	}
+}