File size: 2,609 Bytes
8da2481 | 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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 | defmodule Plausible.Site.Cache do
@moduledoc """
The cache allows lookups by both `domain` and `domain_changed_from`
fields - this is to allow traffic from sites whose domains changed within a certain
grace period (see: `Plausible.Site.Transfer`).
To differentiate cached Site structs from those retrieved directly from the
database, a virtual schema field `from_cache?: true` is set.
This indicates the `Plausible.Site` struct is incomplete in comparison to its
database counterpart -- to spare bandwidth and query execution time,
only selected database columns are retrieved and cached.
The `@cached_schema_fields` attribute defines the list of DB columns
queried on each cache refresh.
Also see tests for more comprehensive examples.
"""
require Logger
import Ecto.Query
alias Plausible.Site
use Plausible.Cache
@cache_name :sites_by_domain
@cached_schema_fields ~w(
id
domain
domain_changed_from
ingest_rate_limit_scale_seconds
ingest_rate_limit_threshold
)a
@impl true
def name(), do: @cache_name
@impl true
def child_id(), do: :cache_sites
@impl true
def count_all() do
from(s in Site.regular())
|> Plausible.Repo.aggregate(:count)
end
@impl true
def base_db_query() do
from s in Site.regular(),
left_join: rg in assoc(s, :revenue_goals),
inner_join: team in assoc(s, :team),
select: {
s.domain,
s.domain_changed_from,
%{struct(s, ^@cached_schema_fields) | from_cache?: true}
},
preload: [revenue_goals: rg, team: team]
end
@impl true
def get_from_source(domain) do
query = from s in base_db_query(), where: s.domain == ^domain
case Plausible.Repo.one(query) do
{_, _, site = %Site{}} -> %Site{site | from_cache?: false}
_any -> nil
end
end
@spec get_site_id(String.t(), Keyword.t()) :: pos_integer() | nil
def get_site_id(domain, opts \\ []) do
case get(domain, opts) do
%{id: site_id} ->
site_id
nil ->
nil
end
end
@spec touch_site!(Site.t(), DateTime.t()) :: Site.t()
def touch_site!(site, now) do
now =
now
|> DateTime.truncate(:second)
|> DateTime.to_naive()
site
|> Ecto.Changeset.change(updated_at: now)
|> Plausible.Repo.update!()
end
@impl true
def unwrap_cache_keys(items) do
Enum.reduce(items, [], fn
{domain, nil, object}, acc ->
[{domain, object} | acc]
{domain, domain_changed_from, object}, acc ->
[{domain, object}, {domain_changed_from, object} | acc]
end)
end
end
|