rationalsied db calls

This commit is contained in:
Andrzej Stepien 2023-08-13 14:59:55 +02:00
parent 9bbc98c0c3
commit 8c9254598d
3 changed files with 77 additions and 54 deletions

View File

@ -1,6 +1,7 @@
import Knex from 'knex'
import logger from '../logger.mjs'
import { isoDate } from "../utilities.mjs"
import config from "../config.mjs"
export const db = Knex({
client: 'sqlite3', // or 'better-sqlite3'
@ -9,6 +10,36 @@ export const db = Knex({
},
useNullAsDefault: true
})
const blocklist = db.union([
db('bad_words').select('word'),
db('medical_dictionary').select('word'),
db('published').select('word')
])
export const getAcceptablePrompts = async (word) => {
logger.trace("getAcceptablePrompt called")
try {
return db('dictionary')
.select('*')
.where({
derivative: 0,
scientific: 0,
})
.modify(queryBuilder=>{
if(word){queryBuilder.andWhere('word',word)}
})
.andWhere('count', '<', config.maxCount)
.andWhere('count', '>', config.minCount)
.andWhere('word', 'not in', blocklist)
.whereRaw('length(word) > 3')
.whereNotNull('pronunciation')
.orderByRaw('count desc')
} catch (error) {
logger.error("getAcceptablePrompts failed!")
throw error
}
}
export const getWords = async () => {
logger.trace("getWords called")
@ -18,14 +49,18 @@ export const getWords = async () => {
.catch(error => { throw error })
}
export const valueExistsInTable = async (table, column, value) => {
export const valueExistsInColumn = async (table, column, value) => {
try {
const number = await db(table)
.count('* as count')
.where(column, value)
return number[0].count > 0
return number[0].count > 0
} catch (error) {
logger.error("valueExistsInTable failed!")
logger.error("valueExistsInColumn failed!")
throw error
}
@ -33,9 +68,44 @@ export const valueExistsInTable = async (table, column, value) => {
export const todaysPromptAlreadyPublished = async () => {
return valueExistsInTable('published', 'date', isoDate())
return valueExistsInColumn('published', 'date', isoDate())
}
export const wordIsAlreadyInBuffer = async (word) => {
return valueExistsInTable('buffer', 'word', word)
return valueExistsInColumn('buffer', 'word', word)
}
export const tableIsNotEmpty = async (table) => {
try {
const number = await db(table)
.count('* as count')
return number[0].count > 0
} catch (error) {
throw error
}
}
export const getPromptFromBuffer = async () => {
logger.trace("getting prompt from buffer")
const oldestWordInBuffer = await db('buffer').select('word').orderBy('timestamp', 'asc').limit(1)
logger.info(`oldest word in buffer: ${oldestWordInBuffer[0].word}`)
try {
const prompt = await getAcceptablePrompts(oldestWordInBuffer[0].word)
return prompt[0]
} catch (error) {
logger.error("getPromptFromBuffer failed!")
throw error
}
}
export const deleteFromBuffer = async (word) => {
logger.trace(`deleteFromBuffer called for word ${word}!`)
try {
return db('buffer')
.where('word', word)
.del()
} catch (error) {
logger.error("deleteFromBuffer failed!")
}
}

View File

@ -1,30 +0,0 @@
import logger from "../logger.mjs"
import { db } from "./db.mjs"
import config from "../config.mjs"
const blocklist = db.union([
db('bad_words').select('word'),
db('medical_dictionary').select('word'),
db('published').select('word')
])
export default async () => {
logger.trace("getAcceptablePrompt called")
try {
return db('dictionary')
.select('*')
.where({
derivative: 0,
scientific: 0,
})
.andWhere('count', '<', config.maxCount)
.andWhere('count', '>', config.minCount)
.andWhere('word', 'not in', blocklist)
.whereRaw('length(word) > 3')
.whereNotNull('pronunciation')
.orderByRaw('count desc')
} catch (error) {
logger.error("getAcceptablePrompts failed!")
throw error
}
}

View File

@ -1,17 +0,0 @@
import { db } from "./db.mjs";
import { isoDate } from "../utilities.mjs";
import logger from "../logger.mjs";
export default async function todaysPromptAlreadyPublished() {
try {
const number = await db('published')
.count('* as count')
.where('date', isoDate())
return number[0].count > 0
} catch (error) {
logger.error("todaysPromptAlreadyPublished failed!")
throw error
}
}
//console.log(await todaysPromptAlreadyPublished())