@@ -86,11 +86,10 @@ export async function cleanWorkspace (
8686 opt : { recruit : boolean , tracker : boolean , removedTx : boolean }
8787) : Promise < void > {
8888 const connection = ( await connect ( transactorUrl , workspaceId , undefined , {
89- mode : 'backup' ,
90- model : 'upgrade'
89+ mode : 'backup'
9190 } ) ) as unknown as CoreClient & BackupClient
9291 try {
93- const ops = new TxOperations ( connection , core . account . System )
92+ const ops = new TxOperations ( connection , core . account . System , workspaceId )
9493
9594 const hierarchy = ops . getHierarchy ( )
9695
@@ -164,6 +163,7 @@ export async function cleanWorkspace (
164163 client . close ( )
165164 }
166165 } catch ( err : any ) {
166+ // TODO: Add force-close
167167 console . trace ( err )
168168 } finally {
169169 await connection . close ( )
@@ -214,6 +214,7 @@ export async function cleanRemovedTransactions (workspaceId: WorkspaceUuid, tran
214214 objectId : { $in : removedDocs . map ( ( it ) => it . objectId ) }
215215 } )
216216 await connection . clean (
217+ workspaceId ,
217218 DOMAIN_TX ,
218219 toRemove . map ( ( it ) => it . _id )
219220 )
@@ -232,8 +233,7 @@ export async function cleanRemovedTransactions (workspaceId: WorkspaceUuid, tran
232233
233234export async function optimizeModel ( workspaceId : WorkspaceUuid , transactorUrl : string ) : Promise < void > {
234235 const connection = ( await connect ( transactorUrl , workspaceId , undefined , {
235- mode : 'backup' ,
236- model : 'upgrade'
236+ mode : 'backup'
237237 } ) ) as unknown as CoreClient & BackupClient
238238 try {
239239 let count = 0
@@ -289,7 +289,7 @@ export async function optimizeModel (workspaceId: WorkspaceUuid, transactorUrl:
289289 }
290290 }
291291
292- await connection . clean ( DOMAIN_TX , toRemove )
292+ await connection . clean ( workspaceId , DOMAIN_TX , toRemove )
293293
294294 count += toRemove . length
295295 console . log ( 'processed' , count )
@@ -298,6 +298,8 @@ export async function optimizeModel (workspaceId: WorkspaceUuid, transactorUrl:
298298 } catch ( err : any ) {
299299 console . trace ( err )
300300 } finally {
301+ // TODO: Add force-close
302+ await connection . sendForceClose ( workspaceId )
301303 await connection . close ( )
302304 }
303305}
@@ -307,7 +309,7 @@ export async function cleanArchivedSpaces (workspaceId: WorkspaceUuid, transacto
307309 } ) ) as unknown as CoreClient & BackupClient
308310 try {
309311 const count = 0
310- const ops = new TxOperations ( connection , core . account . System )
312+ const ops = new TxOperations ( connection , core . account . System , workspaceId )
311313 while ( true ) {
312314 const spaces = await connection . findAll ( core . class . Space , { archived : true } , { limit : 1000 } )
313315 if ( spaces . length === 0 ) {
@@ -370,7 +372,7 @@ export async function fixCommentDoubleIdCreate (workspaceId: WorkspaceUuid, tran
370372 // We have found duplicate one, let's rename it.
371373 const doc = TxProcessor . createDoc2Doc < ChatMessage > ( c as unknown as TxCreateDoc < ChatMessage > )
372374 if ( doc . message !== '' && doc . message . trim ( ) !== '<p></p>' ) {
373- await connection . clean ( DOMAIN_TX , [ c . _id ] )
375+ await connection . clean ( workspaceId , DOMAIN_TX , [ c . _id ] )
374376 if ( oldValue . get ( cid ) === doc . message . trim ( ) ) {
375377 console . log ( 'delete tx' , cid , doc . message )
376378 } else {
@@ -379,9 +381,9 @@ export async function fixCommentDoubleIdCreate (workspaceId: WorkspaceUuid, tran
379381 // Remove previous transaction.
380382 c . objectId = generateId ( )
381383 doc . _id = c . objectId as Ref < ChatMessage >
382- await connection . upload ( DOMAIN_TX , [ c ] )
384+ await connection . upload ( workspaceId , DOMAIN_TX , [ c ] )
383385 // Also we need to create snapsot
384- await connection . upload ( DOMAIN_ACTIVITY , [ doc ] )
386+ await connection . upload ( workspaceId , DOMAIN_ACTIVITY , [ doc ] )
385387 }
386388 }
387389 }
@@ -475,7 +477,7 @@ export async function fixSkills (
475477 // fix skills with + and -
476478 if ( step === '3' ) {
477479 console . log ( 'STEP 3' )
478- const ops = new TxOperations ( connection , core . account . System )
480+ const ops = new TxOperations ( connection , core . account . System , workspaceId )
479481 const regex = / \S + (?: [ - + ] \S + ) + / g
480482 const tagsToClean = ( await connection . findAll ( tags . class . TagElement , {
481483 category : {
@@ -532,7 +534,7 @@ export async function fixSkills (
532534 }
533535 } ) ) as TagElement [ ]
534536 goodTags = goodTags . sort ( ( a , b ) => b . title . length - a . title . length ) . filter ( ( t ) => t . title . length > 2 )
535- const ops = new TxOperations ( connection , core . account . System )
537+ const ops = new TxOperations ( connection , core . account . System , workspaceId )
536538 const tagsToClean = ( await connection . findAll ( tags . class . TagElement , {
537539 category : {
538540 $in : [ 'recruit:category:Other' , 'document:category:Other' , 'tracker:category:Other' ] as Ref < TagCategory > [ ]
@@ -669,8 +671,7 @@ export async function restoreRecruitingTaskTypes (
669671 transactorUrl : string
670672) : Promise < void > {
671673 const connection = ( await connect ( transactorUrl , workspaceId , undefined , {
672- mode : 'backup' ,
673- model : 'upgrade'
674+ mode : 'backup'
674675 } ) ) as unknown as CoreClient & BackupClient
675676 const client = getMongoClient ( mongoUrl )
676677 try {
@@ -772,6 +773,7 @@ export async function restoreRecruitingTaskTypes (
772773 statusCategories . sort ( compareCategories )
773774
774775 const createTxNew : TxCreateDoc < TaskType > = {
776+ _uuid : workspaceId ,
775777 _id : generateId ( ) ,
776778 _class : core . class . TxCreateDoc ,
777779 space : core . space . Tx ,
@@ -822,7 +824,9 @@ export async function restoreRecruitingTaskTypes (
822824 } catch ( err : any ) {
823825 console . trace ( err )
824826 } finally {
827+ // TODO: Add force-close
825828 client . close ( )
829+ await connection . sendForceClose ( workspaceId )
826830 await connection . close ( )
827831 }
828832}
@@ -833,8 +837,7 @@ export async function restoreHrTaskTypesFromUpdates (
833837 transactorUrl : string
834838) : Promise < void > {
835839 const connection = ( await connect ( transactorUrl , workspaceId , undefined , {
836- mode : 'backup' ,
837- model : 'upgrade'
840+ mode : 'backup'
838841 } ) ) as unknown as CoreClient & BackupClient
839842 const client = getMongoClient ( mongoUrl )
840843 try {
@@ -927,6 +930,7 @@ export async function restoreHrTaskTypesFromUpdates (
927930 const ofClassClass = hierarchy . getClass ( recruit . class . Applicant )
928931
929932 await db . collection < TxCreateDoc < Doc > > ( DOMAIN_TX ) . insertOne ( {
933+ _uuid : workspaceId ,
930934 _id : generateId ( ) ,
931935 _class : core . class . TxCreateDoc ,
932936 space : core . space . Tx ,
@@ -946,6 +950,7 @@ export async function restoreHrTaskTypesFromUpdates (
946950 } )
947951
948952 createTaskTypeTx = {
953+ _uuid : workspaceId ,
949954 _id : generateId ( ) ,
950955 _class : core . class . TxCreateDoc ,
951956 space : core . space . Tx ,
@@ -980,6 +985,7 @@ export async function restoreHrTaskTypesFromUpdates (
980985 const ofClassClass = hierarchy . getClass ( recruit . class . Vacancy )
981986
982987 await db . collection < TxCreateDoc < Doc > > ( DOMAIN_TX ) . insertOne ( {
988+ _uuid : workspaceId ,
983989 _id : generateId ( ) ,
984990 _class : core . class . TxCreateDoc ,
985991 space : core . space . Tx ,
@@ -999,6 +1005,7 @@ export async function restoreHrTaskTypesFromUpdates (
9991005 } )
10001006
10011007 const createProjectTypeTx : TxCreateDoc < ProjectType > = {
1008+ _uuid : workspaceId ,
10021009 _id : generateId ( ) ,
10031010 _class : core . class . TxCreateDoc ,
10041011 space : core . space . Tx ,
@@ -1031,6 +1038,7 @@ export async function restoreHrTaskTypesFromUpdates (
10311038 console . trace ( err )
10321039 } finally {
10331040 client . close ( )
1041+ await connection . sendForceClose ( workspaceId )
10341042 await connection . close ( )
10351043 }
10361044}
@@ -1127,7 +1135,7 @@ export async function removeDuplicateIds (
11271135 // await updateId(ctx, wsClient, db, storageAdapter, wsDataId, doc)
11281136 // }
11291137 // }
1130- // await wsClient.sendForceClose()
1138+ // await wsClient.sendForceClose(workspaceId )
11311139 // await wsClient.close()
11321140 // await db.collection<MigrationState>(DOMAIN_MIGRATION).insertOne({
11331141 // _id: generateId(),
0 commit comments