Merge pull request #6 from prisma/feat/no-datasource

feat: Work without datasource name

README.md

@@ -4,7 +4,7 @@ This [Prisma Client Extension](https://www.prisma.io/docs/concepts/components/pr
 
 ## Requirements
 
-Works best with Prisma 5.1+. Can work with earlier versions (4.16.2+) if [no result extensions](https://www.prisma.io/docs/concepts/components/prisma-client/client-extensions/result) are used at the same time.
+Requires Prisma 5.2+.
 
 ## Installation
 
@@ -38,13 +38,11 @@ import { readReplicas } from '@prisma/extension-read-replicas
 
 const prisma = new PrismaClient().$extends(
   readReplicas({
-    db: 'postgresql://replica.example.com:5432/db',
+    url: 'postgresql://replica.example.com:5432/db',
   }),
 )
 ```
 
-Where `db` is the name of your datasource (`datasource` block in the schema file).
-
 All non-transactional read queries will now be executed against the defined replica.  
 Write queries and transactions will be executed against the primary server.
 
@@ -55,7 +53,7 @@ You can also initialize the extension with an array of replica connection string
 ```ts
 const prisma = new PrismaClient().$extends(
   readReplicas({
-    db: [
+    url: [
       'postgresql://replica-1.example.com:5432/db',
       'postgresql://replica-2.example.com:5432/db',
     ],

demo/index.ts

@@ -1,7 +1,7 @@
 import { PrismaClient } from '@prisma/client'
 import { readReplicas } from '..'
 
-const client = new PrismaClient().$extends(readReplicas({ db: process.env.REPLICA_URL! }))
+const client = new PrismaClient().$extends(readReplicas({ url: process.env.REPLICA_URL! }))
 
 async function main() {
   await client.user.deleteMany()

package.json

@@ -39,7 +39,7 @@
   "author": "",
   "license": "Apache-2.0",
   "devDependencies": {
-    "@prisma/client": "5.1.0-integration-fix-result-query-ext.1",
+    "@prisma/client": "5.2.0-dev.72",
     "@swc/core": "^1.3.73",
     "@swc/jest": "^0.2.27",
     "@types/debug": "^4.1.8",
@@ -57,7 +57,7 @@
     "jest": "^29.6.2",
     "lint-staged": "^13.2.3",
     "prettier": "^3.0.0",
-    "prisma": "5.1.0-integration-fix-result-query-ext.1",
+    "prisma": "5.2.0-dev.72",
     "typescript": "^5.1.6"
   },
   "dependencies": {

src/ReplicaManager.ts

@@ -1,7 +1,7 @@
 import { PrismaClient } from '@prisma/client/extension'
 
 type PrismaConstructorOptions = {
-  datasources?: Record<string, { url: string }>
+  datasourceUrl?: string
 }
 
 export type ConfigureReplicaCallback = (client: PrismaClient) => PrismaClient
@@ -12,21 +12,16 @@ interface PrismaClientConstructor {
 type ReplicaManagerOptions = {
   clientConstructor: PrismaClientConstructor
   replicaUrls: string[]
-  datasourceName: string
   configureCallback: ConfigureReplicaCallback | undefined
 }
 
 export class ReplicaManager {
   private _replicaClients: PrismaClient[]
 
-  constructor({ replicaUrls, datasourceName, clientConstructor, configureCallback }: ReplicaManagerOptions) {
-    this._replicaClients = replicaUrls.map((url) => {
+  constructor({ replicaUrls, clientConstructor, configureCallback }: ReplicaManagerOptions) {
+    this._replicaClients = replicaUrls.map((datasourceUrl) => {
       const client = new clientConstructor({
-        datasources: {
-          [datasourceName]: {
-            url,
-          },
-        },
+        datasourceUrl,
       })
 
       if (configureCallback) {

src/extension.ts

@@ -4,7 +4,7 @@ import { Prisma } from '@prisma/client/extension'
 import { ConfigureReplicaCallback, ReplicaManager } from './ReplicaManager'
 
 export type ReplicasOptions = {
-  [datasource: string]: string | string[]
+  url: string | string[]
 }
 
 const debug = createDebug('prisma:replicasExtension')
@@ -31,17 +31,16 @@ export const readReplicas = (options: ReplicasOptions, configureReplicaClient?:
     if (!datasourceName) {
       throw new Error(`Read replicas options must specify a datasource`)
     }
-    let urls = options[datasourceName]
-    if (typeof urls === 'string') {
-      urls = [urls]
-    } else if (!Array.isArray(urls)) {
+    let replicaUrls = options.url
+    if (typeof replicaUrls === 'string') {
+      replicaUrls = [replicaUrls]
+    } else if (!Array.isArray(replicaUrls)) {
       throw new Error(`Replica URLs must be a string or list of strings`)
     }
 
     const replicaManager = new ReplicaManager({
-      replicaUrls: urls,
+      replicaUrls,
       clientConstructor: PrismaClient,
-      datasourceName,
       configureCallback: configureReplicaClient,
     })
 

tests/extension.test.ts

@@ -15,7 +15,7 @@ function createPrisma() {
     .$extends(
       readReplicas(
         {
-          db: process.env.REPLICA_URL!,
+          url: process.env.REPLICA_URL!,
         },
         (client) =>
           (client as PrismaClient).$extends({