mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2024-10-25 07:55:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

Add seed target for inserting seed data

Browse source 
Co-Authored-By: Nathan Sobo <nathan@zed.dev>
This commit is contained in:
Max Brunsfeld 2021-08-05 11:56:36 -07:00
parent 1910a4c1be
commit 775bf8dd33
2 changed files with 115 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
server/migrations/20210805175147_create_chat_tables.sql
View file

@ -4,6 +4,8 @@ CREATE TABLE IF NOT EXISTS "orgs" (
"slug" VARCHAR NOT NULL
);
CREATE UNIQUE INDEX "index_orgs_slug" ON "orgs" ("slug");
CREATE TABLE IF NOT EXISTS "org_memberships" (
"id" SERIAL PRIMARY KEY,
"org_id" INTEGER REFERENCES orgs (id) NOT NULL,
@ -11,8 +13,8 @@ CREATE TABLE IF NOT EXISTS "org_memberships" (
"admin" BOOLEAN NOT NULL
);
CREATE UNIQUE INDEX "index_org_memberships_user_id" ON "org_memberships" ("user_id");
CREATE UNIQUE INDEX "index_org_memberships_org_id" ON "org_memberships" ("org_id");
CREATE INDEX "index_org_memberships_user_id" ON "org_memberships" ("user_id");
CREATE UNIQUE INDEX "index_org_memberships_org_id_and_user_id" ON "org_memberships" ("org_id", "user_id");
CREATE TABLE IF NOT EXISTS "channels" (
"id" SERIAL PRIMARY KEY,
@ -21,7 +23,7 @@ CREATE TABLE IF NOT EXISTS "channels" (
"name" VARCHAR NOT NULL
);
CREATE UNIQUE INDEX "index_channels_owner" ON "channels" ("owner_is_user", "owner_id");
CREATE UNIQUE INDEX "index_channels_owner_and_name" ON "channels" ("owner_is_user", "owner_id", "name");
CREATE TABLE IF NOT EXISTS "channel_memberships" (
"id" SERIAL PRIMARY KEY,
@ -30,8 +32,8 @@ CREATE TABLE IF NOT EXISTS "channel_memberships" (
"admin" BOOLEAN NOT NULL
);
CREATE UNIQUE INDEX "index_channel_memberships_user_id" ON "channel_memberships" ("user_id");
CREATE UNIQUE INDEX "index_channel_memberships_channel_id" ON "channel_memberships" ("channel_id");
CREATE INDEX "index_channel_memberships_user_id" ON "channel_memberships" ("user_id");
CREATE UNIQUE INDEX "index_channel_memberships_channel_id_and_user_id" ON "channel_memberships" ("channel_id", "user_id");
CREATE TABLE IF NOT EXISTS "channel_messages" (
"id" SERIAL PRIMARY KEY,
@ -41,18 +43,4 @@ CREATE TABLE IF NOT EXISTS "channel_messages" (
"sent_at" TIMESTAMP
);
CREATE UNIQUE INDEX "index_channel_messages_channel_id" ON "channel_messages" ("channel_id");
INSERT INTO users (github_login, admin) VALUES ('iamnbutler', true);
DO $$
DECLARE
zed_org_id INTEGER;
max_id INTEGER;
nathan_id INTEGER;
antonio_id INTEGER;
nate_id INTEGER;
BEGIN
INSERT INTO "orgs" (name, slug) VALUES ('Zed', 'zed') RETURNING id into zed_org_id;
END $$;
CREATE INDEX "index_channel_messages_channel_id" ON "channel_messages" ("channel_id");

107
server/src/bin/seed.rs Normal file
View file

@ -0,0 +1,107 @@
use sqlx::postgres::PgPoolOptions;
use tide::log;
#[path = "../env.rs"]
mod env;
#[async_std::main]
async fn main() {
if let Err(error) = env::load_dotenv() {
log::error!(
"error loading .env.toml (this is expected in production): {}",
error
);
}
let database_url = std::env::var("DATABASE_URL").expect("missing DATABASE_URL env var");
let db = PgPoolOptions::new()
.max_connections(5)
.connect(&database_url)
.await
.expect("failed to connect to postgres database");
let zed_users = ["nathansobo", "maxbrunsfeld", "as-cii", "iamnbutler"];
let mut zed_user_ids = Vec::<i32>::new();
for zed_user in zed_users {
zed_user_ids.push(
sqlx::query_scalar(
r#"
INSERT INTO users
(github_login, admin)
VALUES
($1, true)
ON CONFLICT (github_login) DO UPDATE SET
github_login=EXCLUDED.github_login
RETURNING id
"#,
)
.bind(zed_user)
.fetch_one(&db)
.await
.expect("failed to insert user"),
)
}
let zed_org_id: i32 = sqlx::query_scalar(
r#"
INSERT INTO orgs
(name, slug)
VALUES
('Zed', 'zed')
ON CONFLICT (slug) DO UPDATE SET
slug=EXCLUDED.slug
RETURNING id
"#,
)
.fetch_one(&db)
.await
.expect("failed to insert org");
let general_channel_id: i32 = sqlx::query_scalar(
r#"
INSERT INTO channels
(owner_is_user, owner_id, name)
VALUES
(false, $1, 'General')
ON CONFLICT (owner_is_user, owner_id, name) DO UPDATE SET
name=EXCLUDED.name
RETURNING id
"#,
)
.bind(zed_org_id)
.fetch_one(&db)
.await
.expect("failed to insert channel");
for user_id in zed_user_ids {
sqlx::query(
r#"
INSERT INTO org_memberships
(org_id, user_id, admin)
VALUES
($1, $2, true)
ON CONFLICT DO NOTHING
"#,
)
.bind(zed_org_id)
.bind(user_id)
.execute(&db)
.await
.expect("failed to insert org membership");
sqlx::query(
r#"
INSERT INTO channel_memberships
(channel_id, user_id, admin)
VALUES
($1, $2, true)
ON CONFLICT DO NOTHING
"#,
)
.bind(general_channel_id)
.bind(user_id)
.execute(&db)
.await
.expect("failed to insert channel membership");
}
}