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';
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';
ws.on('open', async () => {
console.log('Connected to the server');
ws.send(JSON.stringify({ type: 'subscribe', channel: 'SOL-PERP' }));
ws.send(JSON.stringify({ type: 'subscribe', channel: 'LINK-PERP' }));
ws.send(JSON.stringify({ type: 'subscribe', channel: 'INJ-PERP' }));
// Subscribe and unsubscribe to orderbook channels
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);
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("####################");
});
ws.on('message', (data: WebSocket.Data) => {
try {
const message = JSON.parse(data.toString());
console.log(`Received data from market ${message.channel}`);
// book data is in message.data
console.log(`Received data from channel: ${JSON.stringify(message.channel)}`);
// book and trades data is in message.data
} catch (e) {
console.error('Invalid message:', data);
}

View File

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