Function main

samples/orm/main.ts:7–119

()

import { schema } from './zenstack/schema';

async function main() {
    const db = new ZenStackClient(schema, {
        dialect: new SqliteDialect({ database: new SQLite('./zenstack/dev.db') }),
        computedFields: {
            user: {
                postCount: (eb, { modelAlias }) =>
                    eb
                        .selectFrom('Post')
                        .whereRef('Post.authorId', '=', sql.ref(`${modelAlias}.id`))
                        .select(({ fn }) => fn.countAll<number>().as('postCount')),
            },
        },
        procedures: {
            signUp: ({ client, args }) =>
                client.user.create({
                    data: { ...args },
                }),
            listPublicPosts: ({}) => [],
        },
    }).$use({
        id: 'cost-logger',
        onQuery: async ({ model, operation, args, proceed }) => {
            const start = Date.now();
            const result = await proceed(args as any);
            console.log(`[cost] ${model} ${operation} took ${Date.now() - start}ms`);
            return result;
        },
    });

    // clean up existing data
    await db.post.deleteMany();
    await db.profile.deleteMany();
    await db.user.deleteMany();

    db.user.findMany({ where: { id: '1' } });

    // create users and some posts
    const user1 = await db.user.create({
        data: {
            email: 'yiming@gmail.com',
            role: 'ADMIN',
            posts: {
                create: [
                    {
                        title: 'Post1',
                        content: 'An unpublished post',
                        published: false,
                    },
                    {
                        title: 'Post2',
                        content: 'A published post',
                        published: true,
                    },
                ],
            },
        },
        include: { posts: true },
    });