module DataClassification::Migration

Public Instance Methods

add_data_classification(table, column, classification) click to toggle source
# File lib/data_classification/migration.rb, line 5
def add_data_classification(table, column, classification)
  change_column_comment table, column, data_classification_comment(classification)
end
data_classification_comment(classification) click to toggle source
# File lib/data_classification/migration.rb, line 9
def data_classification_comment(classification)
  fail "This is not a valid data classification" unless DATA_CLASSIFICATIONS.include? classification

  {
    tags: [
      "classification:#{classification}",
    ],
  }.to_json
end