trade/trade-drift-dlob
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

properly apply pmm mode

Browse Source
This commit is contained in:
wphan
2025-01-31 09:52:29 -08:00
parent 49d3d6c08a
commit b29a68fdfc
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/dlobProvider.ts
View File

@@ -3,7 +3,7 @@ import { PublicKey } from '@solana/web3.js';
export type DLOBProvider = { export type DLOBProvider = {
subscribe(): Promise<void>; subscribe(): Promise<void>;
getDLOB(slot: number): Promise<DLOB>; getDLOB(slot: number, protectedMakerView: boolean): Promise<DLOB>;
getUniqueAuthorities(): PublicKey[]; getUniqueAuthorities(): PublicKey[];
getUserAccounts(): Generator<{ getUserAccounts(): Generator<{
userAccount: UserAccount; userAccount: UserAccount;
@@ -20,8 +20,8 @@ export function getDLOBProviderFromUserMap(userMap: UserMap): DLOBProvider {
subscribe: async () => { subscribe: async () => {
await userMap.subscribe(); await userMap.subscribe();
}, },
getDLOB: async (slot: number) => { getDLOB: async (slot: number, protectedMakerView: boolean) => {
return await userMap.getDLOB(slot); return await userMap.getDLOB(slot, protectedMakerView);
}, },
getUniqueAuthorities: () => { getUniqueAuthorities: () => {
return userMap.getUniqueAuthorities(); return userMap.getUniqueAuthorities();
@@ -56,8 +56,8 @@ export function getDLOBProviderFromOrderSubscriber(
subscribe: async () => { subscribe: async () => {
await orderSubscriber.subscribe(); await orderSubscriber.subscribe();
}, },
getDLOB: async (slot: number) => { getDLOB: async (slot: number, protectedMakerView: boolean) => {
return await orderSubscriber.getDLOB(slot); return await orderSubscriber.getDLOB(slot, protectedMakerView);
}, },
getUniqueAuthorities: () => { getUniqueAuthorities: () => {
const authorities = new Set<string>(); const authorities = new Set<string>();

4
src/publishers/dlobPublisher.ts
View File

@@ -590,6 +590,7 @@ const main = async () => {
const handleDebug = async (req: Request, res: Response) => { const handleDebug = async (req: Request, res: Response) => {
const marketIndex = +req.query.marketIndex; const marketIndex = +req.query.marketIndex;
const protectedMakerView = req.query.includePmm === 'true';
let marketType: MarketType = MarketType.PERP; let marketType: MarketType = MarketType.PERP;
let oraclePriceData: OraclePriceData; let oraclePriceData: OraclePriceData;
if (req.query.marketType === 'spot') { if (req.query.marketType === 'spot') {
@@ -600,7 +601,7 @@ const main = async () => {
} }
try { try {
const slot = slotSource.getSlot(); const slot = slotSource.getSlot();
const dlob = await dlobProvider.getDLOB(slot); const dlob = await dlobProvider.getDLOB(slot, protectedMakerView);
const l2 = dlob.getL2({ const l2 = dlob.getL2({
marketIndex, marketIndex,
marketType, marketType,
@@ -623,6 +624,7 @@ const main = async () => {
}, },
l2: l2WithBNToStrings(l2), l2: l2WithBNToStrings(l2),
l3, l3,
includePmm: protectedMakerView,
}; };
res.json(state); res.json(state);