Início / coding / Perguntas / 79464910
Accepted
lmtaq
Asked: 2025-02-25 06:05:24 +0800 CST

Remover bytes nulos para cada atributo de string do ActiveRecord

  • 772

Tenho uma aplicação Rails usando Postgres como banco de dados.

O Postrgres não aceita strings contendo bytes nulos (exemplo: "a \u0000 b"). Tentar salvar esses dados leva ao seguinte erro:

ActiveRecord::StatementInvalid
PG::UntranslatableCharacter: ERROR:  unsupported Unicode escape sequence

Gostaria de configurar uma regra que funcionasse para todos os meus modelos ActiveRecord, garantindo que cada atributo de string removesse bytes nulos dele antes de salvar. Estou pensando em algo que poderia ter um efeito semelhante ao exemplo abaixo:

class MyModel < ApplicationRecord
  before_save :remove_null_bytes

  private

  def remove_null_bytes
    my_field.delete!("\u0000")
  end
end

Mas aplicado a cada atributo de string de cada modelo sem ser forçado a defini-lo repetidamente.

O motivo pelo qual quero que isso seja limitado aos atributos de string do Active Record é que o aplicativo também manipula uploads binários e webhooks de e-mail. Eles podem conter bytes nulos legítimos, e não quero que sejam afetados pela nova regra.

ruby-on-rails

3 respostas

  1. Best Answer

    Quando você quiser normalizar todos os atributos do tipo string em todos os modelos, eu adicionaria isto a ApplicationRecord:

    class ApplicationRecord < ActiveRecord::Base
  normalizes *attribute_names, 
    with: ->(value) { value.delete("\u0000") if value.is_a?(String) }
end

    Observe que isso normalizesfoi introduzido no Ruby on Rails 7.1 e não está disponível em versões mais antigas.

    • 3

  2. Estou pensando em definir algo assim em ApplicationRecord:

    class ApplicationRecord < ActiveRecord::Base
  before_save :remove_null_bytes

  def remove_null_bytes
    changes.keys.each do |attribute_name|
      attr_type = type_for_attribute(attribute_name).type
      send(attribute_name).delete!("\u0000") if attr_type == :string
    end
  end
end

    Não tenho certeza se estou feliz com a metaprogramação e a falta de legibilidade dessa abordagem.

    • 1

  3. Substitua write_attribute em ApplicationRecord para garantir que todos os atributos de string sejam higienizados quando atribuídos:

    não há necessidade de adicionar o método before_save

    class ApplicationRecord < ActiveRecord::Base
  self.abstract_class = true

  def write_attribute(attr_name, value)
    column = self.class.columns_hash[attr_name.to_s]

    if column&.type == :string && value.is_a?(String)
      value = value.delete("\u0000")
    end

    super(attr_name, value)
  end
end
    • 1

relate perguntas