Try that again

This commit is contained in:
Neil Alexander 2022-04-27 14:53:11 +01:00
parent 6ee8507955
commit 655ac3e8fb
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
2 changed files with 8 additions and 12 deletions

View file

@ -105,7 +105,6 @@ func (s *accountDataStatements) SelectAccountDataInRange(
accountDataEventFilter *gomatrixserverlib.EventFilter,
) (data map[string][]string, pos types.StreamPosition, err error) {
data = make(map[string][]string)
pos = r.High()
rows, err := s.selectAccountDataInRangeStmt.QueryContext(ctx, userID, r.Low(), r.High(),
pq.StringArray(filterConvertTypeWildcardToSQL(accountDataEventFilter.Types)),
@ -120,7 +119,6 @@ func (s *accountDataStatements) SelectAccountDataInRange(
var dataType string
var roomID string
var id types.StreamPosition
var highest types.StreamPosition
for rows.Next() {
if err = rows.Scan(&id, &roomID, &dataType); err != nil {
@ -132,12 +130,12 @@ func (s *accountDataStatements) SelectAccountDataInRange(
} else {
data[roomID] = []string{dataType}
}
if id > highest {
highest = id
if id > pos {
pos = id
}
}
if highest < pos {
pos = highest
if pos == 0 {
pos = r.High()
}
return data, pos, rows.Err()
}

View file

@ -96,7 +96,6 @@ func (s *accountDataStatements) SelectAccountDataInRange(
r types.Range,
filter *gomatrixserverlib.EventFilter,
) (data map[string][]string, pos types.StreamPosition, err error) {
pos = r.High()
data = make(map[string][]string)
stmt, params, err := prepareWithFilters(
s.db, nil, selectAccountDataInRangeSQL,
@ -116,7 +115,6 @@ func (s *accountDataStatements) SelectAccountDataInRange(
var dataType string
var roomID string
var id types.StreamPosition
var highest types.StreamPosition
for rows.Next() {
if err = rows.Scan(&id, &roomID, &dataType); err != nil {
@ -128,12 +126,12 @@ func (s *accountDataStatements) SelectAccountDataInRange(
} else {
data[roomID] = []string{dataType}
}
if id > highest {
highest = id
if id > pos {
pos = id
}
}
if highest < pos {
pos = highest
if pos == 0 {
pos = r.High()
}
return data, pos, nil
}