sqliteviz/src/lib/database/_sql.js

import initSqlJs from 'sql.js/dist/sql-wasm.js'
import dbUtils from './_statements'

let SQL = null
const sqlModuleReady = initSqlJs().then(sqlModule => { SQL = sqlModule })

function _getDataSourcesFromSqlResult (sqlResult) {
  if (!sqlResult) {
    return {}
  }
  const dataSorces = {}
  sqlResult.columns.forEach((column, index) => {
    dataSorces[column] = sqlResult.values.map(row => row[index])
  })
  return dataSorces
}

export default class Sql {
  constructor () {
    this.db = null
  }

  static build () {
    return sqlModuleReady
      .then(() => {
        return new Sql()
      })
  }

  createDb (buffer) {
    if (this.db != null) this.db.close()
    this.db = new SQL.Database(buffer)
    return this.db
  }

  open (buffer) {
    this.createDb(buffer && new Uint8Array(buffer))
    return {
      ready: true
    }
  }

  exec (sql, params) {
    if (this.db === null) {
      this.createDb()
    }
    if (!sql) {
      throw new Error('exec: Missing query string')
    }
    const sqlResults = this.db.exec(sql, params)
    return sqlResults.map(result => {
      return {
        columns: result.columns,
        values: _getDataSourcesFromSqlResult(result)
      }
    })
  }

  import (tabName, data, progressCounterId, progressCallback, chunkSize = 1500) {
    if (this.db === null) {
      this.createDb()
    }
    const columns = data.columns
    const rowCount = data.values[columns[0]].length
    this.db.exec(dbUtils.getCreateStatement(tabName, data.values))
    const chunks = dbUtils.generateChunks(data.values, chunkSize)
    const chunksAmount = Math.ceil(rowCount / chunkSize)
    let count = 0
    const insertStr = dbUtils.getInsertStmt(tabName, columns)
    const insertStmt = this.db.prepare(insertStr)

    progressCallback({ progress: 0, id: progressCounterId })
    for (const chunk of chunks) {
      this.db.exec('BEGIN')
      for (const row of chunk) {
        insertStmt.run(row)
      }
      this.db.exec('COMMIT')
      count++
      progressCallback({ progress: 100 * (count / chunksAmount), id: progressCounterId })
    }

    return {
      finish: true
    }
  }

  export () {
    return this.db.export()
  }

  close () {
    if (this.db) {
      this.db.close()
    }
    return {
      finished: true
    }
  }
}