22 lines
624 B
Go
22 lines
624 B
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
)
|
|
|
|
func SeedFriendships(db *sql.DB) error {
|
|
ts := time.Now().UnixMilli()
|
|
_, err := db.Exec(`
|
|
INSERT INTO friendships (user_id, friend_id, star_id, status, intimacy, created_at, updated_at)
|
|
SELECT a.id, b.id, $1, 'accepted', 0, $2, $2
|
|
FROM users a, users b
|
|
WHERE a.id BETWEEN $3 AND $4
|
|
AND b.id BETWEEN $3 AND $4
|
|
AND a.id != b.id
|
|
AND ((a.id - $3) + 1) % 10 = ((b.id - $3) % 10)
|
|
ON CONFLICT (user_id, friend_id, star_id) DO NOTHING
|
|
`, LoadtestStarID, ts, LoadtestUserMin, LoadtestUserMax)
|
|
return err
|
|
}
|