开始接触 ruby 不久,算是绝对的 ruby 新人,所以当写出的代码很长时,总感觉是 ruby 的话应该还有更精简的写法,但是限于我的知识,苦思冥想也没有找到一个更合适的解决方法,问题如下,请各位 ruby 高手不吝指点:
现在的项目中遇到这样一个情景,需要把整理出来的数据进行分页,然后返回给客户端,客户端第一次请求不带参数,默认返回最前边的 n 个元素,然后客户端可以通过把第一次请求的首尾元素以及想要前页还是后页放入 params 来请求该组元素的前后页数据,下面是我写的一个分页逻辑,但是感觉太长,不知是否有可以改进的地方?
参数详解--------------------------------------------------------------------
arr: 要被分页的数据数组 start_element: 上次返回的开始元素 end_element: 上次返回的结束元素 action: 这一次索要的数据是前页还是后页,可以是 prev,或者 next pace: 每一页的数量
代码如下----------------------------------------------------------------------
def paginate (arr, start_element=nil, end_element=nil, action=nil, pace)
if start_element and end_element and action
start_index = arr.index(start_element)
end_index = arr.index(end_element)
case action
when 'next'
prev_start_index = (start_index - pace + 1 > 0) ? (start_index - pace + 1) : 0
prev_end_index = (start_index - 1) > 0 ? (start_index -1) : 0
result = arr[prev_start_index..prev_end_index]
next_start_element = result.first
next_end_element = result.last
when 'prev'
next_start_index = (end_index + 1 < arr.count) ? end_index + 1 : arr.count - 1
next_end_index = (end_index + pace < arr.count) ? end_index + pace : arr.count - 1
result = arr[next_start_index..next_end_index]
next_start_element = result.first
next_end_element = result.last
end
if next_start_element == next_end_element and (next_end_element == start_element || next_start_element == end_element)
result = []
end
[next_start_element,next_end_element,result]
else
result = arr.first(pace)
[result.first, result.last, result]
end
end