feat(upload): spawn worker dynamically, when uploading

This commit is contained in:
Kasper Seweryn 2023-12-05 11:19:26 +01:00 committed by Georg Krause
parent e458ad3817
commit 287bca1a21
3 changed files with 67 additions and 62 deletions

View File

@ -1,6 +1,5 @@
<script setup lang="ts">
import { reactive, ref, computed } from 'vue'
import { whenever, computedAsync, useWebWorkerFn } from '@vueuse/core'
import { UseTimeAgo } from '@vueuse/components'
import { Icon } from '@iconify/vue';
import { useUploadsStore } from '~/ui/stores/upload'
@ -52,31 +51,28 @@ const libraryModalAlertOpen = ref(true)
const serverPath = ref('/srv/funkwhale/data/music')
// Upload
const files = ref<File[]>([])
const combinedFileSize = computed(() => bytesToHumanSize(
files.value.reduce((acc, file) => acc + file.size, 0)
uploads.queue.reduce((acc, { file }) => acc + file.size, 0)
))
const uploads = useUploadsStore()
const processFiles = (fileList: FileList) => {
console.log('processFiles', fileList)
// NOTE: Append fileList elements in reverse order so they appear in the UI in the order, user selected them
for (const file of Array.from(fileList).reverse()) {
files.value.push(file)
}
for (const file of fileList) {
uploads.queueUpload(file)
}
}
const cancel = () => {
libraryOpen.value = false
uploads.cancelAll()
}
</script>
<template>
<div class="flex items-center">
<h1>Upload</h1>
<div class="flex-spacer" />
<h1 class="mr-auto">Upload</h1>
<div class="filesystem-stats">
<div class="filesystem-stats--progress" :style="`--progress: ${filesystemProgress}%`" />
@ -95,7 +91,7 @@ const processFiles = (fileList: FileList) => {
<p> Select a destination for your audio files: </p>
<div class="flex space-between">
<div class="flex justify-between">
<FwCard
v-for="tab in tabs" :key="tab.label"
:title="tab.label"
@ -134,10 +130,10 @@ const processFiles = (fileList: FileList) => {
/>
<!-- Upload path -->
<div v-if="files && files.length > 0">
<div v-if="uploads.queue.length > 0">
<div class="list-header">
<div class="file-count">
{{ files.length }} files, {{ combinedFileSize }}
{{ uploads.queue.length }} files, {{ combinedFileSize }}
</div>
<FwButton color="secondary">All</FwButton>
@ -212,7 +208,7 @@ const processFiles = (fileList: FileList) => {
</template>
<template #actions>
<FwButton @click="libraryOpen = false" color="secondary">Cancel</FwButton>
<FwButton @click="cancel" color="secondary">Cancel</FwButton>
<FwButton @click="libraryOpen = false">
{{ uploads.queue.length ? 'Continue in background' : 'Save and close' }}
</FwButton>
@ -228,35 +224,18 @@ h1 {
font-family: Lato, sans-serif;
}
.flex {
display: flex;
align-items: flex-start;
.flex:not(.flex-col) {
.funkwhale.button {
&:first-child {
margin-left: 0;
}
&:not(.flex-col) {
.funkwhale.button {
&:first-child {
margin-left: 0;
}
&:last-child {
margin-right: 0;
}
&:last-child {
margin-right: 0;
}
}
}
.items-center {
align-items: center;
}
.space-between {
justify-content: space-between;
}
.flex-spacer {
flex: 1;
}
.filesystem-stats {
color: var(--fw-gray-700);
> .flex {
@ -348,14 +327,6 @@ h1 {
}
}
.w-full {
width: 100%;
}
.mr-4 {
margin-right: 1rem;
}
label {
line-height: 1.2;
font-weight: 900;
@ -467,6 +438,4 @@ label {
border-color: transparent !important;
}
}
</style>

View File

@ -1,9 +1,11 @@
import { defineStore, acceptHMRUpdate } from 'pinia'
import { computed, reactive, readonly, ref, watchEffect, markRaw, toRaw } from 'vue'
import { whenever, useWebWorker } from '@vueuse/core'
import { computed, reactive, readonly, ref, watchEffect, markRaw, toRaw, type Ref } from 'vue'
import { whenever, useWebWorker, type UseWebWorkerReturn } from '@vueuse/core'
import { not } from '@vueuse/math'
import axios from 'axios'
import FileMetadataParserWorker from '~/ui/workers/file-metadata-parser.ts?worker'
import FileMetadataParserWorker, { type MetadataParsingResult } from '~/ui/workers/file-metadata-parser.ts?worker'
import { getCoverUrl, getTags, type Tags } from '~/ui/composables/metadata'
@ -28,9 +30,25 @@ interface UploadQueueEntry {
}
export const useUploadsStore = defineStore('uploads', () => {
const uploadQueue: UploadQueueEntry[] = reactive([])
const currentIndex = ref(0)
const currentUpload = computed(() => uploadQueue[currentIndex.value])
const isUploading = computed(() => !!currentUpload.value)
// Tag extraction with a Web Worker
const { post: retrieveMetadata, data: workerData, worker } = useWebWorker(FileMetadataParserWorker)
whenever(workerData, (reactiveData) => {
const worker = ref<UseWebWorkerReturn<MetadataParsingResult>>()
const retrieveMetadata = (entry: Pick<UploadQueueEntry, 'id' | 'file'>) => {
if (!worker.value) worker.value = useWebWorker<MetadataParsingResult>(FileMetadataParserWorker)
worker.value.post(entry)
}
whenever(not(isUploading), () => {
worker.value?.terminate()
worker.value = undefined
})
whenever(() => worker.value?.data, (reactiveData) => {
const data = toRaw(reactiveData)
if (data.status === 'success') {
const id = data.id as number
@ -72,8 +90,6 @@ export const useUploadsStore = defineStore('uploads', () => {
}
})
// TODO: Handle failure with a try/catch block
console.log(`[${entry.id}] import complete!`)
entry.importedAt = new Date()
}
@ -91,11 +107,6 @@ export const useUploadsStore = defineStore('uploads', () => {
retrieveMetadata({ id, file })
}
const uploadQueue: UploadQueueEntry[] = reactive([])
const currentIndex = ref(0)
const currentUpload = computed(() => uploadQueue[currentIndex.value])
const isUploading = computed(() => !!currentUpload.value)
// Upload the file whenever it is available
whenever(currentUpload, (entry) => upload(entry).catch((error) => {
// The tags were missing, so we have cancelled the upload
@ -120,11 +131,21 @@ export const useUploadsStore = defineStore('uploads', () => {
}
})
const cancelAll = () => {
for (const upload of uploadQueue) {
upload.abortController.abort()
}
uploadQueue.length = 0
currentIndex.value = 0
}
// Return public API
return {
isUploading,
queueUpload,
currentUpload,
cancelAll,
queue: readonly(uploadQueue)
}
})

View File

@ -1,6 +1,21 @@
/// <reference lib="webworker" />
import { getCoverUrl, getTags } from '~/ui/composables/metadata'
import { getCoverUrl, getTags, type Tags } from '~/ui/composables/metadata'
export interface MetadataParsingSuccess {
id: number
status: 'success'
tags: Tags
coverUrl: string | undefined
}
export interface MetadataParsingFailure {
id: number
status: 'failure'
error: Error
}
export type MetadataParsingResult = MetadataParsingSuccess | MetadataParsingFailure
const parse = async (id: number, file: File) => {