Skip to content
Snippets Groups Projects
Commit 165e0670 authored by Neil Johnson's avatar Neil Johnson
Browse files

Revert "change monthly_active_users table to be a single column"

This reverts commit ec716a35.
parent ec716a35
No related branches found
No related tags found
No related merge requests found
......@@ -38,18 +38,22 @@ class MonthlyActiveUsersStore(SQLBaseStore):
return count
return self.runInteraction("count_users", _count_users)
def insert_monthly_active_user(self, user_id):
def upsert_monthly_active_user(self, user_id):
"""
Updates or inserts monthly active user member
Arguments:
user_id (str): user to add/update
"""
return self._simple_insert(
return self._simple_upsert(
desc="upsert_monthly_active_user",
table="monthly_active_users",
values={
keyvalues={
"user_id": user_id,
},
values={
"timestamp": int(self._clock.time_msec()),
},
lock=False,
)
@defer.inlineCallbacks
......
......@@ -15,7 +15,9 @@
-- a table of monthly active users, for use where blocking based on mau limits
CREATE TABLE monthly_active_users (
user_id TEXT NOT NULL
user_id TEXT NOT NULL,
timestamp BIGINT NOT NULL
);
CREATE UNIQUE INDEX monthly_active_users_users ON monthly_active_users(user_id);
CREATE INDEX monthly_active_users_time_stamp ON monthly_active_users(timestamp);
......@@ -22,7 +22,7 @@ class MonthlyActiveUsersTestCase(tests.unittest.TestCase):
count = yield self.mau.get_monthly_active_count()
self.assertEqual(0, count)
yield self.mau.insert_monthly_active_user("@user:server")
yield self.mau.upsert_monthly_active_user("@user:server")
count = yield self.mau.get_monthly_active_count()
self.assertEqual(1, count)
......@@ -34,8 +34,8 @@ class MonthlyActiveUsersTestCase(tests.unittest.TestCase):
user_id3 = "@user3:server"
result = yield self.mau.is_user_monthly_active(user_id1)
self.assertFalse(result)
yield self.mau.insert_monthly_active_user(user_id1)
yield self.mau.insert_monthly_active_user(user_id2)
yield self.mau.upsert_monthly_active_user(user_id1)
yield self.mau.upsert_monthly_active_user(user_id2)
result = yield self.mau.is_user_monthly_active(user_id1)
self.assertTrue(result)
result = yield self.mau.is_user_monthly_active(user_id3)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment