File size: 1,999 Bytes
4bbfe8b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | import { Client } from "@planetscale/database"
import { Resource } from "sst/resource"
const tables = ["geo_stat", "model_stat", "provider_stat"] as const
const checkOnly = process.argv.includes("--check")
const client = new Client({ url: databaseUrl() })
const missing = await tables.reduce<Promise<(typeof tables)[number][]>>(async (promise, table) => {
const result = await promise
if (await hasUniqueUsersColumn(table)) {
console.log(`unique_users column already exists on ${table}`)
return result
}
return [...result, table]
}, Promise.resolve([]))
if (missing.length === 0) {
console.log("unique_users columns complete")
process.exit(0)
}
if (checkOnly) {
console.log(`unique_users columns missing on ${missing.join(", ")}`)
process.exit(1)
}
await missing.reduce((promise, table) => promise.then(() => addUniqueUsersColumn(table)), Promise.resolve())
function databaseUrl() {
if (
process.env.PLANETSCALE_HOST &&
process.env.PLANETSCALE_USERNAME &&
process.env.PLANETSCALE_PASSWORD &&
process.env.PLANETSCALE_DATABASE
)
return `mysql://${encodeURIComponent(process.env.PLANETSCALE_USERNAME)}:${encodeURIComponent(
process.env.PLANETSCALE_PASSWORD,
)}@${process.env.PLANETSCALE_HOST}/${process.env.PLANETSCALE_DATABASE}?ssl=${encodeURIComponent(
JSON.stringify({ rejectUnauthorized: true }),
)}`
return process.env.DATABASE_URL ?? Resource.StatsDatabase.url
}
async function hasUniqueUsersColumn(table: (typeof tables)[number]) {
const result = await client.execute<{ column_name: string }>(
"SELECT column_name FROM information_schema.columns WHERE table_schema = database() AND table_name = ? AND column_name = 'unique_users'",
[table],
)
return result.rows.length > 0
}
async function addUniqueUsersColumn(table: (typeof tables)[number]) {
await client.execute(`ALTER TABLE \`${table}\` ADD \`unique_users\` bigint NOT NULL DEFAULT 0`)
console.log(`added unique_users column to ${table}`)
}
|