Add addPrimaryKeyConstraint for AlterTableBuilder

src/schema/alter-table-builder.ts

@@ -36,6 +36,7 @@ import {
 import { AlterTableExecutor } from './alter-table-executor.js'
 import { AlterTableAddForeignKeyConstraintBuilder } from './alter-table-add-foreign-key-constraint-builder.js'
 import { AlterTableDropConstraintBuilder } from './alter-table-drop-constraint-builder.js'
+import { PrimaryConstraintNode } from '../operation-node/primary-constraint-node.js'
 
 /**
  * This builder can be used to create a `alter table` query.
@@ -212,6 +213,23 @@ export class AlterTableBuilder implements ColumnAlteringInterface {
     })
   }
 
+  /**
+   * See {@link CreateTableBuilder.addPrimaryKeyConstraint}
+   */
+  addPrimaryKeyConstraint(
+    constraintName: string,
+    columns: string[]
+  ): AlterTableExecutor {
+    return new AlterTableExecutor({
+      ...this.#props,
+      node: AlterTableNode.cloneWithTableProps(this.#props.node, {
+        addConstraint: AddConstraintNode.create(
+          PrimaryConstraintNode.create(columns, constraintName)
+        ),
+      }),
+    })
+  }
+
   dropConstraint(constraintName: string): AlterTableDropConstraintBuilder {
     return new AlterTableDropConstraintBuilder({
       ...this.#props,

test/node/src/schema.test.ts

@@ -2468,6 +2468,53 @@ for (const dialect of DIALECTS) {
         })
       }
 
+      if (dialect !== 'sqlite') {
+        describe('add primary key constraint', async () => {
+          it('should add a primary key constraint', async () => {
+            const builder = ctx.db.schema
+              .alterTable('test')
+              .addPrimaryKeyConstraint('test_pkey', ['integer_col'])
+
+            testSql(builder, dialect, {
+              postgres: {
+                sql: 'alter table "test" add constraint "test_pkey" primary key ("integer_col")',
+                parameters: [],
+              },
+              mysql: {
+                sql: 'alter table `test` add constraint `test_pkey` primary key (`integer_col`)',
+                parameters: [],
+              },
+              sqlite: NOT_SUPPORTED,
+            })
+
+            await builder.execute()
+          })
+
+          it('should add a primary key constraint for multiple columns', async () => {
+            const builder = ctx.db.schema
+              .alterTable('test')
+              .addPrimaryKeyConstraint('test_pkey', [
+                'integer_col',
+                'varchar_col',
+              ])
+
+            testSql(builder, dialect, {
+              postgres: {
+                sql: 'alter table "test" add constraint "test_pkey" primary key ("integer_col", "varchar_col")',
+                parameters: [],
+              },
+              mysql: {
+                sql: 'alter table `test` add constraint `test_pkey` primary key (`integer_col`, `varchar_col`)',
+                parameters: [],
+              },
+              sqlite: NOT_SUPPORTED,
+            })
+
+            await builder.execute()
+          })
+        })
+      }
+
       if (dialect !== 'sqlite') {
         describe('parse schema name', () => {
           beforeEach(cleanup)