# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
require 'solr/xml'
require 'solr/request/base'
require 'solr/document'
require 'solr/request/update'
class Solr::Request::ModifyDocument < Solr::Request::Update
# Example: ModifyDocument.new(:id => 10, :overwrite => {:field_name => "new value"})
def initialize(update_data)
modes = []
@doc = {}
[:overwrite, :append, :distinct, :increment, :delete].each do |mode|
field_data = update_data[mode]
if field_data
field_data.each do |field_name, field_value|
modes << "#{field_name}:#{mode.to_s.upcase}"
@doc[field_name] = field_value if field_value # if value is nil, omit so it can be removed
end
update_data.delete mode
end
end
@mode = modes.join(",")
# only one key should be left over, the id
@doc[update_data.keys[0].to_s] = update_data.values[0]
end
# returns the request as a string suitable for posting
def to_s
e = Solr::XML::Element.new 'add'
e.add_element(Solr::Document.new(@doc).to_xml)
return e.to_s
end
def handler
"update?mode=#{@mode}"
end
end