Nofated095
/
Q2TG
mirror of https://github.com/Nofated095/Q2TG.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

perf: 通过环境变量设置 websocket

This commit is contained in:
Clansty 2023-11-08 16:18:41 +08:00
parent 7ba8e9116a
commit 4987b15e0e
No known key found for this signature in database
3 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Dockerfile
View File

@ -56,5 +56,6 @@ COPY prisma /app/
RUN pnpm exec prisma generate
COPY --from=build /app/build /app/build
ENV TG_CONNECTION=websocket
ENV DATA_DIR=/app/data
CMD pnpm start

8
src/client/Telegram.ts
View File

@ -13,8 +13,8 @@ import TelegramSession from '../models/TelegramSession';
import { LogLevel } from 'telegram/extensions/Logger';
import { BigInteger } from 'big-integer';
import { IterMessagesParams } from 'telegram/client/messages';
import { PromisedWebSockets } from 'telegram/extensions';
import { ConnectionTCPObfuscated } from 'telegram/network';
import { PromisedNetSockets, PromisedWebSockets } from 'telegram/extensions';
import { ConnectionTCPFull, ConnectionTCPObfuscated } from 'telegram/network';
type MessageHandler = (message: Api.Message) => Promise<boolean | void>;
type ServiceMessageHandler = (message: Api.MessageService) => Promise<boolean | void>;
@ -55,8 +55,8 @@ export default class Telegram {
port: parseInt(process.env.PROXY_PORT),
} : undefined,
autoReconnect: true,
networkSocket: PromisedWebSockets,
connection: ConnectionTCPObfuscated,
networkSocket: process.env.TG_CONNECTION === 'websocket' ? PromisedWebSockets : PromisedNetSockets,
connection: process.env.TG_CONNECTION === 'websocket' ? ConnectionTCPObfuscated : ConnectionTCPFull,
},
);
// this.client.logger.setLevel(LogLevel.WARN);

2
src/controllers/SetupController.ts
View File

@ -9,8 +9,6 @@ import { WorkMode } from '../types/definitions';
import OicqClient from '../client/OicqClient';
import { md5Hex } from '../utils/hashing';
import Instance from '../models/Instance';
import db from '../models/db';
import { CustomFile } from 'telegram/client/uploads';
export default class SetupController {
private readonly setupService: SetupService;