コード例 #1
0
ファイル: data.go プロジェクト: ibiBgOR/GoTrigramNews
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
}
コード例 #2
0
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)
		}
	}
}