From 7cef1846f712e7a4d80124c8ff680b2457c86dbf Mon Sep 17 00:00:00 2001
From: dengbiao <dengbiao@fnuoos.com>
Date: Tue, 21 May 2024 20:36:17 +0800
Subject: [PATCH] update

---
 ...ppy_orchard_reward_exchange_records_dao.go |  3 ++
 ...appy_orchard_reward_exchange_records_db.go | 28 +++++++++++++++++--
 2 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/src/dao/happy_orchard_reward_exchange_records_dao.go b/src/dao/happy_orchard_reward_exchange_records_dao.go
index 110d2d0..c9d12b9 100644
--- a/src/dao/happy_orchard_reward_exchange_records_dao.go
+++ b/src/dao/happy_orchard_reward_exchange_records_dao.go
@@ -2,6 +2,7 @@ package dao
 
 import (
 	"code.fnuoos.com/go_rely_warehouse/zyos_model.git/src/models"
+	"xorm.io/xorm"
 )
 
 type HappyOrchardRewardExchangeRecordsDao interface {
@@ -10,4 +11,6 @@ type HappyOrchardRewardExchangeRecordsDao interface {
 	GetHappyOrchardRewardExchangeRecordsByRecordId(id int) (m *models.HappyOrchardRewardExchangeRecords, err error)
 	GetHappyOrchardRewardExchangeRecordsByUidWithState(uid, state int) (m *models.HappyOrchardRewardExchangeRecords, err error)
 	GetHappyOrchardRewardExchangeRecordsByUidWithStateAndId(uid, id, state int) (m *models.HappyOrchardRewardExchangeRecords, err error)
+	InsertHappyOrchardRewardExchangeRecordsBySess(sess *xorm.Session, m *models.HappyOrchardRewardExchangeRecords) (int, error)
+	UpdateHappyOrchardRewardExchangeRecordsBySess(sess *xorm.Session, m *models.HappyOrchardRewardExchangeRecords, forceColums ...string) (int64, error)
 }
diff --git a/src/implement/happy_orchard_reward_exchange_records_db.go b/src/implement/happy_orchard_reward_exchange_records_db.go
index 9e13720..c42ae35 100644
--- a/src/implement/happy_orchard_reward_exchange_records_db.go
+++ b/src/implement/happy_orchard_reward_exchange_records_db.go
@@ -7,12 +7,36 @@ import (
 	"xorm.io/xorm"
 )
 
+func NewHappyOrchardRewardExchangeRecordsDb(engine *xorm.Engine) dao.HappyOrchardRewardExchangeRecordsDao {
+	return &HappyOrchardRewardExchangeRecordsDb{Db: engine}
+}
+
 type HappyOrchardRewardExchangeRecordsDb struct {
 	Db *xorm.Engine
 }
 
-func NewHappyOrchardRewardExchangeRecordsDb(engine *xorm.Engine) dao.HappyOrchardRewardExchangeRecordsDao {
-	return &HappyOrchardRewardExchangeRecordsDb{Db: engine}
+func (h HappyOrchardRewardExchangeRecordsDb) UpdateHappyOrchardRewardExchangeRecordsBySess(sess *xorm.Session, m *models.HappyOrchardRewardExchangeRecords, forceColums ...string) (int64, error) {
+	var (
+		affected int64
+		err      error
+	)
+	if forceColums != nil {
+		affected, err = sess.Where("id=?", m.Id).Cols(forceColums...).Update(m)
+	} else {
+		affected, err = sess.Where("id=?", m.Id).Update(m)
+	}
+	if err != nil {
+		return 0, err
+	}
+	return affected, nil
+}
+
+func (h HappyOrchardRewardExchangeRecordsDb) InsertHappyOrchardRewardExchangeRecordsBySess(sess *xorm.Session, m *models.HappyOrchardRewardExchangeRecords) (int, error) {
+	_, err := sess.InsertOne(m)
+	if err != nil {
+		return 0, err
+	}
+	return m.Id, nil
 }
 
 func (h HappyOrchardRewardExchangeRecordsDb) GetHappyOrchardRewardExchangeRecords(id int) (m *models.HappyOrchardRewardExchangeRecords, err error) {