helene/FoundKey
2
0
Fork 0
forked from FoundKeyGang/FoundKey
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix bug

Browse source 
48ea805999 (commitcomment-48584326)
This commit is contained in:
syuilo 2021-03-23 14:54:09 +09:00
parent 48ea805999
commit 5c3a56b283
10 changed files with 42 additions and 39 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/db/redis.ts
View file

@ -1,16 +1,19 @@
import * as redis from 'redis';
import config from '../config';
const client = redis.createClient(
config.redis.port,
config.redis.host,
{
password: config.redis.pass,
prefix: config.redis.prefix,
db: config.redis.db || 0
}
);
export function createConnection() {
return redis.createClient(
config.redis.port,
config.redis.host,
{
password: config.redis.pass,
prefix: config.redis.prefix,
db: config.redis.db || 0
}
);
}
client.subscribe(config.host);
export const subsdcriber = createConnection();
subsdcriber.subscribe(config.host);
export default client;
export const redisClient = createConnection();

6
src/misc/app-lock.ts
View file

@ -1,4 +1,4 @@
import redis from '../db/redis';
import { redisClient } from '../db/redis';
import { promisify } from 'util';
/**
@ -7,8 +7,8 @@ import { promisify } from 'util';
const retryDelay = 100;
const lock: (key: string, timeout?: number) => Promise<() => void>
= redis
? promisify(require('redis-lock')(redis, retryDelay))
= redisClient
? promisify(require('redis-lock')(redisClient, retryDelay))
: async () => () => { };
/**

4
src/server/api/endpoints/admin/server-info.ts
View file

@ -2,7 +2,7 @@ import * as os from 'os';
import * as si from 'systeminformation';
import { getConnection } from 'typeorm';
import define from '../../define';
import redis from '../../../../db/redis';
import { redisClient } from '../../../../db/redis';
export const meta = {
requireCredential: true as const,
@ -115,7 +115,7 @@ export default define(meta, async () => {
os: os.platform(),
node: process.version,
psql: await getConnection().query('SHOW server_version').then(x => x[0].server_version),
redis: redis.server_info.redis_version,
redis: redisClient.server_info.redis_version,
cpu: {
model: os.cpus()[0].model,
cores: os.cpus().length

4
src/server/api/endpoints/get-online-users-count.ts
View file

@ -1,5 +1,5 @@
import define from '../define';
import redis from '../../../db/redis';
import { redisClient } from '../../../db/redis';
import config from '../../../config';
export const meta = {
@ -13,7 +13,7 @@ export const meta = {
export default define(meta, (ps, user) => {
return new Promise((res, rej) => {
redis.pubsub('numsub', config.host, (_, x) => {
redisClient.pubsub('numsub', config.host, (_, x) => {
res({
count: x[1]
});

6
src/server/api/limiter.ts
View file

@ -1,5 +1,5 @@
import * as Limiter from 'ratelimiter';
import limiterDB from '../../db/redis';
import { redisClient } from '../../db/redis';
import { IEndpoint } from './endpoints';
import getAcct from '../../misc/acct/render';
import { User } from '../../models/entities/user';
@ -35,7 +35,7 @@ export default (endpoint: IEndpoint, user: User) => new Promise((ok, reject) =>
id: `${user.id}:${key}:min`,
duration: limitation.minInterval,
max: 1,
db: limiterDB!
db: redisClient
});
minIntervalLimiter.get((err, info) => {
@ -63,7 +63,7 @@ export default (endpoint: IEndpoint, user: User) => new Promise((ok, reject) =>
id: `${user.id}:${key}`,
duration: limitation.duration,
max: limitation.max,
db: limiterDB!
db: redisClient
});
limiter.get((err, info) => {

10
src/server/api/service/discord.ts
View file

@ -4,7 +4,7 @@ import { getJson } from '../../../misc/fetch';
import { OAuth2 } from 'oauth';
import config from '../../../config';
import { publishMainStream } from '../../../services/stream';
import redis from '../../../db/redis';
import { redisClient } from '../../../db/redis';
import { v4 as uuid } from 'uuid';
import signin from '../common/signin';
import { fetchMeta } from '../../../misc/fetch-meta';
@ -96,7 +96,7 @@ router.get('/connect/discord', async ctx => {
response_type: 'code'
};
redis.set(userToken, JSON.stringify(params));
redisClient.set(userToken, JSON.stringify(params));
const oauth2 = await getOAuth2();
ctx.redirect(oauth2!.getAuthorizeUrl(params));
@ -118,7 +118,7 @@ router.get('/signin/discord', async ctx => {
httpOnly: true
});
redis.set(sessid, JSON.stringify(params));
redisClient.set(sessid, JSON.stringify(params));
const oauth2 = await getOAuth2();
ctx.redirect(oauth2!.getAuthorizeUrl(params));
@ -145,7 +145,7 @@ router.get('/dc/cb', async ctx => {
}
const { redirect_uri, state } = await new Promise<any>((res, rej) => {
redis.get(sessid, async (_, state) => {
redisClient.get(sessid, async (_, state) => {
res(JSON.parse(state));
});
});
@ -216,7 +216,7 @@ router.get('/dc/cb', async ctx => {
}
const { redirect_uri, state } = await new Promise<any>((res, rej) => {
redis.get(userToken, async (_, state) => {
redisClient.get(userToken, async (_, state) => {
res(JSON.parse(state));
});
});

10
src/server/api/service/github.ts
View file

@ -4,7 +4,7 @@ import { getJson } from '../../../misc/fetch';
import { OAuth2 } from 'oauth';
import config from '../../../config';
import { publishMainStream } from '../../../services/stream';
import redis from '../../../db/redis';
import { redisClient } from '../../../db/redis';
import { v4 as uuid } from 'uuid';
import signin from '../common/signin';
import { fetchMeta } from '../../../misc/fetch-meta';
@ -95,7 +95,7 @@ router.get('/connect/github', async ctx => {
state: uuid()
};
redis.set(userToken, JSON.stringify(params));
redisClient.set(userToken, JSON.stringify(params));
const oauth2 = await getOath2();
ctx.redirect(oauth2!.getAuthorizeUrl(params));
@ -116,7 +116,7 @@ router.get('/signin/github', async ctx => {
httpOnly: true
});
redis.set(sessid, JSON.stringify(params));
redisClient.set(sessid, JSON.stringify(params));
const oauth2 = await getOath2();
ctx.redirect(oauth2!.getAuthorizeUrl(params));
@ -143,7 +143,7 @@ router.get('/gh/cb', async ctx => {
}
const { redirect_uri, state } = await new Promise<any>((res, rej) => {
redis.get(sessid, async (_, state) => {
redisClient.get(sessid, async (_, state) => {
res(JSON.parse(state));
});
});
@ -194,7 +194,7 @@ router.get('/gh/cb', async ctx => {
}
const { redirect_uri, state } = await new Promise<any>((res, rej) => {
redis.get(userToken, async (_, state) => {
redisClient.get(userToken, async (_, state) => {
res(JSON.parse(state));
});
});

10
src/server/api/service/twitter.ts
View file

@ -2,7 +2,7 @@ import * as Koa from 'koa';
import * as Router from '@koa/router';
import { v4 as uuid } from 'uuid';
import autwh from 'autwh';
import redis from '../../../db/redis';
import { redisClient } from '../../../db/redis';
import { publishMainStream } from '../../../services/stream';
import config from '../../../config';
import signin from '../common/signin';
@ -89,7 +89,7 @@ router.get('/connect/twitter', async ctx => {
const twAuth = await getTwAuth();
const twCtx = await twAuth!.begin();
redis.set(userToken, JSON.stringify(twCtx));
redisClient.set(userToken, JSON.stringify(twCtx));
ctx.redirect(twCtx.url);
});
@ -99,7 +99,7 @@ router.get('/signin/twitter', async ctx => {
const sessid = uuid();
redis.set(sessid, JSON.stringify(twCtx));
redisClient.set(sessid, JSON.stringify(twCtx));
ctx.cookies.set('signin_with_twitter_sid', sessid, {
path: '/',
@ -124,7 +124,7 @@ router.get('/tw/cb', async ctx => {
}
const get = new Promise<any>((res, rej) => {
redis.get(sessid, async (_, twCtx) => {
redisClient.get(sessid, async (_, twCtx) => {
res(twCtx);
});
});
@ -153,7 +153,7 @@ router.get('/tw/cb', async ctx => {
}
const get = new Promise<any>((res, rej) => {
redis.get(userToken, async (_, twCtx) => {
redisClient.get(userToken, async (_, twCtx) => {
res(twCtx);
});
});

2
src/server/api/streaming.ts
View file

@ -5,7 +5,7 @@ import MainStreamConnection from './stream';
import { ParsedUrlQuery } from 'querystring';
import authenticate from './authenticate';
import { EventEmitter } from 'events';
import redisClient from '../../db/redis';
import { subsdcriber as redisClient } from '../../db/redis';
module.exports = (server: http.Server) => {
// Init websocket server

4
src/server/web/index.ts
View file

@ -21,7 +21,7 @@ import { Users, Notes, Emojis, UserProfiles, Pages, Channels, Clips } from '../.
import parseAcct from '../../misc/acct/parse';
import { getNoteSummary } from '../../misc/get-note-summary';
import { getConnection } from 'typeorm';
import redis from '../../db/redis';
import { redisClient } from '../../db/redis';
import locales = require('../../../locales');
const markdown = MarkdownIt({
@ -379,7 +379,7 @@ router.get('/info', async ctx => {
os: os.platform(),
node: process.version,
psql: await getConnection().query('SHOW server_version').then(x => x[0].server_version),
redis: redis.server_info.redis_version,
redis: redisClient.server_info.redis_version,
cpu: {
model: os.cpus()[0].model,
cores: os.cpus().length