詳解Electron中如何使用SQLite存儲(chǔ)筆記

更新時(shí)間：2022年11月29日 09:53:55 作者：夜焱辰

這篇文章主要為大家介紹了Electron中如何使用SQLite存儲(chǔ)筆記示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

前言

上一篇，我們使用 remirror 實(shí)現(xiàn)了一個(gè)簡(jiǎn)單的 markdown 編輯器。接下來(lái)，我們要學(xué)習(xí)如何去存儲(chǔ)這些筆記。

當(dāng)然了，你也可以選擇不使用數(shù)據(jù)庫(kù)，不過(guò)若是你以后需要將該應(yīng)用上架到 mac Apple Store ，就需要考慮這個(gè)了。因?yàn)樯霞?mac 應(yīng)用需要啟用 sandbox，當(dāng)你第一次訪問(wèn)筆記中的媒體文件時(shí)，都要打開(kāi)選擇文件的彈窗，通過(guò)讓用戶主動(dòng)選擇來(lái)授權(quán)訪問(wèn)沙箱外的媒體文件。不過(guò)，如果你的媒體文件在第一次選擇插入文檔時(shí)復(fù)制到 sandbox 中，以后訪問(wèn)優(yōu)先從沙箱容器中讀取，那是不需要授權(quán)的。雖然我也可以這么做，但這里考慮到后面的功能，還是選擇使用數(shù)據(jù)庫(kù)，當(dāng)需要導(dǎo)出筆記時(shí)再?gòu)臄?shù)據(jù)庫(kù)中導(dǎo)出。

數(shù)據(jù)庫(kù)的選擇

Electron 應(yīng)用中常使用的數(shù)據(jù)庫(kù)是 SQLite、IndexedDB，IndexedDB 是在前端網(wǎng)頁(yè)中去操作。有的文章里說(shuō) IndexedDB 的性能會(huì)比 SQLite 更好，大家看實(shí)際場(chǎng)景去選擇使用。大多數(shù)桌面應(yīng)用或者 App 需要使用數(shù)據(jù)庫(kù)的時(shí)候一般都是用 SQLite。

npm 上有兩個(gè)最常用的 sqlite3 庫(kù)，一是 better-sqlite3 ，一是 node-sqlite ，兩種各有特點(diǎn)。前者是同步的 api ，執(zhí)行速度快，后者是異步 api ，執(zhí)行速度相對(duì)慢一點(diǎn)。值得注意的是，后者的編譯支持 arm 機(jī)器，而且由于出的比較早，和其他庫(kù)配合使用很方便。

安裝

安裝 node-sqlite

// 倉(cāng)庫(kù)名是 node-sqlite, package 名是 sqlite3
yarn add sqlite3

借助 Knex.js 簡(jiǎn)化數(shù)據(jù)庫(kù)操作

Knex.js是為Postgres，MSSQL，MySQL，MariaDB，SQLite3，Oracle和Amazon Redshift設(shè)計(jì)的 SQL 查詢構(gòu)建器

安裝 Knex.js

yarn add knex

創(chuàng)建表

現(xiàn)在，我們要開(kāi)始設(shè)計(jì)數(shù)據(jù)庫(kù)結(jié)構(gòu)了。我們大概需要 3 張表，筆記本表，筆記表，還有一個(gè)媒體文件表。sqlite 支持 blob 數(shù)據(jù)類型，所以你也可以把媒體文件的二進(jìn)制數(shù)據(jù)存到數(shù)據(jù)庫(kù)中。這里我們就簡(jiǎn)單的記個(gè) id ，把媒體文件存到沙箱內(nèi)。

我們確定一下三張表的表名，notebooks, notes, media, 然后看一下該如何使用 Knex.js 創(chuàng)建表

import { app } from "electron";
import knex, { Knex } from "knex";
import { join } from "path";
import { injectable } from "inversify";
@injectable()
export class LocalDB {
  declare db: Knex;
  async init() {
    this.db = knex({
      client: "sqlite",
      useNullAsDefault: true,
      connection: {
        filename: join(app.getPath("userData"), "local.db"),
      },
    });
    // 新建表
    await this.sync();
  }
  async sync() {
    // notebooks
    await this.db.schema.hasTable("notebooks").then((exist) => {
      if (exist) return;
      return this.db.schema.createTable("notebooks", (table) => {
        table.bigIncrements("id", { primaryKey: true });
        table.string("name");
        table.timestamps(true, true);
      });
    });
    // notes
    await this.db.schema.hasTable("notes").then((exist) => {
      if (exist) return;
      return this.db.schema.createTable("notes", (table) => {
        table.bigIncrements("id", { primaryKey: true });
        table.string("name");
        table.text("content");
        table.bigInteger("notebook_id");
        table.timestamps(true, true);
      });
    });
    // media
    await this.db.schema.hasTable("media").then((exist) => {
      if (exist) return;
      return this.db.schema.createTable("media", (table) => {
        table.bigIncrements("id", { primaryKey: true });
        table.string("name");
        table.string("local_path"); // 本地實(shí)際路徑
        table.string("sandbox_path"); // 沙盒中的地址
        table.bigInteger("note_id");
        table.timestamps(true, true);
      });
    });
  }
}

這里我用了一個(gè) IOC 庫(kù) inversify, 后面遇到 injectable、inject、ioc.get等寫(xiě)法都是和這個(gè)有關(guān)，這里我就不多介紹了，具體用法可以看文檔或其他文章。

注意：三張表中，note 和 media 都一個(gè)外鍵，這里我簡(jiǎn)化了，并沒(méi)有用 api 去創(chuàng)建。

Service

數(shù)據(jù)庫(kù)表創(chuàng)建完了，接下來(lái)我們?yōu)楸淼牟僮鲗?xiě)相關(guān)服務(wù)，這一塊我是參考傳統(tǒng)后端 api 的設(shè)計(jì)去寫(xiě)的，有 Service（數(shù)據(jù)庫(kù)）和 Controller（業(yè)務(wù)），以 Notebook 為例：

import { inject, injectable } from "inversify";
import { LocalDB } from "../db";
interface NotebookModel {
  id: number;
  name: string;
  create_at?: string | null;
  update_at?: string | null;
}
@injectable()
export class NotebooksService {
  name = "notebooks";
  constructor(@inject(LocalDB) public localDB: LocalDB) {}
  async create(data: { name: string }) {
    return await this.localDB.db.table(this.name).insert(data);
  }
  async get(id: number) {
    return await this.localDB.db
      .table<NotebookModel>(this.name)
      .select("*")
      .where("id", "=", id)
      .first();
  }
  async delete(id: number) {
    return await this.localDB.db.table(this.name).where("id", "=", id).delete();
  }
  async update(data: { id: number; name: string }) {
    return await this.localDB.db
      .table(this.name)
      .where("id", "=", data.id)
      .update({ name: data.name });
  }
  async getAll() {
    return await this.localDB.db.table<NotebookModel>(this.name).select("*");
  }
}

Service 只負(fù)責(zé)數(shù)據(jù)庫(kù)的連接和表中數(shù)據(jù)的增刪改查。

Controller

Controller 可以通過(guò)接入 Service 操作數(shù)據(jù)庫(kù)，并做一些業(yè)務(wù)上的工作。

import { inject, injectable } from "inversify";
import { NotebooksService } from "../services/notebooks.service";
import { NotesService } from "../services/notes.service";
@injectable()
export class NotebooksController {
  constructor(
    @inject(NotebooksService) public service: NotebooksService,
    @inject(NotesService) public notesService: NotesService
  ) {}
  async create(name: string) {
    await this.service.create({
      name,
    });
  }
  async delete(id: number) {
    const row = await this.service.get(id);
    if (row) {
      const notes = await this.notesService.getByNotebookId(id);
      if (notes.length) throw Error("delete failed");
      await this.service.delete(id);
    }
  }
  async update(data: { id: number; name: string }) {
    return await this.service.update(data);
  }
  async getAll() {
    return await this.service.getAll();
  }
}

業(yè)務(wù)

如何創(chuàng)建筆記本？

我們先來(lái)實(shí)現(xiàn)創(chuàng)建筆記本，之后的刪除筆記本，更新筆記本名稱等等，依葫蘆畫(huà)瓢就行。我們?cè)诮缑嫔咸砑右粋€(gè)創(chuàng)建按鈕。

點(diǎn)擊后就會(huì)出現(xiàn)這樣一個(gè)彈窗，這里 UI 庫(kù)我是用的 antd 做的。

看一下這個(gè)彈窗部分的邏輯

import { Modal, Form, Input } from "antd";
import React, { forwardRef, useImperativeHandle, useState } from "react";
interface CreateNotebookModalProps {
  onCreateNotebook: (name: string) => Promise<void>;
}
export interface CreateNotebookModalRef {
  setVisible: (visible: boolean) => void;
}
export const CreateNotebookModal = forwardRef<
  CreateNotebookModalRef,
  CreateNotebookModalProps
>((props, ref) => {
  const [modalVisible, setMoalVisible] = useState(false);
  const [form] = Form.useForm();
  const handleOk = () => {
    form.validateFields().then(async (values) => {
      await props.onCreateNotebook(values.name);
      setMoalVisible(false);
    });
  };
  useImperativeHandle(ref, (): CreateNotebookModalRef => {
    return {
      setVisible: setMoalVisible,
    };
  });
  return (
    <Modal
      visible={modalVisible}
      title="創(chuàng)建筆記本"
      onCancel={() => setMoalVisible(false)}
      onOk={handleOk}
      cancelText="取消"
      okText="確定"
      destroyOnClose
    >
      <Form form={form}>
        <Form.Item
          label="筆記本名稱"
          name="name"
          rules={[
            {
              required: true,
              message: "請(qǐng)?zhí)顚?xiě)名稱",
            },
            {
              whitespace: true,
              message: "禁止使用空格",
            },
            { min: 1, max: 100, message: "字符長(zhǎng)度請(qǐng)保持在 1-100 之間" },
          ]}
        >
          <Input />
        </Form.Item>
      </Form>
    </Modal>
  );
});

外部提供的 onCreateNotebook 的實(shí)現(xiàn)：

const handleCreateNotebook = async (name: string) => {
    await window.Bridge?.createNotebook(name);
    const data = await window.Bridge?.getNotebooks();
    if (data) {
      setNotebooks(data);
    }
};

上面出現(xiàn)的 Bridge 是我在第一篇中講的 preload.js 提供的對(duì)象，它可以幫我們和 electron 主進(jìn)程通信。

接來(lái)寫(xiě)，我們具體看一下 preload 和主進(jìn)程部分的實(shí)現(xiàn)：

// preload.ts
import { contextBridge, ipcRenderer, MessageBoxOptions } from "electron";
contextBridge.exposeInMainWorld("Bridge", {
  showMessage: (options: MessageBoxOptions) => {
    ipcRenderer.invoke("showMessage", options);
  },
  createNotebook: (name: string) => {
    return ipcRenderer.invoke("createNotebook", name);
  },
  getNotebooks: () => {
    return ipcRenderer.invoke("getNotebooks");
  },
});

實(shí)際還是用 ipcRenderer 去通信，但是這種方式更好

// main.ts
import { ipcMain } from "electron"
ipcMain.handle("createNotebook", async (e, name: string) => {
  return await ioc.get(NotebooksController).create(name);
});
ipcMain.handle("getNotebooks", async () => {
  return await ioc.get(NotebooksController).getAll();
});