withchao d55d416f68 seq
2024-06-17 17:41:28 +08:00

1.4 KiB
Raw Blame History

mongo

type Seq struct {
	ConversationID string `bson:"conversation_id"`
	MaxSeq         int64  `bson:"max_seq"`
	MinSeq         int64  `bson:"min_seq"`
}
type Seq interface {
  Malloc(ctx context.Context, conversationID string, size int64) (int64, error)
  GetMaxSeq(ctx context.Context, conversationID string) (int64, error)
  GetMinSeq(ctx context.Context, conversationID string) (int64, error)
  SetMinSeq(ctx context.Context, conversationID string, seq int64) error
}
  1. Malloc 申请seq数量返回的已有seq的最大值消息用的第一个seq是返回值+1
  2. GetMaxSeq 获取申请的seq的最大值在发消息的seq小于这个值
  3. GetMinSeq 获取最小的seq用于拉取历史消息
  4. SetMinSeq 设置最小的seq用于拉取历史消息

redis

type RedisSeq struct {
	Curr int64 // 当前的最大seq
	Last int64 // mongodb中申请的最大seq
	Lock *int64 // 锁用于在mongodb中申请seq
}
  1. Malloc 申请seq数量返回的已有seq的最大值消息用的第一个seq是返回值+1如果redis中申请数量够用直接返回并自增对应数量。如果redis中申请数量不够用加锁从mongodb中申请seq。
  2. GetMaxSeq 获取已发消息的最大seq就是Curr的值。如果redis中缓存不存在就通过mongodb获取最大seq。存储在redis中。其中Curr和Last都是这个seq值。
  3. GetMinSeq, SetMinSeq用之前rockscache的方案。