func loadStatements() {
var queries = `
-- name: drop-database
DROP DATABASE IF EXISTS trigramnews;
-- name: create-database
CREATE DATABASE IF NOT EXISTS trigramnews;
-- name: use-database
USE trigramnews;
-- name: create-titles-table
CREATE TABLE IF NOT EXISTS titles (
id INTEGER NOT NULL AUTO_INCREMENT,
title VARCHAR(255) NOT NULL,
PRIMARY KEY (id)
);
-- name: create-trigrams-table
CREATE TABLE IF NOT EXISTS trigrams (
id INTEGER NOT NULL,
trigram CHARACTER(3) NOT NULL,
occurrence INTEGER NOT NULL DEFAULT 1,
UNIQUE (id, trigram),
FOREIGN KEY (id) REFERENCES titles(id)
);
-- name: insert-title
INSERT INTO titles (title) VALUES(?);
-- name: insert-trigram
INSERT INTO trigrams (id, trigram, occurrence) VALUES(?, ?, 1)
ON DUPLICATE KEY UPDATE occurrence = occurrence + 1;
-- name: select-titleids-from-trigram
SELECT id FROM trigrams WHERE trigram = ?;
-- name: select-title
SELECT title FROM titles WHERE id = ? LIMIT 1;
-- name: select-titleid-by-name
SELECT id FROM titles WHERE title = ? LIMIT 1;
-- name: select-all-trigrams-by-title
SELECT tg.trigram, tg.occurrence FROM trigrams as tg, titles as ti WHERE tg.id = ti.id AND ti.title = ?;
-- name: count-all-titles
SELECT count(id) FROM titles
`
dot, err := dotsql.LoadFromString(queries)
if err != nil {
panic(err)
}
statementsFile = dot
}
func BenchmarkDotSQL(b *testing.B) {
db := test.db
dot, err := dotsql.LoadFromString(`
-- name: select
SELECT * from users`)
panicIfErr(err)
var t1, t2 string
for i := 0; i < b.N; i++ {
rows, err := dot.Query(db, "select")
panicIfErr(err)
for rows.Next() {
err := rows.Scan(&t1, &t2)
panicIfErr(err)
}
}
}