forked from Azure/azure-sdk-for-js
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Run format for SB, EH and Core AMQP (Azure#14323)
- Loading branch information
1 parent
d68cc4b
commit 603e521
Showing
10 changed files
with
151 additions
and
120 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
14 changes: 12 additions & 2 deletions
14
sdk/servicebus/service-bus/test/internal/smoketest.spec.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
169 changes: 91 additions & 78 deletions
169
sdk/servicebus/service-bus/test/stress/scenarioLongRunning.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,105 +1,118 @@ | ||
import {hostname} from "os"; | ||
import { captureConsoleOutputToAppInsights, defaultClient, SBStressTestsBase } from "./stressTestsBase"; | ||
import { hostname } from "os"; | ||
import { | ||
captureConsoleOutputToAppInsights, | ||
defaultClient, | ||
SBStressTestsBase | ||
} from "./stressTestsBase"; | ||
import { AbortController, AbortSignalLike } from "@azure/abort-controller"; | ||
import { ServiceBusClient, ServiceBusSender } from "@azure/service-bus"; | ||
import { v4 as uuidv4 } from "uuid"; | ||
|
||
captureConsoleOutputToAppInsights(); | ||
|
||
const stressTest = new SBStressTestsBase({ | ||
testName: "longRunning", | ||
snapshotFocus: [ | ||
"send-info", | ||
"receive-info" | ||
] | ||
testName: "longRunning", | ||
snapshotFocus: ["send-info", "receive-info"] | ||
}); | ||
|
||
async function looper(fn: () => Promise<void>, delay: number, abortSignal: AbortSignalLike) { | ||
const timeout = () => new Promise((resolve) => setTimeout(() => resolve(true), delay)); | ||
const timeout = () => new Promise((resolve) => setTimeout(() => resolve(true), delay)); | ||
|
||
while (!abortSignal.aborted && await timeout()) { | ||
await fn(); | ||
} | ||
while (!abortSignal.aborted && (await timeout())) { | ||
await fn(); | ||
} | ||
} | ||
|
||
async function sendMessagesForever(clientForSender: ServiceBusClient, abortSignal: AbortSignalLike) { | ||
console.log(`Started message sending`); | ||
async function sendMessagesForever( | ||
clientForSender: ServiceBusClient, | ||
abortSignal: AbortSignalLike | ||
) { | ||
console.log(`Started message sending`); | ||
|
||
let sender: ServiceBusSender | undefined; | ||
let sender: ServiceBusSender | undefined; | ||
|
||
return looper(async () => { | ||
if (abortSignal.aborted) { | ||
console.log(`Aborting sending because of abortSignal`); | ||
return; | ||
} | ||
return looper( | ||
async () => { | ||
if (abortSignal.aborted) { | ||
console.log(`Aborting sending because of abortSignal`); | ||
return; | ||
} | ||
|
||
try { | ||
if (sender == null) { | ||
sender = clientForSender.createSender(stressTest.queueName); | ||
} | ||
|
||
const messagesToSend = [{ | ||
messageId: uuidv4(), | ||
body: `Message: ${Date.now()}` | ||
}]; | ||
|
||
stressTest.trackSentMessages(messagesToSend); | ||
await sender.sendMessages(messagesToSend); | ||
} catch (err) { | ||
console.log(`Sending message failed: `, err); | ||
stressTest.trackError("send", err); | ||
sender = undefined; | ||
try { | ||
if (sender == null) { | ||
sender = clientForSender.createSender(stressTest.queueName); | ||
} | ||
}, 1000, abortSignal); | ||
|
||
const messagesToSend = [ | ||
{ | ||
messageId: uuidv4(), | ||
body: `Message: ${Date.now()}` | ||
} | ||
]; | ||
|
||
stressTest.trackSentMessages(messagesToSend); | ||
await sender.sendMessages(messagesToSend); | ||
} catch (err) { | ||
console.log(`Sending message failed: `, err); | ||
stressTest.trackError("send", err); | ||
sender = undefined; | ||
} | ||
}, | ||
1000, | ||
abortSignal | ||
); | ||
} | ||
|
||
async function main() { | ||
const abortController = new AbortController(); | ||
const abortSignal = abortController.signal; | ||
|
||
await stressTest.init(); | ||
|
||
console.log(`Starting with hostname ${hostname}`); | ||
|
||
defaultClient.trackEvent({ | ||
name: "ApplicationStart" | ||
}); | ||
|
||
const clientForReceiver = stressTest.createServiceBusClient(); | ||
|
||
const receiver = clientForReceiver.createReceiver(stressTest.queueName, { | ||
receiveMode: "peekLock" | ||
}); | ||
|
||
console.log(`Subscribing...`); | ||
|
||
const subscription = receiver.subscribe({ | ||
processMessage: async (msg) => { | ||
stressTest.addReceivedMessage([msg]); | ||
await stressTest.completeMessage(receiver, msg); | ||
}, | ||
processError: async (args) => { | ||
console.log(`subscribe error:`, args); | ||
stressTest.trackError("receive", args.error); | ||
} | ||
}, { | ||
autoCompleteMessages: false, | ||
maxConcurrentCalls: 10 | ||
}); | ||
const abortController = new AbortController(); | ||
const abortSignal = abortController.signal; | ||
|
||
await stressTest.init(); | ||
|
||
console.log(`Starting with hostname ${hostname}`); | ||
|
||
defaultClient.trackEvent({ | ||
name: "ApplicationStart" | ||
}); | ||
|
||
const clientForReceiver = stressTest.createServiceBusClient(); | ||
|
||
const receiver = clientForReceiver.createReceiver(stressTest.queueName, { | ||
receiveMode: "peekLock" | ||
}); | ||
|
||
console.log(`Subscribing...`); | ||
|
||
const subscription = receiver.subscribe( | ||
{ | ||
processMessage: async (msg) => { | ||
stressTest.addReceivedMessage([msg]); | ||
await stressTest.completeMessage(receiver, msg); | ||
}, | ||
processError: async (args) => { | ||
console.log(`subscribe error:`, args); | ||
stressTest.trackError("receive", args.error); | ||
} | ||
}, | ||
{ | ||
autoCompleteMessages: false, | ||
maxConcurrentCalls: 10 | ||
} | ||
); | ||
|
||
const clientForSender = stressTest.createServiceBusClient(); | ||
const clientForSender = stressTest.createServiceBusClient(); | ||
|
||
await sendMessagesForever(clientForSender, abortSignal); | ||
defaultClient.flush(); | ||
await sendMessagesForever(clientForSender, abortSignal); | ||
defaultClient.flush(); | ||
|
||
await subscription.close(); | ||
await clientForReceiver.close(); | ||
await clientForSender.close(); | ||
await subscription.close(); | ||
await clientForReceiver.close(); | ||
await clientForSender.close(); | ||
|
||
await stressTest.end(); | ||
await stressTest.end(); | ||
} | ||
|
||
main().catch((err) => { | ||
console.error(err); | ||
process.exit(1); | ||
main().catch((err) => { | ||
console.error(err); | ||
process.exit(1); | ||
}); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.