Dimitrie Stefanescu
115 lines
3.4 KiB
JavaScript
115 lines
3.4 KiB
JavaScript
/**
|
|
* Simple client that streams object info from a Speckle Server.
|
|
* TODO: This should be split from the viewer into its own package.
|
|
*/
|
|
|
|
export default class ObjectLoader {
|
|
|
|
constructor( { serverUrl, streamId, token, objectId } ) {
|
|
this.INTERVAL_MS = 20
|
|
this.TIMEOUT_MS = 180000 // three mins
|
|
|
|
this.serverUrl = serverUrl || window.location.origin
|
|
this.streamId = streamId
|
|
this.objectId = objectId
|
|
this.token = token || localStorage.getItem( 'AuthToken' )
|
|
|
|
this.headers = {
|
|
'Accept': 'text/plain'
|
|
}
|
|
|
|
if( token ) {
|
|
this.headers['Authorization'] = `Bearer ${this.token}`
|
|
}
|
|
|
|
this.requestUrl = `${this.serverUrl}/objects/${this.streamId}/${this.objectId}`
|
|
this.promises = []
|
|
this.intervals = {}
|
|
this.buffer = []
|
|
}
|
|
|
|
dispose() {
|
|
this.buffer = []
|
|
this.intervals.forEach( i => clearInterval( i.interval ) )
|
|
}
|
|
|
|
async getObject( id ){
|
|
if ( this.buffer[id] ) return this.buffer[id]
|
|
|
|
let promise = new Promise( ( resolve, reject ) => {
|
|
this.promises.push( { id, resolve, reject } )
|
|
// Only create a new interval checker if none is already present!
|
|
if ( this.intervals[id] ) {
|
|
this.intervals[id].elapsed = 0 // reset elapsed
|
|
} else {
|
|
let intervalId = setInterval( this.tryResolvePromise.bind( this ), this.INTERVAL_MS, id )
|
|
this.intervals[id] = { interval: intervalId, elapsed: 0 }
|
|
}
|
|
} )
|
|
return promise
|
|
}
|
|
|
|
tryResolvePromise( id ) {
|
|
this.intervals[id].elapsed += this.INTERVAL_MS
|
|
if ( this.buffer[id] ) {
|
|
for ( let p of this.promises.filter( p => p.id === id ) ) {
|
|
p.resolve( this.buffer[id] )
|
|
}
|
|
|
|
clearInterval( this.intervals[id].interval )
|
|
delete this.intervals[id]
|
|
// this.promises = this.promises.filter( p => p.id !== p.id ) // clearing out promises too early seems to nuke loading
|
|
return
|
|
}
|
|
|
|
if ( this.intervals[id].elapsed > this.TIMEOUT_MS ) {
|
|
console.warn( `Timeout resolving ${id}. HIC SVNT DRACONES.` )
|
|
clearInterval( this.intervals[id].interval )
|
|
this.promises.filter( p => p.id === id ).forEach( p => p.reject() )
|
|
this.promises = this.promises.filter( p => p.id !== p.id ) // clear out
|
|
}
|
|
}
|
|
|
|
async * getObjectIterator( ) {
|
|
for await ( let line of this.getRawObjectIterator() ) {
|
|
let { id, obj } = this.processLine( line )
|
|
this.buffer[ id ] = obj
|
|
yield obj
|
|
}
|
|
}
|
|
|
|
processLine( chunk ) {
|
|
var pieces = chunk.split( '\t' )
|
|
return { id: pieces[0], obj: JSON.parse( pieces[1] ) }
|
|
}
|
|
|
|
async * getRawObjectIterator() {
|
|
const decoder = new TextDecoder()
|
|
const response = await fetch( this.requestUrl, { headers: this.headers } )
|
|
const reader = response.body.getReader()
|
|
let { value: chunk, done: readerDone } = await reader.read()
|
|
chunk = chunk ? decoder.decode( chunk ) : ''
|
|
|
|
let re = /\r\n|\n|\r/gm
|
|
let startIndex = 0
|
|
|
|
while ( true ) {
|
|
let result = re.exec( chunk )
|
|
if ( !result ) {
|
|
if ( readerDone ) break
|
|
let remainder = chunk.substr( startIndex );
|
|
( { value: chunk, done: readerDone } = await reader.read() )
|
|
chunk = remainder + ( chunk ? decoder.decode( chunk ) : '' )
|
|
startIndex = re.lastIndex = 0
|
|
continue
|
|
}
|
|
yield chunk.substring( startIndex, result.index )
|
|
startIndex = re.lastIndex
|
|
}
|
|
|
|
if ( startIndex < chunk.length ) {
|
|
yield chunk.substr( startIndex )
|
|
}
|
|
}
|
|
}
|