Skip to content

Commit

Permalink
align with cloud
Browse files Browse the repository at this point in the history
  • Loading branch information
idanasulin2706 committed Nov 25, 2023
1 parent ac72be8 commit a3bdb7b
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions server/memphis_handlers_dls_messages.go
Original file line number Diff line number Diff line change
Expand Up @@ -231,13 +231,13 @@ func (pmh PoisonMessagesHandler) GetDlsMsgsByStationLight(station models.Station
poisonMessages = append(poisonMessages, models.LightDlsMessageResponse{MessageSeq: v.MessageSeq, ID: v.ID, Message: messageDetails})
case "schema":
messageDetails.Size = len(v.MessageDetails.Data) + len(v.MessageDetails.Headers)
schemaMessages = append(schemaMessages, models.LightDlsMessageResponse{MessageSeq: v.MessageSeq, ID: v.ID, Message: v.MessageDetails})
schemaMessages = append(schemaMessages, models.LightDlsMessageResponse{MessageSeq: v.MessageSeq, ID: v.ID, Message: messageDetails})
case "functions":
functionsMessages = append(functionsMessages, models.LightDlsMessageResponse{MessageSeq: v.MessageSeq, ID: v.ID, Message: v.MessageDetails})
functionsMessages = append(functionsMessages, models.LightDlsMessageResponse{MessageSeq: v.MessageSeq, ID: v.ID, Message: messageDetails})
}
}

lenPoison, lenSchema := len(poisonMessages), len(schemaMessages)
lenPoison, lenSchema, lenFunctions := len(poisonMessages), len(schemaMessages), len(functionsMessages)
totalDlsAmount := 0
if len(dlsMsgs) >= 0 {
totalDlsAmount, err = db.CountDlsMsgsByStationAndPartition(station.ID, partitionNumber)
Expand All @@ -254,13 +254,21 @@ func (pmh PoisonMessagesHandler) GetDlsMsgsByStationLight(station models.Station
return schemaMessages[i].Message.TimeSent.After(schemaMessages[j].Message.TimeSent)
})

sort.Slice(functionsMessages, func(i, j int) bool {
return functionsMessages[i].Message.TimeSent.After(functionsMessages[j].Message.TimeSent)
})

if lenPoison > 1000 {
poisonMessages = poisonMessages[:1000]
}

if lenSchema > 1000 {
schemaMessages = schemaMessages[:1000]
}

if lenFunctions > 1000 {
functionsMessages = functionsMessages[:1000]
}
return poisonMessages, schemaMessages, functionsMessages, totalDlsAmount, nil
}

Expand Down

0 comments on commit a3bdb7b

Please sign in to comment.