f8a8dfa459
* feat(backend): small refactoring in integrations * feat(backend): ignore context timeout error
85 lines
2.2 KiB
Go
85 lines
2.2 KiB
Go
package clients
|
|
|
|
import (
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/cloudwatchlogs"
|
|
"openreplay/backend/pkg/messages"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
var reIsException = regexp.MustCompile(`(?i)exception|error`)
|
|
|
|
type cloudwatch struct {
|
|
AwsAccessKeyId string // `json:"aws_access_key_id"`
|
|
AwsSecretAccessKey string // `json:"aws_secret_access_key"`
|
|
LogGroupName string // `json:"log_group_name"`
|
|
Region string // `json:"region"`
|
|
}
|
|
|
|
func (cw *cloudwatch) Request(c *client) error {
|
|
startTs := int64(c.requestData.GetLastMessageTimestamp() + 1) // From next millisecond
|
|
//endTs := utils.CurrentTimestamp()
|
|
sess, err := session.NewSession(aws.NewConfig().
|
|
WithRegion(cw.Region).
|
|
WithCredentials(
|
|
credentials.NewStaticCredentials(cw.AwsAccessKeyId, cw.AwsSecretAccessKey, ""),
|
|
),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
svc := cloudwatchlogs.New(sess)
|
|
|
|
filterOptions := new(cloudwatchlogs.FilterLogEventsInput).
|
|
SetStartTime(startTs). // Inclusively both startTime and endTime
|
|
// SetEndTime(endTs). // Default nil?
|
|
// SetLimit(10000). // Default 10000
|
|
SetLogGroupName(cw.LogGroupName).
|
|
SetFilterPattern("openReplaySessionToken")
|
|
//SetFilterPattern("asayer_session_id")
|
|
|
|
for {
|
|
output, err := svc.FilterLogEvents(filterOptions)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, e := range output.Events {
|
|
if e.Message == nil || e.Timestamp == nil {
|
|
continue
|
|
}
|
|
if !reIsException.MatchString(*e.Message) { // too weak condition ?
|
|
continue
|
|
}
|
|
token, err := GetToken(*e.Message)
|
|
if err != nil {
|
|
c.errChan <- err
|
|
continue
|
|
}
|
|
name := ""
|
|
if e.LogStreamName != nil {
|
|
name = *e.LogStreamName
|
|
}
|
|
timestamp := uint64(*e.Timestamp)
|
|
c.requestData.SetLastMessageTimestamp(timestamp)
|
|
c.evChan <- &SessionErrorEvent{
|
|
//SessionID: sessionID,
|
|
Token: token,
|
|
IntegrationEvent: &messages.IntegrationEvent{
|
|
Source: "cloudwatch",
|
|
Timestamp: timestamp, // e.IngestionTime ??
|
|
Name: name,
|
|
Payload: strings.ReplaceAll(e.String(), "\n", ""),
|
|
},
|
|
}
|
|
}
|
|
|
|
if output.NextToken == nil {
|
|
break
|
|
}
|
|
filterOptions.NextToken = output.NextToken
|
|
}
|
|
return nil
|
|
}
|