Как изменить значение по умолчанию атрибута Struct?
Согласно документации неустановленные атрибуты Struct устанавливаются в nil
:
unset параметры по умолчанию ноль.
Можно ли указать значение по умолчанию для определенных атрибутов?
Например, для следующей структуры
Struct.new("Person", :name, :happy)
Я хотел бы атрибут happy
по умолчанию true
скорее, чем nil
, Как я могу это сделать? Если я сделаю следующее
Struct.new("Person", :name, :happy = true)
я получил
-:1: syntax error, unexpected '=', expecting ')'
Struct.new("Person", :name, :happy = true)
^
-:1: warning: possibly useless use of true in void context
6 ответов
Это также может быть достигнуто путем создания структуры в качестве подкласса и переопределения initialize
со значениями по умолчанию, как в следующем примере:
class Person < Struct.new(:name, :happy)
def initialize(name, happy=true); super end
end
С одной стороны, этот метод приводит к небольшому шаблону; с другой стороны, он делает то, что вы ищете, красиво и лаконично.
Одним побочным эффектом (который может быть либо выгодой, либо раздражением в зависимости от ваших предпочтений / варианта использования) является то, что вы теряете значение по умолчанию. Struct
поведение всех атрибутов по умолчанию nil
- если вы явно не установите их так. По сути, приведенный выше пример name
обязательный параметр, если вы не объявите его как name=nil
Следуя примеру @rintaun, вы можете сделать это с помощью ключевых слов в Ruby 2+.
A = Struct.new(:a, :b, :c) do
def initialize(a:, b: 2, c: 3); super end
end
A.new
# ArgumentError: missing keyword: a
A.new a: 1
# => #<struct A a=1, b=2, c=3>
A.new a: 1, c: 6
# => #<struct A a=1, b=2, c=6>
ОБНОВИТЬ
Код теперь должен быть написан следующим образом для работы.
A = Struct.new(:a, :b, :c) do
def initialize(a:, b: 2, c: 3)
super(a, b, c)
end
end
Я также нашел это:
Person = Struct.new "Person", :name, :happy do
def initialize(*)
super
self.location ||= true
end
end
@Linuxios дал ответ, который отменяет поиск членов. У этого есть пара проблем: вы не можете явно установить член равным nil, и есть дополнительная нагрузка на каждую ссылку на член. Мне кажется, вы действительно просто хотите указать значения по умолчанию при инициализации нового объекта структуры с частичными значениями членов, предоставленными ::new
или же ::[]
,
Вот модуль для расширения Struct дополнительным фабричным методом, который позволяет описывать желаемую структуру с помощью хэша, где ключами являются имена членов и значения по умолчанию, которые необходимо заполнить, если они не предоставлены при инициализации:
# Extend stdlib Struct with a factory method Struct::with_defaults
# to allow StructClasses to be defined so omitted members of new structs
# are initialized to a default instead of nil
module StructWithDefaults
# makes a new StructClass specified by spec hash.
# keys are member names, values are defaults when not supplied to new
#
# examples:
# MyStruct = Struct.with_defaults( a: 1, b: 2, c: 'xyz' )
# MyStruct.new #=> #<struct MyStruct a=1, b=2, c="xyz"
# MyStruct.new(99) #=> #<struct MyStruct a=99, b=2, c="xyz">
# MyStruct[-10, 3.5] #=> #<struct MyStruct a=-10, b=3.5, c="xyz">
def with_defaults(*spec)
new_args = []
new_args << spec.shift if spec.size > 1
spec = spec.first
raise ArgumentError, "expected Hash, got #{spec.class}" unless spec.is_a? Hash
new_args.concat spec.keys
new(*new_args) do
class << self
attr_reader :defaults
end
def initialize(*args)
super
self.class.defaults.drop(args.size).each {|k,v| self[k] = v }
end
end.tap {|s| s.instance_variable_set(:@defaults, spec.dup.freeze) }
end
end
Struct.extend StructWithDefaults
Просто добавьте еще один вариант:
class Result < Struct.new(:success, :errors)
def initialize(*)
super
self.errors ||= []
end
end
Я думаю, что переопределение #initialize
метод является лучшим способом, с вызовом #super(*required_args)
,
Это имеет дополнительное преимущество, заключающееся в возможности использовать аргументы в стиле хеш-функции. Пожалуйста, посмотрите следующий полный и компилирующий пример:
Аргументы в стиле хэш, значения по умолчанию и Ruby Struct
# This example demonstrates how to create Ruby Structs that use
# newer hash-style parameters, as well as the default values for
# some of the parameters, without loosing the benefits of struct's
# implementation of #eql? #hash, #to_s, #inspect, and other
# useful instance methods.
#
# Run this file as follows
#
# > gem install rspec
# > rspec struct_optional_arguments.rb --format documentation
#
class StructWithOptionals < Struct.new(
:encrypted_data,
:cipher_name,
:iv,
:salt,
:version
)
VERSION = '1.0.1'
def initialize(
encrypted_data:,
cipher_name:,
iv: nil,
salt: 'salty',
version: VERSION
)
super(encrypted_data, cipher_name, iv, salt, version)
end
end
require 'rspec'
RSpec.describe StructWithOptionals do
let(:struct) { StructWithOptionals.new(encrypted_data: 'data', cipher_name: 'AES-256-CBC', iv: 'intravenous') }
it 'should be initialized with default values' do
expect(struct.version).to be(StructWithOptionals::VERSION)
end
context 'all fields must be not null' do
%i(encrypted_data cipher_name salt iv version).each do |field|
subject { struct.send(field) }
it field do
expect(subject).to_not be_nil
end
end
end
end
Вы можете сделать это с помощью синглтона:
Person = Struct.new("Person", :name, :happy) do
@@defaults = {:happy => true}
@@vars = [:name, :happy]
def [](i)
return super if(super)
return @@defaults[i] if(@@defaults[i])
return nil
end
@@vars.each do |v|
define_method(v) {return self[v]}
end
end