Optimize account creation
This commit is contained in:
parent
41dbdb8acf
commit
47e0652b4a
|
@ -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)
|
||||||
})
|
})
|
||||||
|
|
|
@ -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) {
|
||||||
|
|
|
@ -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)
|
||||||
}
|
}
|
||||||
|
|
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -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)
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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'
|
||||||
|
|
Loading…
Reference in New Issue