Lua（数据库访问）

Lua 数据库访问方法

Lua 本身不提供内置的数据库访问功能，但可以通过第三方库实现与多种数据库的交互。以下是常见的 Lua 数据库访问方法：

---

使用 LuaSQL 库

LuaSQL 是一个轻量级数据库访问库，支持多种数据库后端（MySQL、PostgreSQL、SQLite、ODBC 等）。

安装方式：

luarocks install luasql-mysql  # MySQL
luarocks install luasql-sqlite3 # SQLite

示例代码（MySQL）：

local luasql = require "luasql.mysql"
local env = luasql.mysql()
local conn = env:connect("database", "username", "password", "host", port)

-- 执行查询
local cursor = conn:execute("SELECT * FROM users")
local row = cursor:fetch({}, "a")
while row do
    print(row.id, row.name)
    row = cursor:fetch(row, "a")
end

-- 关闭连接
cursor:close()
conn:close()
env:close()

---

使用 SQLite 的 lsqlite3 库

对于嵌入式数据库 SQLite，可以使用 lsqlite3 库。

安装方式：

luarocks install lsqlite3

示例代码：

local sqlite3 = require "lsqlite3"
local db = sqlite3.open("test.db")

-- 执行查询
db:exec[[
    CREATE TABLE test (id INTEGER PRIMARY KEY, content TEXT);
    INSERT INTO test VALUES (NULL, 'Hello World');
]]

-- 读取数据
for row in db:nrows("SELECT * FROM test") do
    print(row.id, row.content)
end

db:close()

---

使用 Redis 的 lua-resty-redis 库

如果需要访问 Redis，可以使用 OpenResty 提供的 lua-resty-redis 库。

安装方式：

luarocks install lua-resty-redis

示例代码：

local redis = require "resty.redis"
local red = redis:new()

red:set_timeout(1000) -- 1 second
local ok, err = red:connect("127.0.0.1", 6379)
if not ok then
    ngx.say("failed to connect: ", err)
    return
end

-- 设置和获取值
red:set("dog", "an animal")
local res, err = red:get("dog")
ngx.say(res)  -- 输出 "an animal"

-- 关闭连接
red:close()

---

使用通用数据库驱动（ODBC）

对于需要通过 ODBC 连接的数据库，可以使用 LuaSQL 的 ODBC 驱动。

安装方式：

luarocks install luasql-odbc

示例代码：

local luasql = require "luasql.odbc"
local env = luasql.odbc()
local conn = env:connect("DSN=mydsn;UID=user;PWD=pass")

-- 执行查询
local cursor = conn:execute("SELECT * FROM products")
-- 处理结果...
cursor:close()
conn:close()
env:close()

---

使用 ORM 框架（如 LuaORM）

对于更高级的数据库操作，可以使用 ORM 框架如 LuaORM。

安装方式：

luarocks install luaorm

示例代码：

local orm = require "luaorm"
orm.init("sqlite3", "test.db")

local User = orm.define("users", {
    { name = "id", type = "integer", primary = true },
    { name = "name", type = "text" }
})

-- 创建表
User:create_table()

-- 插入数据
local new_user = User:new{ name = "Alice" }
new_user:save()

-- 查询数据
for _, user in ipairs(User:find_all()) do
    print(user.id, user.name)
end

---

注意事项

1. 连接池管理：在高并发场景下，建议使用连接池管理数据库连接，避免频繁创建和销毁连接。
2. 错误处理：所有数据库操作都应包含错误处理逻辑，确保程序健壮性。
3. SQL 注入：避免直接拼接 SQL 语句，使用参数化查询防止注入攻击。
4. 性能优化：批量操作数据时，使用事务可以提高性能。

这些方法覆盖了 Lua 中访问常见数据库的主要方式，开发者可以根据项目需求选择合适的方案。