mirror of https://github.com/rocboss/paopao-ce
Michael Li
1 year ago
parent
a14921448e
commit
bf4026c5af
@ -0,0 +1,30 @@
|
||||
// Copyright 2023 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package conf
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/alimy/tryst/cache"
|
||||
)
|
||||
|
||||
const (
|
||||
_defaultKeyPoolSize = 128
|
||||
)
|
||||
|
||||
// 以下包含一些在cache中会用到的池化后的key
|
||||
var (
|
||||
KeyUnreadMsg cache.KeyPool[int64]
|
||||
)
|
||||
|
||||
func initCacheKeyPool() {
|
||||
poolSize := _defaultKeyPoolSize
|
||||
if poolSize < CacheSetting.KeyPoolSize {
|
||||
poolSize = CacheSetting.KeyPoolSize
|
||||
}
|
||||
KeyUnreadMsg = cache.MustKeyPool[int64](poolSize, func(key int64) string {
|
||||
return fmt.Sprintf("paopao:unreadmsg:%d", key)
|
||||
})
|
||||
}
|
||||
@ -0,0 +1,66 @@
|
||||
// Copyright 2023 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package cache
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/Masterminds/semver/v3"
|
||||
"github.com/redis/rueidis"
|
||||
"github.com/rocboss/paopao-ce/internal/conf"
|
||||
"github.com/rocboss/paopao-ce/internal/core"
|
||||
"github.com/rocboss/paopao-ce/pkg/utils"
|
||||
)
|
||||
|
||||
var (
|
||||
_webCache core.WebCache = (*redisWebCache)(nil)
|
||||
)
|
||||
|
||||
type redisWebCache struct {
|
||||
cscExpire time.Duration
|
||||
unreadMsgExpire int64
|
||||
c rueidis.Client
|
||||
}
|
||||
|
||||
func (s *redisWebCache) Name() string {
|
||||
return "RedisWebCache"
|
||||
}
|
||||
|
||||
func (s *redisWebCache) Version() *semver.Version {
|
||||
return semver.MustParse("v0.1.0")
|
||||
}
|
||||
|
||||
func (s *redisWebCache) GetUnreadMsgCountResp(uid int64) ([]byte, error) {
|
||||
key := conf.KeyUnreadMsg.Get(uid)
|
||||
res, err := rueidis.MGetCache(s.c, context.Background(), s.cscExpire, []string{key})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
message := res[key]
|
||||
return message.AsBytes()
|
||||
}
|
||||
|
||||
func (s *redisWebCache) PutUnreadMsgCountResp(uid int64, data []byte) error {
|
||||
return s.c.Do(context.Background(), s.c.B().Set().
|
||||
Key(conf.KeyUnreadMsg.Get(uid)).
|
||||
Value(utils.String(data)).
|
||||
ExSeconds(s.unreadMsgExpire).
|
||||
Build()).
|
||||
Error()
|
||||
}
|
||||
|
||||
func (s *redisWebCache) DelUnreadMsgCountResp(uid int64) error {
|
||||
return s.c.Do(context.Background(), s.c.B().Del().Key(conf.KeyUnreadMsg.Get(uid)).Build()).Error()
|
||||
}
|
||||
|
||||
func newWebCache() *redisWebCache {
|
||||
s := conf.CacheSetting
|
||||
return &redisWebCache{
|
||||
cscExpire: s.CientSideCacheExpire,
|
||||
unreadMsgExpire: s.UnreadMsgExpire,
|
||||
c: conf.MustRedisClient(),
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,15 @@
|
||||
// Copyright 2023 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package joint
|
||||
|
||||
import (
|
||||
stdJson "encoding/json"
|
||||
|
||||
"github.com/rocboss/paopao-ce/pkg/json"
|
||||
)
|
||||
|
||||
func RespMarshal(data any) (stdJson.RawMessage, error) {
|
||||
return json.Marshal(data)
|
||||
}
|
||||
@ -0,0 +1,84 @@
|
||||
// Copyright 2023 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package web
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
|
||||
"github.com/alimy/tryst/event"
|
||||
"github.com/rocboss/paopao-ce/internal/core"
|
||||
"github.com/rocboss/paopao-ce/internal/core/ms"
|
||||
"github.com/rocboss/paopao-ce/internal/events"
|
||||
"github.com/rocboss/paopao-ce/internal/model/joint"
|
||||
"github.com/rocboss/paopao-ce/internal/model/web"
|
||||
)
|
||||
|
||||
type cacheUnreadMsgEvent struct {
|
||||
event.UnimplementedEvent
|
||||
ds core.DataService
|
||||
wc core.WebCache
|
||||
uid int64
|
||||
}
|
||||
|
||||
type createMessageEvent struct {
|
||||
event.UnimplementedEvent
|
||||
ds core.DataService
|
||||
wc core.WebCache
|
||||
message *ms.Message
|
||||
}
|
||||
|
||||
func onCacheUnreadMsgEvent(uid int64) {
|
||||
events.OnEvent(&cacheUnreadMsgEvent{
|
||||
ds: _ds,
|
||||
wc: _wc,
|
||||
uid: uid,
|
||||
})
|
||||
}
|
||||
|
||||
func onCreateMessageEvent(data *ms.Message) {
|
||||
events.OnEvent(&createMessageEvent{
|
||||
ds: _ds,
|
||||
wc: _wc,
|
||||
message: data,
|
||||
})
|
||||
}
|
||||
|
||||
func (e *cacheUnreadMsgEvent) Name() string {
|
||||
return "cacheUnreadMsgEvent"
|
||||
}
|
||||
|
||||
func (e *cacheUnreadMsgEvent) Action() error {
|
||||
count, err := e.ds.GetUnreadCount(e.uid)
|
||||
if err != nil {
|
||||
return fmt.Errorf("cacheUnreadMsgEvent action occurs error: %w", err)
|
||||
}
|
||||
resp := &joint.JsonResp{
|
||||
Code: 0,
|
||||
Msg: "success",
|
||||
Data: &web.GetUnreadMsgCountResp{
|
||||
Count: count,
|
||||
},
|
||||
}
|
||||
data, err := json.Marshal(resp)
|
||||
if err != nil {
|
||||
return fmt.Errorf("cacheUnreadMsgEvent action marshal resp occurs error: %w", err)
|
||||
}
|
||||
if err = e.wc.PutUnreadMsgCountResp(e.uid, data); err != nil {
|
||||
return fmt.Errorf("cacheUnreadMsgEvent action put resp data to redis cache occurs error: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (e *createMessageEvent) Name() string {
|
||||
return "createMessageEvent"
|
||||
}
|
||||
|
||||
func (e *createMessageEvent) Action() (err error) {
|
||||
if _, err = e.ds.CreateMessage(e.message); err == nil {
|
||||
err = e.wc.DelUnreadMsgCountResp(e.message.ReceiverUserID)
|
||||
}
|
||||
return
|
||||
}
|
||||
Loading…
Reference in new issue