パラメータ化されたクエリ
シンプルプログラムでは基本的なクエリの実行方法を学びました。実際のアプリケーションでは、ユーザー入力や変数の値に基づいてクエリを実行することが多くなります。このページでは、安全にパラメータを扱う方法を学びます。
ldbcでは、SQLインジェクション攻撃を防ぐために、パラメータ化されたクエリを使用することを強く推奨しています。パラメータ化されたクエリを使うと、SQLコードとデータを分離し、より安全なデータベースアクセスが可能になります。
パラメータの基本
ldbcでは、SQL文にパラメータを埋め込むための2つの主要な方法があります:
- 動的パラメータ - 通常のパラメータとして使用され、SQLインジェクション攻撃を防ぐために
PreparedStatementによって処理される - 識別子のエスケープ - テーブル名・カラム名など識別子をバッククォートで安全に埋め込む(
ident関数を使用)
動的パラメータの追加
まずは、パラメーターを持たないクエリを作成します。
sql"SELECT name, email FROM user".query[(String, String)].to[List]
次にクエリをメソッドに組み込んで、ユーザーが指定するidと一致するデータのみを選択するパラメーターを追加してみましょう。文字列の補間を行うのと同じように、id引数を$idとしてSQL文に挿入します。
val id = 1
sql"SELECT name, email FROM user WHERE id = $id".query[(String, String)].to[List]
Connectorを使用してクエリを実行すると問題なく動作します。
import ldbc.connector.*
// Connectorを作成
val connector = Connector.fromDataSource(datasource)
sql"SELECT name, email FROM user WHERE id = $id"
.query[(String, String)]
.to[List]
.readOnly(connector)
ここでは何が起こっているのでしょうか?文字列リテラルをSQL文字列にドロップしているだけのように見えますが、実際にはPreparedStatementを構築しており、id値は最終的にsetIntの呼び出しによって設定されます。これにより、SQLインジェクション攻撃からアプリケーションを守ることができます。
さまざまな型のパラメータを使用できます:
val id: Int = 1
val name: String = "Alice"
val active: Boolean = true
val createdAt: LocalDateTime = LocalDateTime.now()
sql"INSERT INTO user (id, name, active, created_at) VALUES ($id, $name, $active, $createdAt)"
ldbcでは、各型に対して適切なエンコーダーが用意されており、Scala/Javaの値をSQLの値に安全に変換します。
複数のパラメータ
複数のパラメータも同様に使用できます。
val id = 1
val email = "alice@example.com"
// Connectorを使用して実行
sql"SELECT name, email FROM user WHERE id = $id AND email > $email"
.query[(String, String)]
.to[List]
.readOnly(connector)
クエリーの結合
大きなクエリを構築する場合は、複数のSQLフラグメントを結合することができます。
val baseQuery = sql"SELECT name, email FROM user"
val whereClause = sql"WHERE id > $id"
val orderClause = sql"ORDER BY name ASC"
val query = baseQuery ++ whereClause ++ orderClause
SQLヘルパー関数
ldbcは、複雑なSQL句を簡単に構築するためのヘルパー関数を多数提供しています。
IN句の扱い
SQLでよくある課題は、一連の値をIN句で使用したい場合です。ldbcではin関数を使って簡単に実装できます。
val ids = NonEmptyList.of(1, 2, 3)
// Connectorを使用して実行
(sql"SELECT name, email FROM user WHERE " ++ in("id", ids))
.query[(String, String)]
.to[List]
.readOnly(connector)
これは以下のSQLに相当します:
SELECT name, email FROM user WHERE (id IN (?, ?, ?))
idsはNonEmptyListである必要があることに注意してください。これはIN句が空であってはならないためです。
その他のヘルパー関数
ldbcでは他にも多くの便利な関数を提供しています:
VALUES句の生成
val users = NonEmptyList.of(
(1, "Alice", "alice@example.com"),
(2, "Bob", "bob@example.com")
)
(sql"INSERT INTO user (id, name, email) " ++ values(users))
WHERE句の条件
AND条件とOR条件を簡単に構築できます:
val activeFilter = sql"active = true"
val nameFilter = sql"name LIKE ${"A%"}"
val emailFilter = sql"email IS NOT NULL"
// WHERE (active = true) AND (name LIKE 'A%') AND (email IS NOT NULL)
val query1 = sql"SELECT * FROM user " ++ whereAnd(activeFilter, nameFilter, emailFilter)
// WHERE (active = true) OR (name LIKE 'A%')
val query2 = sql"SELECT * FROM user " ++ whereOr(activeFilter, nameFilter)
SET句の生成
UPDATE文のSET句を簡単に生成できます:
val name = "New Name"
val email = "new@example.com"
val updateValues = set(
sql"name = $name",
sql"email = $email",
sql"updated_at = NOW()"
)
sql"UPDATE user " ++ updateValues ++ sql" WHERE id = 1"
ORDER BY句の生成
val query = sql"SELECT * FROM user " ++ orderBy(sql"name ASC", sql"created_at DESC")
オプションの条件
条件がオプションの場合(存在しない可能性がある場合)は、Optサフィックスが付いた関数を使用できます:
val nameOpt: Option[String] = Some("Alice")
val emailOpt: Option[String] = None
val nameFilter = nameOpt.map(name => sql"name = $name")
val emailFilter = emailOpt.map(email => sql"email = $email")
// nameFilterはSome(...)、emailFilterはNoneなので、WHERE句には「name = ?」のみが含まれます
val query = sql"SELECT * FROM user " ++ whereAndOpt(nameFilter, emailFilter)
識別子のエスケープ
カラム名やテーブル名など、SQL文の構造的な部分をパラメータ化したい場合があります。このような場合、ident 関数を使用します。
動的パラメータ(通常の$value)はPreparedStatementによって処理され、クエリ文字列では?に置き換えられますが、ident は識別子をバッククォートで囲んでSQL文に直接埋め込みます。NUL文字の除去も行うため、識別子を安全に扱えます。
val column = "name"
val table = "user"
// 動的パラメータとして扱うと「SELECT ? FROM user」となってしまう
// sql"SELECT $column FROM user".query[String].to[List]
// ident を使うと「SELECT `name` FROM `user`」となる
sql"SELECT ${ident(column)} FROM ${ident(table)}".query[String].to[List]
ident の一般的な使用例:
// 動的なカラム選択
val sortColumn = "created_at"
sql"SELECT * FROM user ORDER BY ${ident(sortColumn)} DESC"
// 動的なテーブル選択
val schema = "public"
val table = "user"
sql"SELECT * FROM ${ident(schema)}.${ident(table)}"
注意: ident はバッククォートでエスケープを行いますが、信頼できる値(定数・設定値など)に対して使用することを推奨します。ユーザー入力をそのまま識別子として使用する設計は避けてください。
条件付きSQL断片
条件に応じてSQL断片を付加したい場合は、when 関数を使用します。
val limit: Option[Int] = Some(10)
sql"SELECT name, email FROM user" ++ when(limit.isDefined)(sql" LIMIT ${limit.get}")
when(condition)(fragment) は condition が true のときだけ fragment を追加します。false のときは空の断片になります。
複数の条件を組み合わせることもできます:
val nameFilter: Option[String] = Some("Alice")
val activeOnly: Boolean = true
val query =
sql"SELECT * FROM user" ++
when(nameFilter.isDefined)(sql" WHERE name = ${nameFilter.get}") ++
when(activeOnly)(sql" AND active = true")
ページネーション
一覧取得でよく使われる LIMIT / OFFSET を簡潔に書くには、paginate 関数を使用します。
// limit と offset を両方指定
sql"SELECT name, email FROM user " ++ paginate(limit = 20, offset = 40)
// → SELECT name, email FROM user LIMIT ? OFFSET ?
// limit のみ指定
sql"SELECT name, email FROM user " ++ paginate(limit = 20)
// → SELECT name, email FROM user LIMIT ?
ページ番号からオフセットを計算する場合の例:
val pageSize = 20
val page = 3 // 1始まり
sql"SELECT name, email FROM user ORDER BY id " ++ paginate(limit = pageSize, offset = (page - 1) * pageSize)
注意:limitまたはoffsetに負の値を渡すとIllegalArgumentExceptionがスローされます。
次のステップ
これでパラメータ化されたクエリの使い方を理解できました。パラメータを扱えるようになると、より複雑で実用的なデータベースクエリを構築できるようになります。
次はデータ選択に進み、データをさまざまな形式で取得する方法を学びましょう。