-
Notifications
You must be signed in to change notification settings - Fork 0
/
cmdset.go
106 lines (86 loc) · 2.98 KB
/
cmdset.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package redmos
import (
"encoding/json"
"fmt"
"time"
"github.com/redis/go-redis/v9"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
)
// Set 設值行為, 以索引字串與資料到主要/次要資料庫中儲存資料, 使用上有以下幾點須注意
// - 需要事先建立好資料結構, 並填寫到泛型類型T中, 請不要填入指標類型
// - 資料結構如果包含 Save 結構或是符合 Saver 介面, 會套用儲存判斷機制, 減少不必要的儲存操作
// - 資料結構的成員都需要設定好`bson:xxxxx`屬性
// - 執行前設定好 MajorEnable, MinorEnable
// - 執行前設定好 Meta, 這需要事先建立好與 Metaer 介面符合的元資料結構
// - 執行前設定好 Expire, 若不設置或是設置為0表示不過期
// - 執行前設定好 Key 並且不能為空字串
// - 執行前設定好 Data 並且不能為nil
type Set[T any] struct {
Behave // 行為物件
MajorEnable bool // 啟用主要資料庫
MinorEnable bool // 啟用次要資料庫
Meta Metaer // 元資料
Expire time.Duration // 過期時間, 若為0表示不過期
Key string // 索引值
Data *T // 資料物件
cmd *redis.StatusCmd // 命令結果
}
// Prepare 前置處理
func (this *Set[T]) Prepare() error {
if this.Meta == nil {
return fmt.Errorf("set prepare: meta nil")
} // if
if this.Key == "" {
return fmt.Errorf("set prepare: key empty")
} // if
if this.Data == nil {
return fmt.Errorf("set prepare: data nil")
} // if
if save, ok := any(this.Data).(Saver); ok && save.GetSave() == false {
return nil
} // if
if this.MajorEnable {
key := this.Meta.MajorKey(this.Key)
data, err := json.Marshal(this.Data)
if err != nil {
return fmt.Errorf("set prepare: %w: %v", err, this.Key)
} // if
this.cmd = this.Major().Set(this.Ctx(), key, data, this.Expire)
} // if
if this.MinorEnable {
if this.Meta.MinorTable() == "" {
return fmt.Errorf("set prepare: table empty")
} // if
if this.Meta.MinorField() == "" {
return fmt.Errorf("set prepare: field empty")
} // if
} // if
return nil
}
// Complete 完成處理
func (this *Set[T]) Complete() error {
if this.Meta == nil {
return fmt.Errorf("set complete: meta nil")
} // if
if save, ok := any(this.Data).(Saver); ok && save.GetSave() == false {
return nil
} // if
if this.MajorEnable {
data, err := this.cmd.Result()
if err != nil {
return fmt.Errorf("set complete: %w: %v", err, this.Key)
} // if
if data != RedisOk {
return fmt.Errorf("set complete: save to redis failed: %v", this.Key)
} // if
} // if
if this.MinorEnable {
table := this.Meta.MinorTable()
field := this.Meta.MinorField()
key := this.Meta.MinorKey(this.Key)
filter := bson.D{{Key: field, Value: key}}
this.Minor().Operate(table, mongo.NewReplaceOneModel().SetUpsert(true).SetFilter(filter).SetReplacement(this.Data))
} // if
return nil
}