Optimize account creation

pull/128/head
Chocobozzz 2017-11-16 18:40:50 +01:00
parent 41dbdb8acf
commit 47e0652b4a
No known key found for this signature in database
GPG Key ID: 583A612D890159BE
8 changed files with 44 additions and 26 deletions

View File

@ -38,5 +38,5 @@ const rl = createInterface({
rl.on('line', line => { rl.on('line', line => {
const log = JSON.parse(line) const log = JSON.parse(line)
logLevels[log.level](log.message) logLevels[log.level](log.message, log.stack)
}) })

View File

@ -70,7 +70,7 @@ usersRouter.post('/',
usersRouter.post('/register', usersRouter.post('/register',
ensureUserRegistrationAllowed, ensureUserRegistrationAllowed,
usersRegisterValidator, usersRegisterValidator,
asyncMiddleware(registerUser) asyncMiddleware(registerUserRetryWrapper)
) )
usersRouter.put('/me', usersRouter.put('/me',
@ -113,7 +113,7 @@ async function getUserVideos (req: express.Request, res: express.Response, next:
async function createUserRetryWrapper (req: express.Request, res: express.Response, next: express.NextFunction) { async function createUserRetryWrapper (req: express.Request, res: express.Response, next: express.NextFunction) {
const options = { const options = {
arguments: [ req, res ], arguments: [ req ],
errorMessage: 'Cannot insert the user with many retries.' errorMessage: 'Cannot insert the user with many retries.'
} }
@ -123,7 +123,7 @@ async function createUserRetryWrapper (req: express.Request, res: express.Respon
return res.type('json').status(204).end() return res.type('json').status(204).end()
} }
async function createUser (req: express.Request, res: express.Response, next: express.NextFunction) { async function createUser (req: express.Request) {
const body: UserCreate = req.body const body: UserCreate = req.body
const user = db.User.build({ const user = db.User.build({
username: body.username, username: body.username,
@ -139,7 +139,18 @@ async function createUser (req: express.Request, res: express.Response, next: ex
logger.info('User %s with its channel and account created.', body.username) logger.info('User %s with its channel and account created.', body.username)
} }
async function registerUser (req: express.Request, res: express.Response, next: express.NextFunction) { async function registerUserRetryWrapper (req: express.Request, res: express.Response, next: express.NextFunction) {
const options = {
arguments: [ req ],
errorMessage: 'Cannot insert the user with many retries.'
}
await retryTransactionWrapper(registerUser, options)
return res.type('json').status(204).end()
}
async function registerUser (req: express.Request) {
const body: UserCreate = req.body const body: UserCreate = req.body
const user = db.User.build({ const user = db.User.build({
@ -152,7 +163,8 @@ async function registerUser (req: express.Request, res: express.Response, next:
}) })
await createUserAccountAndChannel(user) await createUserAccountAndChannel(user)
return res.type('json').status(204).end()
logger.info('User %s with its channel and account registered.', body.username)
} }
async function getUserInformation (req: express.Request, res: express.Response, next: express.NextFunction) { async function getUserInformation (req: express.Request, res: express.Response, next: express.NextFunction) {

View File

@ -2,10 +2,9 @@ import * as passwordGenerator from 'password-generator'
import { UserRole } from '../../shared' import { UserRole } from '../../shared'
import { logger, mkdirpPromise, rimrafPromise } from '../helpers' import { logger, mkdirpPromise, rimrafPromise } from '../helpers'
import { createUserAccountAndChannel } from '../lib' import { createUserAccountAndChannel } from '../lib'
import { createLocalAccount } from '../lib/user' import { createLocalAccountWithoutKeys } from '../lib/user'
import { applicationExist, clientsExist, usersExist } from './checker' import { applicationExist, clientsExist, usersExist } from './checker'
import { CACHE, CONFIG, LAST_MIGRATION_VERSION, SERVER_ACCOUNT_NAME } from './constants' import { CACHE, CONFIG, LAST_MIGRATION_VERSION, SERVER_ACCOUNT_NAME } from './constants'
import { database as db } from './database' import { database as db } from './database'
async function installApplication () { async function installApplication () {
@ -136,5 +135,6 @@ async function createApplicationIfNotExist () {
const applicationInstance = await db.Application.create({ migrationVersion: LAST_MIGRATION_VERSION }) const applicationInstance = await db.Application.create({ migrationVersion: LAST_MIGRATION_VERSION })
logger.info('Creating application account.') logger.info('Creating application account.')
return createLocalAccount(SERVER_ACCOUNT_NAME, null, applicationInstance.id, undefined)
return createLocalAccountWithoutKeys(SERVER_ACCOUNT_NAME, null, applicationInstance.id, undefined)
} }

View File

@ -5,16 +5,17 @@ import { database as db } from '../initializers'
import { CONFIG } from '../initializers/constants' import { CONFIG } from '../initializers/constants'
import { UserInstance } from '../models' import { UserInstance } from '../models'
import { createVideoChannel } from './video-channel' import { createVideoChannel } from './video-channel'
import { logger } from '../helpers/logger'
async function createUserAccountAndChannel (user: UserInstance, validateUser = true) { async function createUserAccountAndChannel (user: UserInstance, validateUser = true) {
const res = await db.sequelize.transaction(async t => { const { account, videoChannel } = await db.sequelize.transaction(async t => {
const userOptions = { const userOptions = {
transaction: t, transaction: t,
validate: validateUser validate: validateUser
} }
const userCreated = await user.save(userOptions) const userCreated = await user.save(userOptions)
const accountCreated = await createLocalAccount(user.username, user.id, null, t) const accountCreated = await createLocalAccountWithoutKeys(user.username, user.id, null, t)
const videoChannelName = `Default ${userCreated.username} channel` const videoChannelName = `Default ${userCreated.username} channel`
const videoChannelInfo = { const videoChannelInfo = {
@ -25,18 +26,23 @@ async function createUserAccountAndChannel (user: UserInstance, validateUser = t
return { account: accountCreated, videoChannel } return { account: accountCreated, videoChannel }
}) })
return res // Set account keys, this could be long so process after the account creation and do not block the client
const { publicKey, privateKey } = await createPrivateAndPublicKeys()
account.set('publicKey', publicKey)
account.set('privateKey', privateKey)
account.save().catch(err => logger.error('Cannot set public/private keys of local account %d.', account.id, err))
return { account, videoChannel }
} }
async function createLocalAccount (name: string, userId: number, applicationId: number, t: Sequelize.Transaction) { async function createLocalAccountWithoutKeys (name: string, userId: number, applicationId: number, t: Sequelize.Transaction) {
const { publicKey, privateKey } = await createPrivateAndPublicKeys()
const url = getActivityPubUrl('account', name) const url = getActivityPubUrl('account', name)
const accountInstance = db.Account.build({ const accountInstance = db.Account.build({
name, name,
url, url,
publicKey, publicKey: null,
privateKey, privateKey: null,
followersCount: 0, followersCount: 0,
followingCount: 0, followingCount: 0,
inboxUrl: url + '/inbox', inboxUrl: url + '/inbox',
@ -56,5 +62,5 @@ async function createLocalAccount (name: string, userId: number, applicationId:
export { export {
createUserAccountAndChannel, createUserAccountAndChannel,
createLocalAccount createLocalAccountWithoutKeys
} }

View File

@ -25,9 +25,7 @@ async function createVideoChannel (videoChannelInfo: VideoChannelCreate, account
// Do not forget to add Account information to the created video channel // Do not forget to add Account information to the created video channel
videoChannelCreated.Account = account videoChannelCreated.Account = account
await sendCreateVideoChannel(videoChannelCreated, t) // No need to seed this empty video channel to followers
await shareVideoChannelByServer(videoChannelCreated, t)
return videoChannelCreated return videoChannelCreated
} }

View File

@ -75,7 +75,7 @@ export default function defineAccount (sequelize: Sequelize.Sequelize, DataTypes
}, },
publicKey: { publicKey: {
type: DataTypes.STRING(CONSTRAINTS_FIELDS.ACCOUNTS.PUBLIC_KEY.max), type: DataTypes.STRING(CONSTRAINTS_FIELDS.ACCOUNTS.PUBLIC_KEY.max),
allowNull: false, allowNull: true,
validate: { validate: {
publicKeyValid: value => { publicKeyValid: value => {
const res = isAccountPublicKeyValid(value) const res = isAccountPublicKeyValid(value)

View File

@ -15,7 +15,7 @@ describe('Test config', function () {
let server = null let server = null
before(async function () { before(async function () {
this.timeout(120000) this.timeout(10000)
await flushTests() await flushTests()
server = await runServer(1) server = await runServer(1)
@ -29,9 +29,11 @@ describe('Test config', function () {
}) })
it('Should have a correct config on a server with registration enabled and a users limit', async function () { it('Should have a correct config on a server with registration enabled and a users limit', async function () {
await registerUser(server.url, 'user1', 'super password') await Promise.all([
await registerUser(server.url, 'user2', 'super password') registerUser(server.url, 'user1', 'super password'),
await registerUser(server.url, 'user3', 'super password') registerUser(server.url, 'user2', 'super password'),
registerUser(server.url, 'user3', 'super password')
])
const res = await getConfig(server.url) const res = await getConfig(server.url)
const data = res.body const data = res.body

View File

@ -3,7 +3,7 @@ export * from './clients'
export * from './config' export * from './config'
export * from './login' export * from './login'
export * from './miscs' export * from './miscs'
export * from './pods' export * from './follows'
export * from './request-schedulers' export * from './request-schedulers'
export * from './requests' export * from './requests'
export * from './servers' export * from './servers'