feat: delete token traits via worker (#1333)

delete token traits via worker

src/job-handlers/delete-traits.ts

@@ -0,0 +1,29 @@
+import Queue from 'bull'
+import connection from '../connection'
+import { JobData } from '../jobs/delete-traits'
+
+export default async (job: Queue.Job<JobData>, done: Queue.DoneCallback) => {
+    if (!connection.isInitialized) {
+        await connection.initialize().catch((err) => {
+            throw err
+        })
+    }
+
+    const em = connection.manager
+
+    await em.query(
+        `
+        DELETE FROM trait_token WHERE token_id = $1
+    `,
+        [job.data.id]
+    )
+
+    await em.query(
+        `
+        DELETE FROM token_rarity WHERE token_id = $1
+    `,
+        [job.data.id]
+    )
+
+    done()
+}

src/job-handlers/worker.ts

@@ -12,6 +12,7 @@ import { traitsQueue } from '../jobs/compute-traits'
 import { fetchCollectionExtraQueue } from '../jobs/fetch-collection-extra'
 import { invalidateExpiredListings } from '../jobs/invalidate-expired-listings'
 import { rarityQueue } from '../jobs/rarity-ranker'
+import { deleteTraitsQueue } from '../jobs/delete-traits'
 
 async function main() {
     if (!connection.isInitialized) {
@@ -35,6 +36,7 @@ async function main() {
     fetchBalanceQueue.process(5, `${__dirname}/fetch-balance.js`)
     fetchCollectionExtraQueue.process(5, `${__dirname}/fetch-collection-extra.js`)
     invalidateExpiredListings.process(1, `${__dirname}/invalidate-expired-listings.js`)
+    deleteTraitsQueue.process(1, `${__dirname}/delete-traits.js`)
 
     traitsQueue.on('global:failed', (job, err) => {
         throwError(`traitsQueue:Job ${job.id} failed with error: ${err.message}`, 'warning')

src/jobs/delete-traits.ts

@@ -0,0 +1,20 @@
+import Queue from 'bull'
+import { redisConfig } from './common'
+
+export type JobData = { id: string }
+
+export const deleteTraitsQueue = new Queue<JobData>('deleteTraits', {
+    defaultJobOptions: { removeOnComplete: true },
+    redis: redisConfig,
+    settings: {
+        maxStalledCount: 1,
+    },
+})
+
+export const deleteTokenTraits = async (id: string) => {
+    deleteTraitsQueue.add({ id }).catch(() => {
+        // eslint-disable-next-line no-console
+        console.log('Closing connection as Redis is not available')
+        deleteTraitsQueue.close(true)
+    })
+}

src/server-extension/resolvers/delete_token_traits.ts

@@ -2,6 +2,7 @@
 import { Query, Resolver, Arg } from 'type-graphql'
 import 'reflect-metadata'
 import { EntityManager } from 'typeorm'
+import { deleteTokenTraits } from '../../jobs/delete-traits'
 
 @Resolver()
 export class DeleteTokenTraitsResolver {
@@ -14,21 +15,7 @@ export class DeleteTokenTraitsResolver {
         })
         tokenId: string
     ): Promise<boolean> {
-        const manager = await this.tx()
-
-        await manager.query(
-            `
-            DELETE FROM trait_token WHERE token_id = $1
-        `,
-            [tokenId]
-        )
-
-        await manager.query(
-            `
-            DELETE FROM token_rarity WHERE token_id = $1
-        `,
-            [tokenId]
-        )
+        await deleteTokenTraits(tokenId)
 
         return true
     }