@drift-labs/sdk
Version:
SDK for Drift Protocol
46 lines (45 loc) • 2 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.TestBulkAccountLoader = void 0;
const bulkAccountLoader_1 = require("./bulkAccountLoader");
class TestBulkAccountLoader extends bulkAccountLoader_1.BulkAccountLoader {
async loadChunk(accountsToLoadChunks) {
if (accountsToLoadChunks.length === 0) {
return;
}
const accounts = [];
for (const accountsToLoadChunk of accountsToLoadChunks) {
for (const accountToLoad of accountsToLoadChunk) {
const account = await this.connection.getAccountInfoAndContext(accountToLoad.publicKey, this.commitment);
accounts.push(account);
const newSlot = account.context.slot;
if (newSlot > this.mostRecentSlot) {
this.mostRecentSlot = newSlot;
}
if (accountToLoad.callbacks.size === 0) {
return;
}
const key = accountToLoad.publicKey.toBase58();
const prev = this.bufferAndSlotMap.get(key);
if (prev && newSlot < prev.slot) {
return;
}
let newBuffer = undefined;
if (account.value) {
newBuffer = account.value.data;
}
if (!prev) {
this.bufferAndSlotMap.set(key, { slot: newSlot, buffer: newBuffer });
this.handleAccountCallbacks(accountToLoad, newBuffer, newSlot);
return;
}
const oldBuffer = prev.buffer;
if (newBuffer && (!oldBuffer || !newBuffer.equals(oldBuffer))) {
this.bufferAndSlotMap.set(key, { slot: newSlot, buffer: newBuffer });
this.handleAccountCallbacks(accountToLoad, newBuffer, newSlot);
}
}
}
}
}
exports.TestBulkAccountLoader = TestBulkAccountLoader;