setOperationsMixedAllAsFunctionWithSubquery() — drizzle-orm Function Reference

Source Code

integration-tests/tests/sqlite/durable-objects/index.ts lines 2981–3032

	async setOperationsMixedAllAsFunctionWithSubquery(): Promise<void> {
		try {
			await this.beforeEach();
			await setupSetOperationTest(this.db);

			const sq = union(
				this.db.select({ id: users2Table.id, name: users2Table.name }).from(users2Table).where(eq(users2Table.id, 1)),
				except(
					this.db.select({ id: users2Table.id, name: users2Table.name }).from(users2Table).where(
						gte(users2Table.id, 5),
					),
					this.db.select({ id: users2Table.id, name: users2Table.name }).from(users2Table).where(eq(users2Table.id, 7)),
				),
				this.db.select().from(citiesTable).where(gt(citiesTable.id, 1)),
			)
				.orderBy(asc(sql`id`))
				.as('sq');

			const result = await this.db.select().from(sq).limit(4).offset(1);

			expect(result).length(4);

			expect(result).deep.equal([
				{ id: 2, name: 'London' },
				{ id: 3, name: 'Tampa' },
				{ id: 5, name: 'Ben' },
				{ id: 6, name: 'Jill' },
			]);

			expect(() => {
				union(
					this.db
						.select({ id: users2Table.id, name: users2Table.name })
						.from(users2Table).where(eq(users2Table.id, 1)),
					except(
						this.db
							.select({ id: users2Table.id, name: users2Table.name })
							.from(users2Table).where(gte(users2Table.id, 5)),
						this.db
							.select({ id: users2Table.id, name: users2Table.name })
							.from(users2Table).where(eq(users2Table.id, 7)),
					),
					this.db
						.select({ name: users2Table.name, id: users2Table.id })
						.from(citiesTable).where(gt(citiesTable.id, 1)),
				).orderBy(asc(sql`id`)).run();
			}).throw();
		} catch (error: any) {
			console.error(error);
			throw new Error(`setOperationsMixedAllAsFunctionWithSubquery error`);
		}
	}