77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
|
import pymongo
|
||
|
|
from config import DB
|
||
|
|
from utils.meta import _meta
|
||
|
|
from utils.meta import MetaKey
|
||
|
|
|
||
|
|
|
||
|
|
def create_indexes():
|
||
|
|
if "trash" not in DB.collection_names():
|
||
|
|
DB.create_collection("trash", capped=True, size=50 << 20) # 50 MB
|
||
|
|
|
||
|
|
DB.command("compact", "activities")
|
||
|
|
DB.activities.create_index([(_meta(MetaKey.NOTIFICATION), pymongo.ASCENDING)])
|
||
|
|
DB.activities.create_index(
|
||
|
|
[(_meta(MetaKey.NOTIFICATION_UNREAD), pymongo.ASCENDING)]
|
||
|
|
)
|
||
|
|
DB.activities.create_index([("remote_id", pymongo.ASCENDING)])
|
||
|
|
DB.activities.create_index([("activity.object.id", pymongo.ASCENDING)])
|
||
|
|
DB.activities.create_index([("meta.thread_root_parent", pymongo.ASCENDING)])
|
||
|
|
DB.activities.create_index(
|
||
|
|
[
|
||
|
|
("meta.thread_root_parent", pymongo.ASCENDING),
|
||
|
|
("meta.deleted", pymongo.ASCENDING),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
DB.activities.create_index(
|
||
|
|
[("activity.object.id", pymongo.ASCENDING), ("meta.deleted", pymongo.ASCENDING)]
|
||
|
|
)
|
||
|
|
DB.cache2.create_index(
|
||
|
|
[
|
||
|
|
("path", pymongo.ASCENDING),
|
||
|
|
("type", pymongo.ASCENDING),
|
||
|
|
("arg", pymongo.ASCENDING),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
DB.cache2.create_index("date", expireAfterSeconds=3600 * 12)
|
||
|
|
|
||
|
|
# Index for the block query
|
||
|
|
DB.activities.create_index(
|
||
|
|
[
|
||
|
|
("box", pymongo.ASCENDING),
|
||
|
|
("type", pymongo.ASCENDING),
|
||
|
|
("meta.undo", pymongo.ASCENDING),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
# Index for count queries
|
||
|
|
DB.activities.create_index(
|
||
|
|
[
|
||
|
|
("box", pymongo.ASCENDING),
|
||
|
|
("type", pymongo.ASCENDING),
|
||
|
|
("meta.undo", pymongo.ASCENDING),
|
||
|
|
("meta.deleted", pymongo.ASCENDING),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
DB.activities.create_index([("box", pymongo.ASCENDING)])
|
||
|
|
|
||
|
|
# Outbox query
|
||
|
|
DB.activities.create_index(
|
||
|
|
[
|
||
|
|
("box", pymongo.ASCENDING),
|
||
|
|
("type", pymongo.ASCENDING),
|
||
|
|
("meta.undo", pymongo.ASCENDING),
|
||
|
|
("meta.deleted", pymongo.ASCENDING),
|
||
|
|
("meta.public", pymongo.ASCENDING),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
DB.activities.create_index(
|
||
|
|
[
|
||
|
|
("type", pymongo.ASCENDING),
|
||
|
|
("activity.object.type", pymongo.ASCENDING),
|
||
|
|
("activity.object.inReplyTo", pymongo.ASCENDING),
|
||
|
|
("meta.deleted", pymongo.ASCENDING),
|
||
|
|
]
|
||
|
|
)
|