blockworks-foundation
/
mango-alerts
mirror of https://github.com/blockworks-foundation/mango-alerts.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

getAllMarginAccounts in reduceMangoGroups

This commit is contained in:
Ralfs 2021-05-07 11:03:57 +03:00
parent 89d6abef36
commit fe3ef3410b
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/index.ts
View File

@ -7,7 +7,7 @@ import cors from "@koa/cors";
import * as cron from "node-cron"; import * as cron from "node-cron";
import {MongoClient, ObjectId} from "mongodb"; import {MongoClient, ObjectId} from "mongodb";
import { MangoClient, IDS } from '@blockworks-foundation/mango-client'; import { MangoClient, MarginAccount, IDS } from '@blockworks-foundation/mango-client';
import { Connection, PublicKey } from '@solana/web3.js'; import { Connection, PublicKey } from '@solana/web3.js';
import { UserError } from './errors'; import { UserError } from './errors';
@ -25,6 +25,7 @@ const client = new MangoClient();
const clusterIds = IDS[cluster]; const clusterIds = IDS[cluster];
const connection = new Connection(config.rpcEndpoint || IDS.cluster_urls[cluster], 'singleGossip'); const connection = new Connection(config.rpcEndpoint || IDS.cluster_urls[cluster], 'singleGossip');
const dexProgramId = new PublicKey(clusterIds.dex_program_id); const dexProgramId = new PublicKey(clusterIds.dex_program_id);
const mangoProgramId = new PublicKey(clusterIds.mango_program_id);
app.use(cors()); app.use(cors());
app.use(bodyParser()); app.use(bodyParser());
@ -104,14 +105,15 @@ app.listen(config.port, () => {
const handleAlert = async (alert: any, mangoGroups: any[], db: any) => { const handleAlert = async (alert: any, mangoGroups: any[], db: any) => {
try { try {
const mangoGroupMapping = mangoGroups[alert.mangoGroupPk];
const marginAccountPk = new PublicKey(alert.marginAccountPk); const marginAccountPk = new PublicKey(alert.marginAccountPk);
const marginAccount = await client.getMarginAccount(connection, marginAccountPk, dexProgramId); const marginAccount = mangoGroupMapping.marginAccounts.find((ma: MarginAccount) => ma.publicKey.equals(marginAccountPk));
const collateralRatio = marginAccount.getCollateralRatio(mangoGroups[alert.mangoGroupPk]['mangoGroup'], mangoGroups[alert.mangoGroupPk]['prices']); const collateralRatio = marginAccount.getCollateralRatio(mangoGroupMapping['mangoGroup'], mangoGroupMapping['prices']);
if ((100 * collateralRatio) <= alert.collateralRatioThresh) { if ((100 * collateralRatio) <= alert.collateralRatioThresh) {
let message = MESSAGE.replace('@ratio@', alert.collateralRatioThresh); let message = MESSAGE.replace('@ratio@', alert.collateralRatioThresh);
message += marginAccount.toPrettyString( message += marginAccount.toPrettyString(
mangoGroups[alert.mangoGroupPk]['mangoGroup'], mangoGroupMapping['mangoGroup'],
mangoGroups[alert.mangoGroupPk]['prices'] mangoGroupMapping['prices']
); );
message += '\nVisit https://trade.mango.markets/' message += '\nVisit https://trade.mango.markets/'
const alertSent = await sendAlert(alert, message); const alertSent = await sendAlert(alert, message);
@ -131,7 +133,7 @@ const runCron = async () => {
try { try {
const alerts = await db.collection('alerts').find({open: true}).toArray(); const alerts = await db.collection('alerts').find({open: true}).toArray();
const uniqueMangoGroupPks: string[] = [...new Set(alerts.map(alert => alert.mangoGroupPk))]; const uniqueMangoGroupPks: string[] = [...new Set(alerts.map(alert => alert.mangoGroupPk))];
const mangoGroups:any = await reduceMangoGroups(client, connection, uniqueMangoGroupPks); const mangoGroups:any = await reduceMangoGroups(client, connection, mangoProgramId, uniqueMangoGroupPks);
alerts.forEach(async (alert) => { alerts.forEach(async (alert) => {
handleAlert(alert, mangoGroups, db); handleAlert(alert, mangoGroups, db);
}); });

4
src/utils.ts
View File

@ -92,12 +92,14 @@ export const sendAlert = async (alert: any, message: string) => {
return true; return true;
} }
export const reduceMangoGroups = async (client: MangoClient, connection: Connection, mangoGroupPks: string[]) => { export const reduceMangoGroups = async (client: MangoClient, connection: Connection, mangoProgramId: PublicKey, mangoGroupPks: string[]) => {
const mangoGroups:any = {}; const mangoGroups:any = {};
for (let mangoGroupPk of mangoGroupPks) { for (let mangoGroupPk of mangoGroupPks) {
const mangoGroup = await client.getMangoGroup(connection, new PublicKey(mangoGroupPk)); const mangoGroup = await client.getMangoGroup(connection, new PublicKey(mangoGroupPk));
const marginAccounts = await client.getAllMarginAccounts(connection, mangoProgramId, mangoGroup);
mangoGroups[mangoGroupPk] = { mangoGroups[mangoGroupPk] = {
mangoGroup, mangoGroup,
marginAccounts,
prices: await mangoGroup.getPrices(connection), prices: await mangoGroup.getPrices(connection),
}; };
} }