Convert primary keys to UUID string

Author: claabs

src/entity/MarkovOptions.ts

@@ -1,10 +1,11 @@
-import { Entity, BaseEntity, PrimaryGeneratedColumn, Column, OneToOne } from 'typeorm';
+import { Entity, BaseEntity, Column, OneToOne, PrimaryColumn, Generated } from 'typeorm';
 import { MarkovRoot } from './MarkovRoot';
 
 @Entity()
 export class MarkovOptions extends BaseEntity {
-  @PrimaryGeneratedColumn()
-  id: number;
+  @PrimaryColumn({ type: 'text' })
+  @Generated('uuid')
+  id: string;
 
   @Column()
   stateSize: number;

src/entity/MarkovRoot.ts

@@ -2,19 +2,20 @@ import {
   Entity,
   BaseEntity,
   OneToOne,
-  PrimaryGeneratedColumn,
   OneToMany,
   JoinColumn,
-  ValueTransformer,
+  PrimaryColumn,
+  Generated,
 } from 'typeorm';
 import { MarkovCorpusEntry } from './MarkovCorpusEntry';
 import { MarkovFragment } from './MarkovFragment';
 import { MarkovOptions } from './MarkovOptions';
 
 @Entity()
 export class MarkovRoot extends BaseEntity {
-  @PrimaryGeneratedColumn()
-  id: number;
+  @PrimaryColumn({ type: 'text' })
+  @Generated('uuid')
+  id: string;
 
   @OneToMany(() => MarkovCorpusEntry, (entry) => entry.markov, { nullable: true })
   corpus: MarkovCorpusEntry[];

src/index.ts

@@ -23,7 +23,7 @@ export type MarkovConstructorOptions = {
   /**
    * Used to set the options database ID manually
    */
-  id?: number;
+  id?: string;
 
   /**
    * The stateSize is the number of words for each "link" of the generated sentence.
@@ -38,7 +38,7 @@ export type MarkovConstructorProps = {
   /**
    * Used to set a root database ID manually
    */
-  id?: number;
+  id?: string;
 
   /**
    * Global Markov corpus generation options
@@ -109,7 +109,7 @@ export default class Markov {
 
   public options: MarkovOptions | MarkovDataMembers;
 
-  public id: number;
+  public id: string;
 
   private defaultOptions: MarkovDataMembers = {
     stateSize: 2,
@@ -150,7 +150,7 @@ export default class Markov {
   }
 
   private construct() {
-    this.id = this.constructorProps?.id || 1;
+    this.id = this.constructorProps?.id || '1';
 
     // Save options
     this.options = Object.assign(this.defaultOptions, this.constructorProps?.options);

test/test.ts

@@ -195,7 +195,7 @@ describe('Markov class', () => {
     });
 
     it('should export the original database values', async () => {
-      markov = new Markov();
+      markov = new Markov({ id: '1', options: { id: '1' } });
       connection = await createConnection();
       await markov.addData(data);
 

test/v4-export.json

@@ -1,5 +1,5 @@
 {
-    "id": 1,
+    "id": "1",
     "corpus": [
         {
             "id": 1,
@@ -664,7 +664,7 @@
         }
     ],
     "options": {
-        "id": 1,
+        "id": "1",
         "stateSize": 2
     }
 }