--
-- Copyright 2019 The Android Open Source Project
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
--     https://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
--
/* Create the file RSS table. */
CREATE VIEW file_rss AS
SELECT ts,
       LEAD(ts, 1, ts) OVER(PARTITION BY track_id ORDER BY ts) - ts AS dur,
       upid,
       value AS file
FROM counter
JOIN process_counter_track
  ON counter.track_id = process_counter_track.id
WHERE process_counter_track.name IN ("mem.rss.file", "rss_stat.mm_filepages");

/* Create the anon RSS table. */
CREATE VIEW anon_rss AS
SELECT ts,
       LEAD(ts, 1, ts) OVER(PARTITION BY track_id ORDER BY ts) - ts AS dur,
       upid,
       value AS anon
FROM counter
JOIN process_counter_track
  ON counter.track_id = process_counter_track.id
WHERE process_counter_track.name IN ("mem.rss.anon", "rss_stat.mm_anonpages");

/* Create the oom adj table. */
CREATE VIEW oom_adj AS
SELECT ts,
       LEAD(ts, 1, ts) OVER(PARTITION BY track_id ORDER BY ts) - ts AS dur,
       upid,
       value AS oom_score_adj
FROM counter
JOIN process_counter_track
  ON counter.track_id = process_counter_track.id
WHERE process_counter_track.name = 'oom_score_adj';

/* Harmonise the three tables above into a single table. */
CREATE VIRTUAL TABLE anon_file USING span_join(anon_rss PARTITIONED upid, file_rss PARTITIONED upid);

CREATE VIRTUAL TABLE anon_file_oom USING span_join(anon_file PARTITIONED upid, oom_adj PARTITIONED upid);

/* For each span, compute the RSS (for a given upid) for each category of oom_adj */
CREATE VIEW rss_spans_for_oom_upid AS
SELECT ts,
       dur,
       upid,
       CASE WHEN oom_score_adj < 0 THEN file + anon ELSE 0 END AS rss_oom_lt_zero,
       CASE WHEN oom_score_adj <= 0 THEN file + anon ELSE 0 END AS rss_oom_eq_zero,
       CASE WHEN oom_score_adj < 20 THEN file + anon ELSE 0 END AS rss_fg,
       CASE WHEN oom_score_adj < 700 THEN file + anon ELSE 0 END AS rss_bg,
       CASE WHEN oom_score_adj < 900 THEN file + anon ELSE 0 END AS rss_low_bg,
       file + anon AS rss_cached
FROM anon_file_oom;

/* Convert the raw RSS values to the change in RSS (for a given upid) over time */
CREATE VIEW rss_spans_rss_change AS
SELECT ts,
       dur,
       upid,
       rss_oom_lt_zero - lag(rss_oom_lt_zero, 1, 0) OVER win AS rss_oom_lt_zero_diff,
       rss_oom_eq_zero - lag(rss_oom_eq_zero, 1, 0) OVER win AS rss_oom_eq_zero_diff,
       rss_fg - lag(rss_fg, 1, 0) OVER win AS rss_fg_diff,
       rss_bg - lag(rss_bg, 1, 0) OVER win AS rss_bg_diff,
       rss_low_bg - lag(rss_low_bg, 1, 0) OVER win AS rss_low_bg_diff,
       rss_cached - lag(rss_cached, 1, 0) OVER win AS rss_cached_diff
FROM rss_spans_for_oom_upid
WINDOW win AS (PARTITION BY upid ORDER BY ts);

/*
* Compute a rolling sum of anon + file for each category of process state.
* (note: we no longer consider upid in the windows which means we are now
* computing a rolling sum for all the processes).
*/
CREATE VIEW output AS
SELECT ts,
       lead(ts, 1, ts + dur) OVER win - ts AS dur,
       SUM(rss_oom_lt_zero_diff) OVER win AS rss_oom_lt_zero,
       SUM(rss_oom_eq_zero_diff) OVER win AS rss_oom_eq_zero,
       SUM(rss_fg_diff) OVER win AS rss_fg,
       SUM(rss_bg_diff) OVER win AS rss_bg,
       SUM(rss_low_bg_diff) OVER win AS rss_low_bg,
       SUM(rss_cached_diff) OVER win AS rss_cached
FROM rss_spans_rss_change
WINDOW win AS (ORDER BY ts)
ORDER BY ts;

/*
* Print out the final result (note: dur = 0 is excluded to account for times
* where multiple processes report a new memory value at the same timestamp)
*/
SELECT *
FROM output
WHERE dur > 0;