railbird/railbird-gql
9
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Queue uploads

Browse Source
This commit is contained in:
Ivan Malison 2024-02-05 01:42:32 -07:00
parent f44a1597c0
commit 000918844a
1 changed files with 59 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

94
component/video/camera.tsx
View File

@ -24,55 +24,79 @@ type Dictionary<KeyType extends string | number | symbol, ValueType> = {
class StreamUploadManager<TCacheShape> { class StreamUploadManager<TCacheShape> {
client: ApolloClient<TCacheShape>; client: ApolloClient<TCacheShape>;
videoId: number; videoId: number;
nextUploadIdToRequest: number; nextUploadIdToRequest: number = 0;
highestUploadLinkObtained: number; highestUploadLinkObtained: number = -1;
prefetchedUploadLinks: Dictionary<number, string>; prefetchedUploadLinks: Dictionary<number, string> = {};
uploadQueue: Array<() => Promise<void>> = [];
isUploading: boolean = false;
constructor(client: ApolloClient<TCacheShape>, streamId: number) { constructor(client: ApolloClient<TCacheShape>, videoId: number) {
this.client = client; this.client = client;
this.videoId = streamId; this.videoId = videoId;
this.highestUploadLinkObtained = -1;
this.prefetchedUploadLinks = {};
} }
async uploadChunk({ filepath, index }: { filepath: string; index: number }) { enqueueUploadTask(task: () => Promise<void>) {
const uploadUrl = await this.getUploadLink(index); this.uploadQueue.push(task);
const uploadRequest = RNFS.uploadFiles({ this.processUploadQueue();
toUrl: uploadUrl, }
files: [{ filepath: filepath }],
method: "PUT", async processUploadQueue() {
headers: { if (this.isUploading || this.uploadQueue.length === 0) {
"Content-Type": "application/octet-stream", return;
}, }
begin: (res) => { this.isUploading = true;
console.log("Start upload", res); const task = this.uploadQueue.shift();
}, try {
progress: (res) => { if (task) await task();
console.log("Uploading", res); } catch (error) {
}, console.error("Error processing upload task", error);
}); } finally {
console.log(JSON.stringify(uploadRequest)); this.isUploading = false;
const result = await uploadRequest.promise.catch((err) => this.processUploadQueue();
console.log("Upload error!", err),
);
if (response.statusCode == 200) {
console.log(`${filepath} Uploaded`);
} else {
console.log("SERVER ERROR");
} }
} }
async uploadChunk({ filepath, index }: { filepath: string; index: number }) {
this.enqueueUploadTask(async () => {
const uploadUrl = await this.getUploadLink(index);
const uploadRequest = RNFS.uploadFiles({
toUrl: uploadUrl,
files: [{ filepath: filepath }],
method: "PUT",
headers: {
"Content-Type": "application/octet-stream",
},
begin: (res) => {
console.log("Start upload", res);
},
progress: (res) => {
console.log("Uploading", res);
},
});
console.log(JSON.stringify(uploadRequest));
const result = await uploadRequest.promise;
if (result.statusCode === 200) {
console.log(`${filepath} Uploaded`);
} else {
console.error("SERVER ERROR");
}
});
}
async getUploadLink(chunkId: number): Promise<string> { async getUploadLink(chunkId: number): Promise<string> {
if (this.prefetchedUploadLinks[chunkId]) {
return this.prefetchedUploadLinks[chunkId];
}
return this.requestUploadLink(chunkId); return this.requestUploadLink(chunkId);
} }
async requestUploadLink(chunkId: number): Promise<string> { async requestUploadLink(chunkId: number): Promise<string> {
console.log(`Requesting ${chunkId}`); console.log(`Requesting upload link for chunk ${chunkId}`);
const result = await this.client.mutate({ const result = await this.client.mutate({
mutation: gql.GetUploadLinkDocument, mutation: gql.GetUploadLinkDocument,
variables: { videoId: this.videoId, chunkIndex: chunkId }, variables: { videoId: this.videoId, chunkIndex: chunkId },
}); });
console.log(JSON.stringify(result.data)); this.prefetchedUploadLinks[chunkId] = result.data.getUploadLink.uploadUrl;
return result.data.getUploadLink.uploadUrl; return result.data.getUploadLink.uploadUrl;
} }
} }
@ -140,8 +164,8 @@ export default function CameraScreen({
const device = useCameraDevice("back"); const device = useCameraDevice("back");
const format = useCameraFormat(device, [ const format = useCameraFormat(device, [
{ videoResolution: { width: 3048, height: 2160 } }, { videoResolution: { width: 1920, height: 1080 } },
{ fps: 60 }, { fps: 30 },
]); ]);
// TODO(#60): setOrientation should be called when changes are detected // TODO(#60): setOrientation should be called when changes are detected