如何将 Ruby 函数放入 SQLite3 查询中？

Question

我有一个需要放入 SQLite3 查询的函数。

我有以下方法：

def levenshtein(a, b)
  case
    when a.empty? then b.length
    when b.empty? then a.length
    else [(a[0] == b[0] ? 0 : 1) + levenshtein(a[1..-1], b[1..-1]),
      1 + levenshtein(a[1..-1], b),
      1 + levenshtein(a, b[1..-1])].min
  end
end

我想做一个看起来像这样的查询：

@results = The_db.where('levenshtein("name", ?) < 3', '#{userinput}')

我想在The_db 中找到name 的值，其中name 列的值与用户输入之间的编辑距离小于3。问题是我不知道如何在查询中使用 Ruby 函数。这甚至可能吗？

Answer 1

看看create_function method。您可以使用它来创建这样的自定义函数（您已经定义了 levenshtein Ruby 方法）：

db.create_function "levenshtein", 2 do |func, a, b|
  func.result = levenshtein(a, b)
end

然后您可以在 SQL 中使用它：

# first set up some data
db.execute 'create table names (name varchar(30))'
%w{Sam Ian Matt John Albert}.each do |n|
  db.execute 'insert into names values (?)', n
end

#Then use the custom function
puts db.execute('select * from names where levenshtein(name, ?) < 3', 'Jan')

在这个例子中，输出是

Sam
Ian
John

我没有在 Rails 中测试过，但我认为它应该可以工作，查询字符串只是传递到数据库。您可以使用 ActiveRecord::Base.connection.raw_connection 获取 Sqlite db 对象。 （我不知道如何配置 Rails 以便所有 ActiveRecord 连接都定义了函数——如果你在控制器中添加函数似乎确实可以工作，但这并不理想）。

已经展示了这个可以是如何完成的，我不确定它是否应该在一个网络应用程序中完成。你可能不想在生产环境中使用 Sqlite。如果您使用例如它可能会很有用。 Postgres 在生产中具有自己的 levenshtein 函数（如 the Tin Man’s answer 中所建议的那样），但希望在开发中使用 Sqlite。

Answer 2

如果您需要检索列的值以将其插入到 Ruby 函数中，那么您必须首先检索该值。

DBM 无法在您运行的代码中调用您的方法；您必须拥有该值，将其传递给您的方法，然后在辅助查询中使用结果。

或使用内置 Levenshtein function 的 DBM，如 PostgreSQL 或在纯 SQL 中定义它。