mirror of
				https://github.com/openimsdk/open-im-server.git
				synced 2025-11-04 03:13:15 +08:00 
			
		
		
		
	fix:get max seqs
This commit is contained in:
		
							parent
							
								
									04a5c97e31
								
							
						
					
					
						commit
						e595bb0230
					
				
							
								
								
									
										16
									
								
								pkg/common/storage/cache/redis/seq.go
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										16
									
								
								pkg/common/storage/cache/redis/seq.go
									
									
									
									
										vendored
									
									
								
							@ -22,6 +22,7 @@ import (
 | 
			
		||||
	"github.com/openimsdk/tools/errs"
 | 
			
		||||
	"github.com/openimsdk/tools/utils/stringutil"
 | 
			
		||||
	"github.com/redis/go-redis/v9"
 | 
			
		||||
	"sync"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func NewSeqCache(rdb redis.UniversalClient) cache.SeqCache {
 | 
			
		||||
@ -62,17 +63,21 @@ func (c *seqCache) getSeq(ctx context.Context, conversationID string, getkey fun
 | 
			
		||||
 | 
			
		||||
func (c *seqCache) getSeqs(ctx context.Context, items []string, getkey func(s string) string) (m map[string]int64, err error) {
 | 
			
		||||
	m = make(map[string]int64, len(items))
 | 
			
		||||
	var lock sync.Mutex
 | 
			
		||||
 | 
			
		||||
	keys := make([]string, len(items))
 | 
			
		||||
	for i, v := range items {
 | 
			
		||||
		keys[i] = getkey(v)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	manager := NewRedisShardManager(c.rdb)
 | 
			
		||||
	if err = manager.ProcessKeysBySlot(ctx, keys, func(ctx context.Context, _ int64, keys []string) error {
 | 
			
		||||
		res, err := c.rdb.MGet(ctx, keys...).Result()
 | 
			
		||||
		if err != nil && !errors.Is(err, redis.Nil) {
 | 
			
		||||
		return nil, errs.Wrap(err)
 | 
			
		||||
			return errs.Wrap(err)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
	// len(res) == len(items)
 | 
			
		||||
		// len(res) <= len(items)
 | 
			
		||||
		for i := range res {
 | 
			
		||||
			strRes, ok := res[i].(string)
 | 
			
		||||
			if !ok {
 | 
			
		||||
@ -80,9 +85,16 @@ func (c *seqCache) getSeqs(ctx context.Context, items []string, getkey func(s st
 | 
			
		||||
			}
 | 
			
		||||
			val := stringutil.StringToInt64(strRes)
 | 
			
		||||
			if val != 0 {
 | 
			
		||||
				lock.Lock()
 | 
			
		||||
				m[items[i]] = val
 | 
			
		||||
				lock.Unlock()
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return nil
 | 
			
		||||
	}); err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return m, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user