mirror of
				https://github.com/openimsdk/open-im-server.git
				synced 2025-11-04 11:22:10 +08:00 
			
		
		
		
	fix uncorrect condition.
This commit is contained in:
		
							parent
							
								
									18f4e2956f
								
							
						
					
					
						commit
						fa1a6e15c6
					
				@ -162,7 +162,7 @@ func (s *groupServer) BatchGetIncrementalGroupMember(ctx context.Context, req *p
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	groups, err := s.db.FindGroup(ctx, groupIDs)
 | 
						groups, err := s.db.FindGroup(ctx, groupIDs)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return nil, err
 | 
							return nil, errs.Wrap(err)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	for _, group := range groups {
 | 
						for _, group := range groups {
 | 
				
			||||||
@ -190,7 +190,7 @@ func (s *groupServer) BatchGetIncrementalGroupMember(ctx context.Context, req *p
 | 
				
			|||||||
		Versions: func(ctx context.Context, groupIDs []string, versions []uint64, limits []int) (map[string]*model.VersionLog, error) {
 | 
							Versions: func(ctx context.Context, groupIDs []string, versions []uint64, limits []int) (map[string]*model.VersionLog, error) {
 | 
				
			||||||
			vLogs, err := s.db.BatchFindMemberIncrVersion(ctx, groupIDs, versions, limits)
 | 
								vLogs, err := s.db.BatchFindMemberIncrVersion(ctx, groupIDs, versions, limits)
 | 
				
			||||||
			if err != nil {
 | 
								if err != nil {
 | 
				
			||||||
				return nil, err
 | 
									return nil, errs.Wrap(err)
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
			for groupID, vlog := range vLogs {
 | 
								for groupID, vlog := range vLogs {
 | 
				
			||||||
@ -252,7 +252,7 @@ func (s *groupServer) BatchGetIncrementalGroupMember(ctx context.Context, req *p
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	resp, err = opt.Build()
 | 
						resp, err = opt.Build()
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return nil, err
 | 
							return nil, errs.Wrap(err)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	for groupID, val := range resp.RespList {
 | 
						for groupID, val := range resp.RespList {
 | 
				
			||||||
 | 
				
			|||||||
@ -48,7 +48,7 @@ func (o *BatchOption[A, B]) validVersions() []bool {
 | 
				
			|||||||
	valids := make([]bool, len(o.VersionIDs))
 | 
						valids := make([]bool, len(o.VersionIDs))
 | 
				
			||||||
	for i, versionID := range o.VersionIDs {
 | 
						for i, versionID := range o.VersionIDs {
 | 
				
			||||||
		objID, err := primitive.ObjectIDFromHex(versionID)
 | 
							objID, err := primitive.ObjectIDFromHex(versionID)
 | 
				
			||||||
		valids[i] = err == nil && (!objID.IsZero()) && o.VersionNumbers[i] > 0
 | 
							valids[i] = (err == nil && (!objID.IsZero()) && o.VersionNumbers[i] > 0)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	return valids
 | 
						return valids
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@ -75,51 +75,41 @@ func (o *BatchOption[A, B]) getVersions(tags *[]int) (versions map[string]*model
 | 
				
			|||||||
				dIDs = append(dIDs, o.TargetKeys[i])
 | 
									dIDs = append(dIDs, o.TargetKeys[i])
 | 
				
			||||||
				versionNums = append(versionNums, o.VersionNumbers[i])
 | 
									versionNums = append(versionNums, o.VersionNumbers[i])
 | 
				
			||||||
				limits = append(limits, syncLimit)
 | 
									limits = append(limits, syncLimit)
 | 
				
			||||||
 | 
					 | 
				
			||||||
				// version, err := o.Versions(o.Ctx, []string{o.VersionKeys[i]}, []uint64{o.VersionNumbers[i]}, syncLimit)
 | 
					 | 
				
			||||||
				// if err != nil {
 | 
					 | 
				
			||||||
				// 	return nil, err
 | 
					 | 
				
			||||||
				// }
 | 
					 | 
				
			||||||
				// versions[o.VersionKeys[i]] = version[o.VersionKeys[i]]
 | 
					 | 
				
			||||||
			} else {
 | 
								} else {
 | 
				
			||||||
				(*tags)[i] = tagFull
 | 
									(*tags)[i] = tagFull
 | 
				
			||||||
				dIDs = append(dIDs, o.TargetKeys[i])
 | 
									dIDs = append(dIDs, o.TargetKeys[i])
 | 
				
			||||||
				versionNums = append(versionNums, 0)
 | 
									versionNums = append(versionNums, 0)
 | 
				
			||||||
				limits = append(limits, 0)
 | 
									limits = append(limits, 0)
 | 
				
			||||||
 | 
					 | 
				
			||||||
				// version, err := o.Versions(o.Ctx, []string{o.VersionKeys[i]}, []uint64{0}, 0)
 | 
					 | 
				
			||||||
				// if err != nil {
 | 
					 | 
				
			||||||
				// 	return nil, err
 | 
					 | 
				
			||||||
				// }
 | 
					 | 
				
			||||||
				// versions[o.VersionKeys[i]] = version[o.VersionKeys[i]]
 | 
					 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		versions, err = o.Versions(o.Ctx, dIDs, versionNums, limits)
 | 
							versions, err = o.Versions(o.Ctx, dIDs, versionNums, limits)
 | 
				
			||||||
		if err != nil {
 | 
							if err != nil {
 | 
				
			||||||
			return nil, errs.Wrap(err)
 | 
								return nil, errs.Wrap(err)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
		return versions, nil
 | 
							return versions, nil
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	} else {
 | 
						} else {
 | 
				
			||||||
		caches, err := o.CacheMaxVersions(o.Ctx, o.TargetKeys)
 | 
							caches, err := o.CacheMaxVersions(o.Ctx, o.TargetKeys)
 | 
				
			||||||
		if err != nil {
 | 
							if err != nil {
 | 
				
			||||||
			return nil, err
 | 
								return nil, errs.Wrap(err)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		objIDs := make([]primitive.ObjectID, len(o.VersionIDs))
 | 
							objIDs := make([]primitive.ObjectID, len(o.VersionIDs))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		for i, versionID := range o.VersionIDs {
 | 
							for i, versionID := range o.VersionIDs {
 | 
				
			||||||
			objID, _ := primitive.ObjectIDFromHex(versionID)
 | 
								objID, _ := primitive.ObjectIDFromHex(versionID)
 | 
				
			||||||
			objIDs[i] = objID
 | 
								objIDs[i] = objID
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		equals := o.equalIDs(objIDs)
 | 
							equals := o.equalIDs(objIDs)
 | 
				
			||||||
		for i, valid := range valids {
 | 
							for i, valid := range valids {
 | 
				
			||||||
			if !valid {
 | 
								if !valid {
 | 
				
			||||||
				(*tags)[i] = tagFull
 | 
									(*tags)[i] = tagFull
 | 
				
			||||||
				// versions[o.VersionKeys[i]] = caches[o.VersionKeys[i]]
 | 
					 | 
				
			||||||
			} else if !equals[i] {
 | 
								} else if !equals[i] {
 | 
				
			||||||
				(*tags)[i] = tagFull
 | 
									(*tags)[i] = tagFull
 | 
				
			||||||
				// versions[o.VersionKeys[i]] = caches[o.VersionKeys[i]]
 | 
					 | 
				
			||||||
			} else if o.VersionNumbers[i] == uint64(caches[o.TargetKeys[i]].Version) {
 | 
								} else if o.VersionNumbers[i] == uint64(caches[o.TargetKeys[i]].Version) {
 | 
				
			||||||
				(*tags)[i] = tagEqual
 | 
									(*tags)[i] = tagEqual
 | 
				
			||||||
				// versions[o.VersionKeys[i]] = caches[o.VersionKeys[i]]
 | 
					 | 
				
			||||||
			} else {
 | 
								} else {
 | 
				
			||||||
				(*tags)[i] = tagQuery
 | 
									(*tags)[i] = tagQuery
 | 
				
			||||||
				dIDs = append(dIDs, o.TargetKeys[i])
 | 
									dIDs = append(dIDs, o.TargetKeys[i])
 | 
				
			||||||
@ -127,16 +117,20 @@ func (o *BatchOption[A, B]) getVersions(tags *[]int) (versions map[string]*model
 | 
				
			|||||||
				limits = append(limits, syncLimit)
 | 
									limits = append(limits, syncLimit)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
				delete(caches, o.TargetKeys[i])
 | 
									delete(caches, o.TargetKeys[i])
 | 
				
			||||||
				// versions[o.VersionKeys[i]] = version[o.VersionKeys[i]]
 | 
					 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
		version, err := o.Versions(o.Ctx, dIDs, versionNums, limits)
 | 
					
 | 
				
			||||||
		if err != nil {
 | 
							if dIDs != nil {
 | 
				
			||||||
			return nil, err
 | 
								versionMap, err := o.Versions(o.Ctx, dIDs, versionNums, limits)
 | 
				
			||||||
		}
 | 
								if err != nil {
 | 
				
			||||||
		for k, v := range version {
 | 
									return nil, errs.Wrap(err)
 | 
				
			||||||
			caches[k] = v
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								for k, v := range versionMap {
 | 
				
			||||||
 | 
									caches[k] = v
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		versions = caches
 | 
							versions = caches
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	return versions, nil
 | 
						return versions, nil
 | 
				
			||||||
@ -144,13 +138,13 @@ func (o *BatchOption[A, B]) getVersions(tags *[]int) (versions map[string]*model
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
func (o *BatchOption[A, B]) Build() (*B, error) {
 | 
					func (o *BatchOption[A, B]) Build() (*B, error) {
 | 
				
			||||||
	if err := o.check(); err != nil {
 | 
						if err := o.check(); err != nil {
 | 
				
			||||||
		return nil, err
 | 
							return nil, errs.Wrap(err)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	tags := make([]int, len(o.TargetKeys))
 | 
						tags := make([]int, len(o.TargetKeys))
 | 
				
			||||||
	versions, err := o.getVersions(&tags)
 | 
						versions, err := o.getVersions(&tags)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return nil, err
 | 
							return nil, errs.Wrap(err)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	fullMap := make(map[string]bool)
 | 
						fullMap := make(map[string]bool)
 | 
				
			||||||
@ -193,7 +187,7 @@ func (o *BatchOption[A, B]) Build() (*B, error) {
 | 
				
			|||||||
		if len(insertIds) > 0 {
 | 
							if len(insertIds) > 0 {
 | 
				
			||||||
			insertList, err := o.Find(o.Ctx, targetKey, insertIds)
 | 
								insertList, err := o.Find(o.Ctx, targetKey, insertIds)
 | 
				
			||||||
			if err != nil {
 | 
								if err != nil {
 | 
				
			||||||
				return nil, err
 | 
									return nil, errs.Wrap(err)
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
			insertListMap[targetKey] = insertList
 | 
								insertListMap[targetKey] = insertList
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
@ -203,7 +197,7 @@ func (o *BatchOption[A, B]) Build() (*B, error) {
 | 
				
			|||||||
		if len(updateIds) > 0 {
 | 
							if len(updateIds) > 0 {
 | 
				
			||||||
			updateList, err := o.Find(o.Ctx, targetKey, updateIds)
 | 
								updateList, err := o.Find(o.Ctx, targetKey, updateIds)
 | 
				
			||||||
			if err != nil {
 | 
								if err != nil {
 | 
				
			||||||
				return nil, err
 | 
									return nil, errs.Wrap(err)
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
			updateListMap[targetKey] = updateList
 | 
								updateListMap[targetKey] = updateList
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
				
			|||||||
@ -504,7 +504,7 @@ func (g *groupDatabase) FindMemberIncrVersion(ctx context.Context, groupID strin
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
func (g *groupDatabase) BatchFindMemberIncrVersion(ctx context.Context, groupIDs []string, versions []uint64, limits []int) (map[string]*model.VersionLog, error) {
 | 
					func (g *groupDatabase) BatchFindMemberIncrVersion(ctx context.Context, groupIDs []string, versions []uint64, limits []int) (map[string]*model.VersionLog, error) {
 | 
				
			||||||
	if len(groupIDs) == 0 {
 | 
						if len(groupIDs) == 0 {
 | 
				
			||||||
		return nil, errs.New("groupIDs is nil.")
 | 
							return nil, errs.Wrap(errs.New("groupIDs is nil."))
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// convert []uint64 to []uint
 | 
						// convert []uint64 to []uint
 | 
				
			||||||
@ -535,7 +535,7 @@ func (g *groupDatabase) FindMaxGroupMemberVersionCache(ctx context.Context, grou
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
func (g *groupDatabase) BatchFindMaxGroupMemberVersionCache(ctx context.Context, groupIDs []string) (map[string]*model.VersionLog, error) {
 | 
					func (g *groupDatabase) BatchFindMaxGroupMemberVersionCache(ctx context.Context, groupIDs []string) (map[string]*model.VersionLog, error) {
 | 
				
			||||||
	if len(groupIDs) == 0 {
 | 
						if len(groupIDs) == 0 {
 | 
				
			||||||
		return nil, nil
 | 
							return nil, errs.Wrap(errs.New("groupIDs is nil in Cache."))
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	versionLogs, err := g.cache.BatchFindMaxGroupMemberVersion(ctx, groupIDs)
 | 
						versionLogs, err := g.cache.BatchFindMaxGroupMemberVersion(ctx, groupIDs)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user