fix uncorrect condition.

pull/2417/head
Monet Lee 1 year ago
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 dIDs != nil {
versionMap, err := o.Versions(o.Ctx, dIDs, versionNums, limits)
if err != nil { if err != nil {
return nil, err return nil, errs.Wrap(err)
} }
for k, v := range version {
for k, v := range versionMap {
caches[k] = v 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…
Cancel
Save