检索根据 redis 排序集排序的 Django 查询集的最高效方法

Question

我有一个 redis 排序集，其中包含对象 ID 及其各自的排序分数。我需要检索这些对象的 Django 查询集，根据排序集的分数进行排序。最有效的方法是什么？我的数据库是 postgresql。

---

目前，我正在分别检索 redis 排序集和 未排序 查询集。接下来，我根据排序集在 python 中的分数对查询集进行排序。可以在这里有意义地削减步骤，例如以某种方式在检索点对查询集进行排序等？

    obj_ids_with_scr = get_ids_w_scr() #retrieves redis sorted set with scores
    obj_ids = map(itemgetter(0),obj_ids_with_scr) # filters sorted set for just the obj ids
    queryset = Widget.objects.filter(id__in=obj_ids) #unsorted queryset
    a = dict(obj_ids_with_scr) #turning the sorted set into a dictionary
    for obj_pk, sort_score in a:
        obj = queryset.get(id=obj_pk) #get object with id equalling obj_pk 
        sort_score = obj  #assign the object to the 'score' value of this key-value 
    result = a.values() #making a list of all values, that are now in sorted order
    return result

Answer 1

如果您使用的是 postgres，则可以通过定义自己的 field function in postgres 来模拟 mysql 的 order by field 语法 here：

CREATE FUNCTION field(anyelement, VARIADIC anyarray) RETURNS numeric AS $$
  SELECT
    COALESCE(
     ( SELECT i FROM generate_subscripts($2, 1) gs(i)
       WHERE $2[i] = $1 ),
     0);
$$ LANGUAGE SQL STABLE

然后用类似这样的方式调用：

queryset = queryset.extra(
    select={'manual': "FIELD(id, %s)" % (','.join(map(str, obj_ids)),) },
    order_by=['manual']
)

---

n.b.如果您愿意，也可以像这样优化您的 python 代码：

widgets = Widgets.objects.in_bulk(obj_ids)
sorted_widgets = [ widgets[x] for x in obj_ids if x in widgets ]

In-bulk 返回一个 dict 映射 id 到 Widget 实例，这对于您正在执行的排序类型很有用。你可以阅读更多关于它的信息here。