updated exmaple and bug fix

This commit is contained in:
Nour Alharithi
2023-11-17 14:19:00 -08:00
parent 3a64d82acf
commit 285d5b32f9
2 changed files with 22 additions and 8 deletions

View File

@@ -1,23 +1,35 @@
import WebSocket from 'ws'; import WebSocket from 'ws';
const ws = new WebSocket('wss://master.dlob.drift.trade/ws'); const ws = new WebSocket('wss://dlob.drift.trade/ws');
import { sleep } from '../src/utils/utils'; import { sleep } from '../src/utils/utils';
ws.on('open', async () => { ws.on('open', async () => {
console.log('Connected to the server'); console.log('Connected to the server');
ws.send(JSON.stringify({ type: 'subscribe', channel: 'SOL-PERP' }));
ws.send(JSON.stringify({ type: 'subscribe', channel: 'LINK-PERP' })); // Subscribe and unsubscribe to orderbook channels
ws.send(JSON.stringify({ type: 'subscribe', channel: 'INJ-PERP' })); ws.send(JSON.stringify({ type: 'subscribe', marketType: 'perp', channel: 'orderbook', market: 'SOL-PERP' }));
ws.send(JSON.stringify({ type: 'subscribe', marketType: 'perp', channel: 'orderbook', market: 'INJ-PERP' }));
ws.send(JSON.stringify({ type: 'subscribe', marketType: 'perp', channel: 'orderbook', market: 'BTC-PERP' }));
ws.send(JSON.stringify({ type: 'subscribe', marketType: 'spot', channel: 'orderbook', market: 'SOL' }));
await sleep(5000); await sleep(5000);
ws.send(JSON.stringify({ type: 'unsubscribe', channel: 'SOL-PERP' })); ws.send(JSON.stringify({ type: 'unsubscribe', marketType: 'perp', channel: 'orderbook', market: 'SOL-PERP' }));
console.log("####################");
// Subscribe to trades data
ws.send(JSON.stringify({ type: 'subscribe', marketType: 'perp', channel: 'trades', market: 'SOL-PERP' }));
ws.send(JSON.stringify({ type: 'subscribe', marketType: 'spot', channel: 'trades', market: 'SOL' }));
await sleep(5000);
ws.send(JSON.stringify({ type: 'unsubscribe', marketType: 'perp', channel: 'trades', market: 'SOL-PERP' }));
console.log("####################"); console.log("####################");
}); });
ws.on('message', (data: WebSocket.Data) => { ws.on('message', (data: WebSocket.Data) => {
try { try {
const message = JSON.parse(data.toString()); const message = JSON.parse(data.toString());
console.log(`Received data from market ${message.channel}`); console.log(`Received data from channel: ${JSON.stringify(message.channel)}`);
// book data is in message.data // book and trades data is in message.data
} catch (e) { } catch (e) {
console.error('Invalid message:', data); console.error('Invalid message:', data);
} }

View File

@@ -159,7 +159,9 @@ async function main() {
`last_update_${redisChannel}` `last_update_${redisChannel}`
); );
if (lastMessage !== null) { if (lastMessage !== null) {
ws.send(JSON.stringify({ redisChannel, data: lastMessage })); ws.send(
JSON.stringify({ channel: redisChannel, data: lastMessage })
);
} }
} }
break; break;