Gravando arquivos em BD usando PDO do PHP

Introdução

Existem duas formas mais convencionais de se armazenar arquivos em sistemas de informação: gravando em um diretório reservado ou gravando no banco de dados. Gravando em um diretório, existem alguns riscos como o arquivo poder ser acessado por usuários do servidor, ou o arquivo ser acessado por qualquer pessoa da Web se ficar em um diretório visível na Web. Guardando o arquivo em BD, existe o inconveniente do BD poder ficar com um tamanho muito grande, embora ofereça uma segurança adicional. Neste artigo veremos como gravar e recuperar arquivos do BD usando PDO. Para mais detalhes sobre PDO, leia antes o artigo PDO.

---

Criação da Tabela do BD

Antes de tudo, é preciso criar uma tabela no BD que terá um ou mais campos (colunas) que armazenarão o arquivo propriamente dito. O campo mais indicado para armazenar este tipo de informação é chamado "BLOB". No MySQL, por exemplo, existem as variações: TINYBLOB, BLOB, MEDIUMBLOB e LARGEBLOB (cada um com uma capacidade diferente, aproximadamente, 256B, 64KB, 16MB e 4GB respectivamente). Outra informação importante de ser armazenada no BD é o tipo de arquivo, de preferencia o mime-type. Opcionalmente, guarda-se também um nome de arquivo. Exemplo:

CREATE TABLE arquivos (
  id INT NOT NULL AUTO_INCREMENT PRIMARY KEY,
  nome VARCHAR(128) NOT NULL DEFAULT '',
  mimetype VARCHAR(60) NOT NULL DEFAULT 'application/octet-stream',
  conteudo BLOB NOT NULL
);

---

Inserindo o arquivo no BD

Para inserir o arquivo no BD, ele precisa ter vindo de algum lugar. Pode ser via Upload, pode ser um arquivo que estava no servidor, pode ser um arquivo gerado dinamicamente e armazenado em uma variável, ou pode ser um arquivo que está em uma variável. Em todos os casos, será necessário gerar um handle do arquivo, ou seja, ele precisa estar no servidor para gerar um ponteiro de arquivo (com fopen).

Se o arquivo veio por Upload sem erros, ele pode está numa pasta temporária do servidor, e o handle de arquivo pode ser criado assim:

$handle = fopen($_FILE['arquivo']['tmp_name'], 'rb');

E o mimetype e o nome do arquivo podem ser obtidos assim:

$mimetype = $_FILE['arquivo']['type'];
$nome = basename($_FILE['arquivo']['name']);

Se o arquivo já estava no servidor, basta abrir o hanle passando o caminho ao arquivo:

$caminho = '/caminho/ate/o/arquivo';
$handle = fopen($caminho, 'rb');

E se o arquivo está em uma variável, pode ser gerado um handle assim:

$variavel = 'exemplo';     // conteudo do arquivo
$mimetype = 'text/plain';  // mime-type do arquivo

$handle = tmpfile();
fwrite($handle, $variavel);

Com o handle montado, basta montar a SQL de insert e executá-la:

// Montando o handle do arquivo
$handle = ...
$mimetype = ...
$nome = ...

// Preparando o INSERT
$pdo = new PDO(...);
$stmt = $pdo->prepare('INSERT INTO arquivos (nome, mimetype, conteudo) VALUES (:nome, :mimetype, :conteudo)');

$stmt->bindParam(':nome', $nome);
$stmt->bindParam(':mimetype', $mimetype);
$stmt->bindParam(':conteudo', $handle, PDO::PARAM_LOB);

// Inserindo o arquivo no BD
$pdo->beginTransaction();
$stmt->execute();
$pdo->commit();

---

Consultando um arquivo do BD

Assim como foi usado um handle de arquivo para armazenar o arquivo no BD, quando ele é consultado também é devolvido um hanle de arquivo.

// Definindo qual arquivo sera consultado
$id = ...

// Consultando o arquivo
$pdo = new PDO(...);
$stmt = $pdo->prepare('SELECT nome, mimetype, conteudo FROM arquivos WHERE id=:id LIMIT 1');
$stmt->execute(array(':id' => $id));

// Atribuindo o resultado a variaveis
$stmt->bindColumn(1, $nome, PDO::PARAM_STR);
$stmt->bindColumn(2, $mimetype, PDO::PARAM_STR);
$stmt->bindColumn(3, $handle, PDO::PARAM_LOB);
$stmt->fetch(PDO::FETCH_BOUND);

Com o handle em mãos novamente, você é capaz de enviar o arquivo para o usuário:

header('Content-Type: '.$mimetype);
header('Content-Disposition: inline; filename="'.$nome.'"');
header('Cache-Control: no-cache, no-store, must-revalidate');
header('Pragma: no-cache');
header('Expires: 0');
fpassthru($handle);
exit(0);

Para forçar o download, basta trocar "inline" por "attachement".