Mercurial Hosting > linkmystyle
diff src/lib/Reporting.luan @ 0:8f4df159f06b
start public repo
author | Franklin Schmidt <fschmidt@gmail.com> |
---|---|
date | Fri, 11 Jul 2025 20:57:49 -0600 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/lib/Reporting.luan Fri Jul 11 20:57:49 2025 -0600 @@ -0,0 +1,166 @@ +local Luan = require "luan:Luan.luan" +local error = Luan.error +local stringify = Luan.stringify or error() +local pairs = Luan.pairs or error() +local ipairs = Luan.ipairs or error() +local range = Luan.range or error() +local Io = require "luan:Io.luan" +local uri = Io.uri or error() +local Number = require "luan:Number.luan" +local long = Number.long or error() +local Time = require "luan:Time.luan" +local time_now = Time.now or error() +local Table = require "luan:Table.luan" +local copy = Table.copy or error() +local sort = Table.sort or error() +local Thread = require "luan:Thread.luan" +local Lucene = require "luan:lucene/Lucene.luan" +local Shared = require "site:/lib/Shared.luan" +local compressed = Shared.compressed or error() +local Utils = require "site:/lib/Utils.luan" +local list_to_set = Utils.list_to_set or error() +local Logging = require "luan:logging/Logging.luan" +local logger = Logging.logger "Reporting" + + +local Reporting = {} + +local dir = uri("site:/private/local/reporting") + +local db = Lucene.index( dir ) +Reporting.db = db + +db.indexed_fields.owner = Lucene.type.string +db.indexed_fields.value = Lucene.type.string +db.indexed_fields.day = Lucene.type.long + + +local day = Time.period{days=1} +local days = Time.period{days=30} +local offset_from_GMT = Time.period{hours=7} + +local function today() + local today = (time_now() - offset_from_GMT) // day * day + offset_from_GMT + return long(today) +end + +local function by_count(rec1,rec2) + return rec1.count > rec2.count +end + +local limit = 20 + +function Reporting.get_data( query, sum_by ) + local first_day = long(today() - days) + query = query.." +day:["..first_day.." TO *]" + local data + if sum_by == nil then + data = db.search( query, 1, 1000000 ) + else + local t = {} + db.advanced_search( query, function(_,doc_fn,_) + local doc = doc_fn() + local key = doc[sum_by] + if key == nil then + logger.warn("missing '"..sum_by.."' in "..stringify(doc)) + return + end + local count = doc.count or error() + local found = t[key] + if found == nil then + t[key] = { + [sum_by] = key + count = count + } + else + found.count = found.count + count + end + end ) + data = {} + for _, val in pairs(t) do + data[#data+1] = val + end + end + + -- for graph + if sum_by == nil or sum_by == "day" then + local last_day = today() + local first_day = last_day - days + local map = {} + for _, el in ipairs(data) do + map[long(el.day)] = el.count + end + data = {} + for date in range(first_day,last_day,day) do + date = long(date) + local count = map[date] or 0 + data[#data+1] = {date,count} + end + return data + else + sort(data,by_count) + local t = {nil} + local n = 0 + for _, el in ipairs(data) do + t[#t+1] = { x=el[sum_by], y=el.count } + n = n + 1 + if n == limit then + break + end + end + return t + end +end + +local function delete_old() + logger.warn "delete_old" + local expired = long( today() - Time.period{days=31} ) + db.delete( "day:[* TO "..expired.."}" ) +end +-- delete_old() +Thread.schedule( delete_old, { repeating_delay=Time.period{days=1} } ) + + +-- for ads + +db.indexed_fields.ad_visit_owner_id = Lucene.type.long + +local visit_period = Time.period{hours=1} +-- visit_period = Time.period{seconds=1} + +function Reporting.maybe_track_visit(owner) + local doc = db.get_document{ ad_visit_owner_id = owner.id } + return doc == nil or doc.time + visit_period < time_now() +end + +function Reporting.should_track_visit(owner) + return db.run_in_transaction( function() + local now = time_now() + local doc = db.get_document{ ad_visit_owner_id = owner.id } + if doc == nil then + db.save{ + type = "ad_visit" + ad_visit_owner_id = owner.id + time = now + } + return true + elseif doc.time + visit_period < now then + doc.time = now + db.save(doc) + return true + else + return false + end + end ) +end + +function Reporting.owners_with_visits() + local owner_ids = list_to_set{} + local docs = db.search("type:ad_visit",1,1000000) + for _, doc in ipairs(docs) do + owner_ids[doc.ad_visit_owner_id] = true + end + return owner_ids +end + +return Reporting