evan/rats-search
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(store): fix potential out of records on find request

Browse Source
This commit is contained in:
Alexey Kasyanchuk 2018-06-15 16:29:17 +03:00
parent 1fbc889011
commit 9502e53fc1
1 changed files with 6 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/background/store.js
View File

@ -1,5 +1,6 @@
const objectHash = require('object-hash');
const EventEmitter = require('events');
const forBigTable = require('./forBigTable')
module.exports = class P2PStore extends EventEmitter {
constructor(p2p, sphinx)
@ -127,7 +128,6 @@ module.exports = class P2PStore extends EventEmitter {
_pushToDb(value, callback)
{
this.emit('store', value)
const data = this.sphinx.escape(JSON.stringify(value.data))
this.sphinx.query(
`insert into store(id, hash, peerId, data` + (value.index || value.data._index ? ', storeIndex' : '') + `)
@ -142,6 +142,7 @@ module.exports = class P2PStore extends EventEmitter {
if(callback)
callback()
})
this.emit('store', value)
}
store(obj)
@ -176,19 +177,10 @@ module.exports = class P2PStore extends EventEmitter {
return true
}
find(index)
async find(index)
{
return new Promise((resolve) => {
this.sphinx.query(`select * from store where match(${this.sphinx.escape(index)}) LIMIT 50000`, (err, records) => {
if(err)
{
console.log(err)
resolve(false)
return
}
resolve(records.map( ({data, peerid}) => Object.assign(JSON.parse(data), { _peerId: peerid }) ))
})
})
const records = []
await forBigTable(this.sphinx, 'store', (record) => records.push(record), null, 1000, `and match(${this.sphinx.escape(index)})`)
return records.map( ({data, peerid}) => Object.assign(JSON.parse(data), { _peerId: peerid }) )
}
}