cep-all-shield-addons-counts.lua

-- 
-- General Filter for 'seen today' for all Study and Shield addons
-- 
-- 

require "cjson"
require "hyperloglog"
require "string"

addonId = "57-perception-shield-study@mozilla.com"
countStr = "57-perception-shield-study_count"

addon_day_counts = {}

-- date => count
addon_day_hlls = {} 

-- proposed structure:
-- 
-- ADDON_ID:  { date => count}
-- 

-- CEP: process an individual message
function process_message()
  local addonJson = read_message("Fields[environment.addons]")
  if not addonJson then
    return -1
  end

  -- exact match, wants to be regex
  if string.find(addonJson, addonId, 1, true) ~= nil then
    local day = read_message("Fields[submissionDate]")
    local hll = addon_day_hlls[day]
    if not hll then
      hll = hyperloglog.new()
      addon_day_hlls[day] = hll
    end
    hll:add(read_message("Fields[clientId]"))
  end

  return 0
end


-- CEP:  once in a while
function timer_event()
  local count = 0
  local earliest_day = nil
  for day, hll in pairs(addon_day_hlls) do
    addon_day_counts[day] = hll:count()
    count = count + 1
    if not earliest_day or day < earliest_day then
      earliest_day = day
    end
  end
  inject_payload("json", countStr, cjson.encode(addon_day_counts))
  if count > 30 then
    addon_day_hlls[earliest_day] = nil
  end
end