mirror of
https://github.com/apache/lucene.git
synced 2025-02-10 03:55:46 +00:00
b016188d22
git-svn-id: https://svn.apache.org/repos/asf/lucene/solr/trunk@569952 13f79535-47bb-0310-9956-ffa450edef68
52 lines
1.6 KiB
Ruby
Executable File
52 lines
1.6 KiB
Ruby
Executable File
# 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
|