如何让TypeORM实体列类型适配MySQL与SQLite数据库？

这是个很常见的TypeORM跨数据库适配场景，针对你需要在MySQL用mediumblob、SQLite用blob的需求，这里有几个实用的实现方式：

方法1：在@Column中用动态函数返回对应类型 #

TypeORM允许@Column的type参数接收一个函数，你可以通过实体元数据获取当前连接的数据库类型，动态返回对应列类型：

import { Entity, PrimaryGeneratedColumn, Column } from "typeorm";

@Entity()
export class File {
  @PrimaryGeneratedColumn()
  id: number;

  @Column({
    type: (columnMeta) => {
      // 从实体元数据中获取当前连接的数据库类型
      const dbType = columnMeta.entityMetadata.connection.options.type;
      return dbType === "mysql"? "mediumblob" : "blob";
    },
    nullable: false // 根据你的需求添加其他列配置
  })
  data: Buffer;
}

优点：实体定义集中，不需要修改测试代码；缺点：依赖实体元数据的连接信息，需要确保连接初始化顺序正确，适合大部分常规场景。

方法2：测试连接初始化时动态修改实体元数据 #

如果你不想改动生产环境的实体定义，可以在测试代码创建SQLite连接后，手动修改File实体的列类型：

import { createConnection } from "typeorm";
import { File } from "./path/to/File.entity";

describe("CR on File and blobs", () => {
  it("should store arbitrary binary file", async () => {
    const conn = await createConnection({
      type: "sqlite",
      database: ":memory:",
      entities: [File],
      synchronize: true,
    });

    // 找到File实体的元数据和data列
    const fileEntityMeta = conn.entityMetadatas.find(meta => meta.name === "File");
    const dataColumn = fileEntityMeta.columns.find(col => col.propertyName === "data");
    
    if (dataColumn) {
      // 修改列类型为SQLite支持的blob
      dataColumn.type = "blob";
      // 重新同步数据库schema（强制覆盖）
      await conn.synchronize(true);
    }

    // 执行你的测试逻辑
    const fileRepo = conn.getRepository(File);
    const testFile = new File();
    testFile.data = Buffer.from("sample binary content");
    await fileRepo.save(testFile);

    const savedFile = await fileRepo.findOne(testFile.id);
    expect(savedFile.data.toString()).toBe("sample binary content");

    await conn.close();
  });
});

优点：完全不影响生产实体代码，测试逻辑独立；缺点：需要手动操作实体元数据，代码稍显繁琐。

方法3：通过环境变量判断切换类型 #

如果你的测试和生产环境通过NODE_ENV区分，可以直接在实体中根据环境变量返回对应类型：

import { Entity, PrimaryGeneratedColumn, Column } from "typeorm";

// 根据环境变量返回对应blob类型
const getBlobColumnType = () => {
  return process.env.NODE_ENV === "test"? "blob" : "mediumblob";
};

@Entity()
export class File {
  @PrimaryGeneratedColumn()
  id: number;

  @Column({ type: getBlobColumnType() })
  data: Buffer;
}

优点：实现最简单，无需依赖TypeORM内部API；缺点：依赖环境变量的正确设置，需要确保测试脚本启动时NODE_ENV=test。

额外说明 #

SQLite的blob类型没有MySQL那样的大小限制（理论上只要磁盘空间足够，可存储任意大小的二进制数据），所以用它替代mediumblob完全能满足你的测试需求。

内容的提问来源于stack exchange，提问作者svobol13