File size: 3,131 Bytes
c27e67a | 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 106 107 | defmodule Plausible.Workers.AcceptTrafficUntil do
@moduledoc """
A worker meant to be run once a day that sends out e-mail notifications to site
owners assuming:
- their sites still receive traffic (i.e. have stats for yesterday)
- `site.accept_traffic_until` is approaching either tomorrow or exactly in 7 days
Users having no sites or sites that receive no traffic, won't be notified.
We make a tiny effort here to make sure we send the same notification at most once a day.
"""
use Oban.Worker, queue: :check_accept_traffic_until
import Ecto.Query
alias Plausible.Repo
alias Plausible.ClickhouseRepo
def dry_run(date) do
perform(nil, date, true)
end
@impl Oban.Worker
def perform(_job, today \\ Date.utc_today(), dry_run? \\ false) do
tomorrow = today |> Date.add(+1)
next_week = today |> Date.add(+7)
# send at most one notification per user, per day
sent_today_query =
from s in "sent_accept_traffic_until_notifications",
where: s.user_id == parent_as(:users).id and s.sent_on == ^today,
select: true
notifications =
Repo.all(
from t in Plausible.Teams.Team,
inner_join: u in assoc(t, :owners),
as: :users,
inner_join: s in assoc(t, :sites),
where: t.accept_traffic_until == ^tomorrow or t.accept_traffic_until == ^next_week,
where: not exists(sent_today_query),
select: %{
id: u.id,
email: u.email,
deadline: t.accept_traffic_until,
site_ids: fragment("array_agg(?.id)", s),
name: u.name,
team: t
},
group_by: [u.id, t.id]
)
for notification <- notifications do
case {has_stats?(notification.site_ids, today), notification.deadline} do
{true, ^tomorrow} ->
if dry_run? do
IO.puts("Will send final notification to #{notification.email}")
else
notification
|> store_sent(today)
|> PlausibleWeb.Email.approaching_accept_traffic_until_tomorrow()
|> Plausible.Mailer.send()
end
{true, ^next_week} ->
if dry_run? do
IO.puts("Will send weekly notification to #{notification.email}")
else
notification
|> store_sent(today)
|> PlausibleWeb.Email.approaching_accept_traffic_until()
|> Plausible.Mailer.send()
end
_ ->
nil
end
end
{:ok, Enum.count(notifications)}
end
defp has_stats?(site_ids, today) do
ago_2d = Date.add(today, -2)
ClickhouseRepo.exists?(
from e in "events_v2",
where: fragment("toDate(?) >= ?", e.timestamp, ^ago_2d),
where: e.site_id in ^site_ids
)
end
defp store_sent(notification, today) do
Repo.insert_all(
"sent_accept_traffic_until_notifications",
[
%{
user_id: notification.id,
sent_on: today
}
],
on_conflict: :nothing,
conflict_target: [:user_id, :sent_on]
)
notification
end
end
|