Mastodonを読む
はじめに †
タイムライン取得処理、ここからはサーバ側の処理になります。久しぶりにRubyに戻ってきました。
app/controllers/app/v1/timelines_controller.rb †
前回確認したように、ホームのタイムラインを取得する際には、「/api/v1/timelines/home」にアクセスされます。対応するメソッドはTimelinesControllerのhomeです。
1
2
3
4
5
6
7
8
9
10
11
12
13
| -
|
|
|
|
|
|
|
|
|
|
|
!
| def home
@statuses = Feed.new(:home, current_account).get(limit_param(DEFAULT_STATUSES_LIMIT), params[:max_id], params[:since_id])
@statuses = cache_collection(@statuses)
set_maps(@statuses)
next_path = api_v1_home_timeline_url(pagination_params(max_id: @statuses.last.id)) unless @statuses.empty?
prev_path = api_v1_home_timeline_url(pagination_params(since_id: @statuses.first.id)) unless @statuses.empty?
set_pagination_headers(next_path, prev_path)
render :index
end
|
app/models/feed.rb †
Feedはmodelsの下にあります。（Feed自体は対応するデータベーステーブルがあるわけではありません）
そんなに長くないので全部貼り付け。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
| -
-
|
|
!
|
-
|
|
|
|
|
|
!
|
|
|
-
|
!
|
-
|
!
!
| class Feed
def initialize(type, account)
@type = type
@account = account
end
def get(limit, max_id = nil, since_id = nil)
max_id = '+inf' if max_id.blank?
since_id = '-inf' if since_id.blank?
unhydrated = redis.zrevrangebyscore(key, "(#{max_id}", "(#{since_id}", limit: [0, limit], with_scores: true).map(&:last).map(&:to_i)
status_map = Status.where(id: unhydrated).cache_ids.map { |s| [s.id, s] }.to_h
unhydrated.map { |id| status_map[id] }.compact
end
private
def key
FeedManager.instance.key(@type, @account.id)
end
def redis
Redis.current
end
end
|
Redisはインメモリのデータストア、FeedManagerはapp/libにあります。zrevrangebyscoreはキーに対して指定された範囲のリストを返すようです。
で、保存してるIDリストに対応するStatusモデルを返すと。
Feedのキャッシュ †
なんか引っかかります。そう、IDリストっていつの間にキャッシュされたの？ということです。よくわからないのでとりあえず「redis」でgrepしてみたところ次のコードが引っかかりました。
app/services/precompute_feed_service.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
| -
-
|
|
-
-
-
|
|
!
!
!
|
|
|
-
|
!
!
| class PrecomputeFeedService < BaseService
def call(_, account)
redis.pipelined do
Status.as_home_timeline(account).limit(FeedManager::MAX_ITEMS / 4).each do |status|
next if status.direct_visibility? || FeedManager.instance.filter?(:home, status, account.id)
redis.zadd(FeedManager.instance.key(:home, account.id), status.id, status.reblog? ? status.reblog_of_id : status.id)
end
end
end
private
def redis
Redis.current
end
end
|
うむ、こいつがキャッシュしてるっぽい。
キャッシュしてるところが見つかったので、今度は、じゃあ誰がPrecomputeFeedService呼んでるの？ということになります。またまたgrepのお力を借りると今度は次のコードが引っかかりました。
app/workers/regeneration_worker.rb
1
2
3
4
5
6
7
8
9
| -
|
|
|
|
-
|
!
!
| class RegenerationWorker
include Sidekiq::Worker
sidekiq_options queue: 'pull', backtrace: true, unique: :until_executed
def perform(account_id, _ = :home)
PrecomputeFeedService.new.call(:home, Account.find(account_id))
end
end
|
でこいつは(ry、とさらにgrepしたところ、
app/controllers/application_controller.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
| -
|
|
|
-
|
|
-
!
|
-
!
!
| class ApplicationController < ActionController::Base
before_action :set_user_activity
def set_user_activity
return unless !current_user.nil? && (current_user.current_sign_in_at.nil? || current_user.current_sign_in_at < 24.hours.ago)
current_user.update_tracked_fields(request)
RegenerationWorker.perform_async(current_user.account_id) if current_user.last_sign_in_at < 14.days.ago
end
|
これ以上は追いかけませんが、まあ動いているから動くのでしょう。
app/views/app/v1/timelines/index.rabl †
さて、余談気味な話が長くなりましたが、話をtimelines#homeに戻して、homeメソッドの最後に「render :index」があるのでindex.rablが実行されます。
1
2
|
| collection @statuses
extends('api/v1/statuses/show')
|
RABLはhome#indexの時にも出てきたJSONを作るためのDSLです。あちらではやや特殊な使われ方がされていましたが、今回はStatusリストを返すのが目的なので普通のRABLの記述になっています。淡々とJSONを構築しているだけなので省略します。生成されるJSONの仕様はこちらにあります。
おわりに †
今回はサーバ側で行われるタイムライン取得処理を見てきました。基本的にはデータベースから取得してJSONを返すだけですが、ホームタイムラインの場合は気づかぬうちにキャッシュが行われ、それが使われていました。Mastodonは読んでいるとこういう「気づかぬうちに行われている」というものが多いように思います。まあそれはそもそものRailsもそうか(笑)
![[PukiWiki]](image/pukiwiki.png "[PukiWiki]")
