e2net/iceshrimp-legacy
2
0
Fork 0
Code Issues Pull requests Packages Projects Releases Activity

Introduce followed log and following log

Browse source
This commit is contained in:
Akihiko Odaki 2018-04-02 21:57:36 +09:00
parent 7942f76acb
commit 319e905bf9
14 changed files with 101 additions and 117 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/models/followed-log.ts Normal file
View file

@ -0,0 +1,11 @@
import { ObjectID } from 'mongodb';
import db from '../db/mongodb';
const FollowedLog = db.get<IFollowedLog>('followedLogs');
export default FollowedLog;
export type IFollowedLog = {
_id: ObjectID;
userId: ObjectID;
count: number;
};

11
src/models/following-log.ts Normal file
View file

@ -0,0 +1,11 @@
import { ObjectID } from 'mongodb';
import db from '../db/mongodb';
const FollowingLog = db.get<IFollowingLog>('followingLogs');
export default FollowingLog;
export type IFollowingLog = {
_id: ObjectID;
userId: ObjectID;
count: number;
};

1
src/models/following.ts
View file

@ -8,7 +8,6 @@ export default Following;
export type IFollowing = {
_id: mongo.ObjectID;
createdAt: Date;
deletedAt: Date;
followeeId: mongo.ObjectID;
followerId: mongo.ObjectID;
};

12
src/models/user.ts
View file

@ -234,8 +234,7 @@ export const pack = (
_user.isFollowing = (async () => {
const follow = await Following.findOne({
followerId: meId,
followeeId: _user.id,
deletedAt: { $exists: false }
followeeId: _user.id
});
return follow !== null;
})();
@ -244,8 +243,7 @@ export const pack = (
_user.isFollowed = (async () => {
const follow2 = await Following.findOne({
followerId: _user.id,
followeeId: meId,
deletedAt: { $exists: false }
followeeId: meId
});
return follow2 !== null;
})();
@ -275,15 +273,13 @@ export const pack = (
// Get following you know count
_user.followingYouKnowCount = Following.count({
followeeId: { $in: myFollowingIds },
followerId: _user.id,
deletedAt: { $exists: false }
followerId: _user.id
});
// Get followers you know count
_user.followersYouKnowCount = Following.count({
followeeId: _user.id,
followerId: { $in: myFollowingIds },
deletedAt: { $exists: false }
followerId: { $in: myFollowingIds }
});
}
}

4
src/post/distribute.ts
View file

@ -48,9 +48,7 @@ export default async (user, mentions, post) => {
// Fetch all followers
const followers = await Following
.find({
followeeId: user._id,
// 削除されたドキュメントは除く
deletedAt: { $exists: false }
followeeId: user._id
}, {
followerId: true,
_id: false

12
src/processor/http/follow.ts
View file

@ -3,6 +3,8 @@ import { sign } from 'http-signature';
import { URL } from 'url';
import User, { isLocalUser, pack as packUser } from '../../models/user';
import Following from '../../models/following';
import FollowingLog from '../../models/following-log';
import FollowedLog from '../../models/followed-log';
import event from '../../publishers/stream';
import notify from '../../publishers/notify';
import context from '../../remote/activitypub/renderer/context';
@ -21,6 +23,11 @@ export default ({ data }, done) => Following.findOne({ _id: data.following }).th
}
}),
promisedFollower.then(({ followingCount }) => FollowingLog.insert({
userId: followerId,
count: followingCount + 1
})),
// Increment followers count
User.update({ _id: followeeId }, {
$inc: {
@ -28,6 +35,11 @@ export default ({ data }, done) => Following.findOne({ _id: data.following }).th
}
}),
promisedFollowee.then(({ followersCount }) => FollowedLog.insert({
userId: followerId,
count: followersCount + 1
})),
// Notify
promisedFollowee.then(followee => followee.host === null ?
notify(followeeId, followerId, 'follow') : null),

4
src/server/api/common/get-friends.ts
View file

@ -6,9 +6,7 @@ export default async (me: mongodb.ObjectID, includeMe: boolean = true) => {
// SELECT followee
const myfollowing = await Following
.find({
followerId: me,
// 削除されたドキュメントは除く
deletedAt: { $exists: false }
followerId: me
}, {
fields: {
followeeId: true

66
src/server/api/endpoints/aggregation/users/followers.ts
View file

@ -2,8 +2,9 @@
* Module dependencies
*/
import $ from 'cafy';
import { ObjectID } from 'mongodb';
import User from '../../../../../models/user';
import Following from '../../../../../models/following';
import FollowedLog from '../../../../../models/followed-log';
/**
* Aggregate followers of a user
@ -29,47 +30,36 @@ module.exports = (params) => new Promise(async (res, rej) => {
return rej('user not found');
}
const startTime = new Date(new Date().setMonth(new Date().getMonth() - 1));
const following = await Following
.find({
followeeId: user._id,
$or: [
{ deletedAt: { $exists: false } },
{ deletedAt: { $gt: startTime } }
]
}, {
sort: { createdAt: -1 },
fields: {
_id: false,
followerId: false,
followeeId: false
}
});
const today = new Date();
const graph = [];
today.setMinutes(0);
today.setSeconds(0);
today.setMilliseconds(0);
let cursorDate = new Date(today.getTime());
let cursorTime = cursorDate.setDate(new Date(today.getTime()).getDate() + 1);
for (let i = 0; i < 30; i++) {
let day = new Date(new Date().setDate(new Date().getDate() - i));
day = new Date(day.setMilliseconds(999));
day = new Date(day.setSeconds(59));
day = new Date(day.setMinutes(59));
day = new Date(day.setHours(23));
// day = day.getTime();
graph.push(FollowedLog.findOne({
_id: { $lt: ObjectID.createFromTime(cursorTime / 1000) },
userId: user._id
}, {
sort: { _id: -1 },
}).then(log => {
cursorDate = new Date(today.getTime());
cursorTime = cursorDate.setDate(today.getDate() - i);
const count = following.filter(f =>
f.createdAt < day && (f.deletedAt == null || f.deletedAt > day)
).length;
graph.push({
date: {
year: day.getFullYear(),
month: day.getMonth() + 1, // In JavaScript, month is zero-based.
day: day.getDate()
},
count: count
});
return {
date: {
year: cursorDate.getFullYear(),
month: cursorDate.getMonth() + 1, // In JavaScript, month is zero-based.
day: cursorDate.getDate()
},
count: log ? log.count : 0
};
}));
}
res(graph);
res(await Promise.all(graph));
});

65
src/server/api/endpoints/aggregation/users/following.ts
View file

@ -2,8 +2,9 @@
* Module dependencies
*/
import $ from 'cafy';
import { ObjectID } from 'mongodb';
import User from '../../../../../models/user';
import Following from '../../../../../models/following';
import FollowingLog from '../../../../../models/following-log';
/**
* Aggregate following of a user
@ -29,46 +30,36 @@ module.exports = (params) => new Promise(async (res, rej) => {
return rej('user not found');
}
const startTime = new Date(new Date().setMonth(new Date().getMonth() - 1));
const following = await Following
.find({
followerId: user._id,
$or: [
{ deletedAt: { $exists: false } },
{ deletedAt: { $gt: startTime } }
]
}, {
sort: { createdAt: -1 },
fields: {
_id: false,
followerId: false,
followeeId: false
}
});
const today = new Date();
const graph = [];
today.setMinutes(0);
today.setSeconds(0);
today.setMilliseconds(0);
let cursorDate = new Date(today.getTime());
let cursorTime = cursorDate.setDate(new Date(today.getTime()).getDate() + 1);
for (let i = 0; i < 30; i++) {
let day = new Date(new Date().setDate(new Date().getDate() - i));
day = new Date(day.setMilliseconds(999));
day = new Date(day.setSeconds(59));
day = new Date(day.setMinutes(59));
day = new Date(day.setHours(23));
graph.push(FollowingLog.findOne({
_id: { $lt: ObjectID.createFromTime(cursorTime / 1000) },
userId: user._id
}, {
sort: { _id: -1 },
}).then(log => {
cursorDate = new Date(today.getTime());
cursorTime = cursorDate.setDate(today.getDate() - i);
const count = following.filter(f =>
f.createdAt < day && (f.deletedAt == null || f.deletedAt > day)
).length;
graph.push({
date: {
year: day.getFullYear(),
month: day.getMonth() + 1, // In JavaScript, month is zero-based.
day: day.getDate()
},
count: count
});
return {
date: {
year: cursorDate.getFullYear(),
month: cursorDate.getMonth() + 1, // In JavaScript, month is zero-based.
day: cursorDate.getDate()
},
count: log ? log.count : 0
};
}));
}
res(graph);
res(await Promise.all(graph));
});

3
src/server/api/endpoints/following/create.ts
View file

@ -42,8 +42,7 @@ module.exports = (params, user) => new Promise(async (res, rej) => {
// Check if already following
const exist = await Following.findOne({
followerId: follower._id,
followeeId: followee._id,
deletedAt: { $exists: false }
followeeId: followee._id
});
if (exist !== null) {

9
src/server/api/endpoints/following/delete.ts
View file

@ -42,8 +42,7 @@ module.exports = (params, user) => new Promise(async (res, rej) => {
// Check not following
const exist = await Following.findOne({
followerId: follower._id,
followeeId: followee._id,
deletedAt: { $exists: false }
followeeId: followee._id
});
if (exist === null) {
@ -51,12 +50,8 @@ module.exports = (params, user) => new Promise(async (res, rej) => {
}
// Delete following
await Following.update({
await Following.findOneAndDelete({
_id: exist._id
}, {
$set: {
deletedAt: new Date()
}
});
// Send response

3
src/server/api/endpoints/users/followers.ts
View file

@ -46,8 +46,7 @@ module.exports = (params, me) => new Promise(async (res, rej) => {
// Construct query
const query = {
followeeId: user._id,
deletedAt: { $exists: false }
followeeId: user._id
} as any;
// ログインしていてかつ iknow フラグがあるとき

3
src/server/api/endpoints/users/following.ts
View file

@ -46,8 +46,7 @@ module.exports = (params, me) => new Promise(async (res, rej) => {
// Construct query
const query = {
followerId: user._id,
deletedAt: { $exists: false }
followerId: user._id
} as any;
// ログインしていてかつ iknow フラグがあるとき

14
test/api.js
View file

@ -609,20 +609,6 @@ describe('API', () => {
res.should.have.status(204);
}));
it('過去にフォロー歴があった状態でフォローできる', async(async () => {
const hima = await insertHimawari();
const me = await insertSakurako();
await db.get('following').insert({
followeeId: hima._id,
followerId: me._id,
deletedAt: new Date()
});
const res = await request('/following/create', {
userId: hima._id.toString()
}, me);
res.should.have.status(204);
}));
it('既にフォローしている場合は怒る', async(async () => {
const hima = await insertHimawari();
const me = await insertSakurako();