Lua 常用库系列：第五篇【数据库库】

萧山财神

Lua 常用库系列：第五篇【数据库库】

在现代应用开发中，数据库操作是一个不可或缺的部分。Lua 提供了多种库用于与数据库进行交互，常见的包括 LuaSQL、luasqlite3 等。本文将详细介绍 Lua 的数据库库及其常用函数和使用方法。

数据库库概述

LuaSQL 是一个基于 Lua 的数据库接口库，支持多种数据库，包括 MySQL、PostgreSQL、SQLite 等。本文主要介绍如何使用 LuaSQL 进行数据库操作。

一、LuaSQL 安装

在使用 LuaSQL 之前，需要先安装该库。可以使用 LuaRocks 进行安装：

luarocks install luasql-mysql
luarocks install luasql-postgres
luarocks install luasql-sqlite3

安装完成后，即可在 Lua 脚本中引用 LuaSQL 库。

二、MySQL 数据库操作

以下示例演示如何使用 LuaSQL 连接 MySQL 数据库并执行基本的数据库操作。

1. 连接数据库

local luasql = require "luasql.mysql"
​
local env = luasql.mysql()
local conn = env:connect("database_name", "username", "password", "hostname", port)
​
if conn then
    print("成功连接到数据库")
else
    print("连接数据库失败")
end

2. 执行查询

local cursor = conn:execute("SELECT * FROM table_name")
​
-- 遍历结果集
local row = cursor:fetch({}, "a")
while row do
    print(string.format("id: %d, name: %s", row.id, row.name))
    row = cursor:fetch({}, "a")
end
​
cursor:close()

3. 插入数据

local res = conn:execute("INSERT INTO table_name (id, name) VALUES (1, 'Lua')")
if res then
    print("数据插入成功")
else
    print("数据插入失败")
end

4. 更新数据

local res = conn:execute("UPDATE table_name SET name = 'LuaSQL' WHERE id = 1")
if res then
    print("数据更新成功")
else
    print("数据更新失败")
end

5. 删除数据

local res = conn:execute("DELETE FROM table_name WHERE id = 1")
if res then
    print("数据删除成功")
else
    print("数据删除失败")
end

6. 关闭连接

conn:close()
env:close()三、SQLite 数据库操作

SQLite 是一种轻量级、嵌入式的数据库，广泛应用于移动设备和小型应用。以下示例演示如何使用 LuaSQL 连接 SQLite 数据库并执行基本的数据库操作。

1. 连接数据库

local luasql = require "luasql.sqlite3"
​
local env = luasql.sqlite3()
local conn = env:connect("database_name.db")
​
if conn then
    print("成功连接到数据库")
else
    print("连接数据库失败")
end

2. 执行查询

local cursor = conn:execute("SELECT * FROM table_name")

-- 遍历结果集
local row = cursor:fetch({}, "a")
while row do
    print(string.format("id: %d, name: %s", row.id, row.name))
    row = cursor:fetch({}, "a")
end

cursor:close()

3. 插入数据

local res = conn:execute("INSERT INTO table_name (id, name) VALUES (1, 'Lua')")
if res then
    print("数据插入成功")
else
    print("数据插入失败")
end

4. 更新数据

local res = conn:execute("UPDATE table_name SET name = 'LuaSQL' WHERE id = 1")
if res then
    print("数据更新成功")
else
    print("数据更新失败")
end

5. 删除数据

local res = conn:execute("DELETE FROM table_name WHERE id = 1")
if res then
    print("数据删除成功")
else
    print("数据删除失败")
end

6. 关闭连接

conn:close()
env:close()四、实用小工具

在数据库操作中，一些实用的小工具函数可以提高开发效率。以下是两个常用的小工具函数示例：

• 获取查询结果集到表：
  
  

function fetchAll(cursor)
    local results = {}
    local row = cursor:fetch({}, "a")
    while row do
        table.insert(results, row)
        row = cursor:fetch({}, "a")
    end
    return results
end

local cursor = conn:execute("SELECT * FROM table_name")
local results = fetchAll(cursor)
for _, row in ipairs(results) do
    print(string.format("id: %d, name: %s", row.id, row.name))
end
cursor:close()

• 批量插入数据：
  
  

function batchInsert(conn, tableName, columns, valuesList)
    local columnsStr = table.concat(columns, ", ")
    for _, values in ipairs(valuesList) do
        local valuesStr = table.concat(values, "', '")
        local sql = string.format("INSERT INTO %s (%s) VALUES ('%s')", tableName, columnsStr, valuesStr)
        conn:execute(sql)
    end
end

local columns = {"id", "name"}
local valuesList = {
    {1, "Lua"},
    {2, "LuaSQL"},
    {3, "SQLite"}
}
batchInsert(conn, "table_name", columns, valuesList)总结

本文介绍了 Lua 数据库库中的 MySQL 和 SQLite 数据库操作的常用函数及其使用方法。这些函数提供了强大的数据库操作能力，帮助开发者高效地实现各种数据库相关任务。

下一篇文章将介绍 Lua 的 JSON 处理库，敬请期待。

Lua 常用库系列：第六篇【JSON 处理库】