23 lines
360 B
Go
23 lines
360 B
Go
package kinesis
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go/aws/request"
|
|
)
|
|
|
|
var readDuration = 5 * time.Second
|
|
|
|
func init() {
|
|
ops := []string{
|
|
opGetRecords,
|
|
}
|
|
initRequest = func(r *request.Request) {
|
|
for _, operation := range ops {
|
|
if r.Operation.Name == operation {
|
|
r.ApplyOptions(request.WithResponseReadTimeout(readDuration))
|
|
}
|
|
}
|
|
}
|
|
}
|