mirror of
https://github.com/idanoo/GoScrobble.git
synced 2024-11-28 03:15:16 +00:00
186 lines
5.8 KiB
JavaScript
186 lines
5.8 KiB
JavaScript
|
/*
|
||
|
Copyright 2018 Google LLC
|
||
|
|
||
|
Use of this source code is governed by an MIT-style
|
||
|
license that can be found in the LICENSE file or at
|
||
|
https://opensource.org/licenses/MIT.
|
||
|
*/
|
||
|
import { assert } from 'workbox-core/_private/assert.js';
|
||
|
import { DBWrapper } from 'workbox-core/_private/DBWrapper.js';
|
||
|
import '../_version.js';
|
||
|
const DB_VERSION = 3;
|
||
|
const DB_NAME = 'workbox-background-sync';
|
||
|
const OBJECT_STORE_NAME = 'requests';
|
||
|
const INDEXED_PROP = 'queueName';
|
||
|
/**
|
||
|
* A class to manage storing requests from a Queue in IndexedDB,
|
||
|
* indexed by their queue name for easier access.
|
||
|
*
|
||
|
* @private
|
||
|
*/
|
||
|
export class QueueStore {
|
||
|
/**
|
||
|
* Associates this instance with a Queue instance, so entries added can be
|
||
|
* identified by their queue name.
|
||
|
*
|
||
|
* @param {string} queueName
|
||
|
* @private
|
||
|
*/
|
||
|
constructor(queueName) {
|
||
|
this._queueName = queueName;
|
||
|
this._db = new DBWrapper(DB_NAME, DB_VERSION, {
|
||
|
onupgradeneeded: this._upgradeDb,
|
||
|
});
|
||
|
}
|
||
|
/**
|
||
|
* Append an entry last in the queue.
|
||
|
*
|
||
|
* @param {Object} entry
|
||
|
* @param {Object} entry.requestData
|
||
|
* @param {number} [entry.timestamp]
|
||
|
* @param {Object} [entry.metadata]
|
||
|
* @private
|
||
|
*/
|
||
|
async pushEntry(entry) {
|
||
|
if (process.env.NODE_ENV !== 'production') {
|
||
|
assert.isType(entry, 'object', {
|
||
|
moduleName: 'workbox-background-sync',
|
||
|
className: 'QueueStore',
|
||
|
funcName: 'pushEntry',
|
||
|
paramName: 'entry',
|
||
|
});
|
||
|
assert.isType(entry.requestData, 'object', {
|
||
|
moduleName: 'workbox-background-sync',
|
||
|
className: 'QueueStore',
|
||
|
funcName: 'pushEntry',
|
||
|
paramName: 'entry.requestData',
|
||
|
});
|
||
|
}
|
||
|
// Don't specify an ID since one is automatically generated.
|
||
|
delete entry.id;
|
||
|
entry.queueName = this._queueName;
|
||
|
await this._db.add(OBJECT_STORE_NAME, entry);
|
||
|
}
|
||
|
/**
|
||
|
* Prepend an entry first in the queue.
|
||
|
*
|
||
|
* @param {Object} entry
|
||
|
* @param {Object} entry.requestData
|
||
|
* @param {number} [entry.timestamp]
|
||
|
* @param {Object} [entry.metadata]
|
||
|
* @private
|
||
|
*/
|
||
|
async unshiftEntry(entry) {
|
||
|
if (process.env.NODE_ENV !== 'production') {
|
||
|
assert.isType(entry, 'object', {
|
||
|
moduleName: 'workbox-background-sync',
|
||
|
className: 'QueueStore',
|
||
|
funcName: 'unshiftEntry',
|
||
|
paramName: 'entry',
|
||
|
});
|
||
|
assert.isType(entry.requestData, 'object', {
|
||
|
moduleName: 'workbox-background-sync',
|
||
|
className: 'QueueStore',
|
||
|
funcName: 'unshiftEntry',
|
||
|
paramName: 'entry.requestData',
|
||
|
});
|
||
|
}
|
||
|
const [firstEntry] = await this._db.getAllMatching(OBJECT_STORE_NAME, {
|
||
|
count: 1,
|
||
|
});
|
||
|
if (firstEntry) {
|
||
|
// Pick an ID one less than the lowest ID in the object store.
|
||
|
entry.id = firstEntry.id - 1;
|
||
|
}
|
||
|
else {
|
||
|
// Otherwise let the auto-incrementor assign the ID.
|
||
|
delete entry.id;
|
||
|
}
|
||
|
entry.queueName = this._queueName;
|
||
|
await this._db.add(OBJECT_STORE_NAME, entry);
|
||
|
}
|
||
|
/**
|
||
|
* Removes and returns the last entry in the queue matching the `queueName`.
|
||
|
*
|
||
|
* @return {Promise<Object>}
|
||
|
* @private
|
||
|
*/
|
||
|
async popEntry() {
|
||
|
return this._removeEntry({ direction: 'prev' });
|
||
|
}
|
||
|
/**
|
||
|
* Removes and returns the first entry in the queue matching the `queueName`.
|
||
|
*
|
||
|
* @return {Promise<Object>}
|
||
|
* @private
|
||
|
*/
|
||
|
async shiftEntry() {
|
||
|
return this._removeEntry({ direction: 'next' });
|
||
|
}
|
||
|
/**
|
||
|
* Returns all entries in the store matching the `queueName`.
|
||
|
*
|
||
|
* @param {Object} options See {@link module:workbox-background-sync.Queue~getAll}
|
||
|
* @return {Promise<Array<Object>>}
|
||
|
* @private
|
||
|
*/
|
||
|
async getAll() {
|
||
|
return await this._db.getAllMatching(OBJECT_STORE_NAME, {
|
||
|
index: INDEXED_PROP,
|
||
|
query: IDBKeyRange.only(this._queueName),
|
||
|
});
|
||
|
}
|
||
|
/**
|
||
|
* Deletes the entry for the given ID.
|
||
|
*
|
||
|
* WARNING: this method does not ensure the deleted enry belongs to this
|
||
|
* queue (i.e. matches the `queueName`). But this limitation is acceptable
|
||
|
* as this class is not publicly exposed. An additional check would make
|
||
|
* this method slower than it needs to be.
|
||
|
*
|
||
|
* @private
|
||
|
* @param {number} id
|
||
|
*/
|
||
|
async deleteEntry(id) {
|
||
|
await this._db.delete(OBJECT_STORE_NAME, id);
|
||
|
}
|
||
|
/**
|
||
|
* Removes and returns the first or last entry in the queue (based on the
|
||
|
* `direction` argument) matching the `queueName`.
|
||
|
*
|
||
|
* @return {Promise<Object>}
|
||
|
* @private
|
||
|
*/
|
||
|
async _removeEntry({ direction }) {
|
||
|
const [entry] = await this._db.getAllMatching(OBJECT_STORE_NAME, {
|
||
|
direction,
|
||
|
index: INDEXED_PROP,
|
||
|
query: IDBKeyRange.only(this._queueName),
|
||
|
count: 1,
|
||
|
});
|
||
|
if (entry) {
|
||
|
await this.deleteEntry(entry.id);
|
||
|
return entry;
|
||
|
}
|
||
|
}
|
||
|
/**
|
||
|
* Upgrades the database given an `upgradeneeded` event.
|
||
|
*
|
||
|
* @param {Event} event
|
||
|
* @private
|
||
|
*/
|
||
|
_upgradeDb(event) {
|
||
|
const db = event.target.result;
|
||
|
if (event.oldVersion > 0 && event.oldVersion < DB_VERSION) {
|
||
|
if (db.objectStoreNames.contains(OBJECT_STORE_NAME)) {
|
||
|
db.deleteObjectStore(OBJECT_STORE_NAME);
|
||
|
}
|
||
|
}
|
||
|
const objStore = db.createObjectStore(OBJECT_STORE_NAME, {
|
||
|
autoIncrement: true,
|
||
|
keyPath: 'id',
|
||
|
});
|
||
|
objStore.createIndex(INDEXED_PROP, INDEXED_PROP, { unique: false });
|
||
|
}
|
||
|
}
|