101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func SaveOrgMembers(orgID uint64, members []uint64) {
|
|
defer measure("SaveOrgMembers", time.Now())
|
|
tx := db.MustBegin()
|
|
|
|
var ids = []string{}
|
|
for _, id := range members {
|
|
ids = append(ids, strconv.FormatUint(id, 10))
|
|
}
|
|
tx.MustExec(fmt.Sprintf(`
|
|
delete from org_members
|
|
where
|
|
org_id = %d and
|
|
user_id not in (%s)
|
|
`, orgID, strings.Join(ids, ", ")))
|
|
|
|
var values = []string{}
|
|
for _, id := range members {
|
|
values = append(values, fmt.Sprintf("(%d, %d)", orgID, id))
|
|
}
|
|
tx.MustExec(`
|
|
insert into org_members (org_id, user_id)
|
|
values ` + strings.Join(values, ", ") + `
|
|
on duplicate key update org_id = org_id
|
|
`)
|
|
|
|
if err := tx.Commit(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func SaveTeamMembers(orgID, teamID uint64, members []uint64) {
|
|
defer measure("SaveTeamMembers", time.Now())
|
|
tx := db.MustBegin()
|
|
|
|
var ids = []string{}
|
|
for _, id := range members {
|
|
ids = append(ids, strconv.FormatUint(id, 10))
|
|
}
|
|
tx.MustExec(fmt.Sprintf(`
|
|
delete from team_members
|
|
where
|
|
org_id = %d and
|
|
team_id = %d and
|
|
user_id not in (%s)
|
|
`, orgID, teamID, strings.Join(ids, ", ")))
|
|
|
|
var values = []string{}
|
|
for _, id := range members {
|
|
values = append(values, fmt.Sprintf("(%d, %d, %d)", orgID, teamID, id))
|
|
}
|
|
tx.MustExec(`
|
|
insert into team_members (org_id, team_id, user_id)
|
|
values ` + strings.Join(values, ", ") + `
|
|
on duplicate key update org_id = org_id
|
|
`)
|
|
|
|
if err := tx.Commit(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func SaveTeamRepos(orgID, teamID uint64, repos []uint64) {
|
|
defer measure("SaveTeamRepos", time.Now())
|
|
tx := db.MustBegin()
|
|
|
|
var ids = []string{}
|
|
for _, id := range repos {
|
|
ids = append(ids, strconv.FormatUint(id, 10))
|
|
}
|
|
tx.MustExec(fmt.Sprintf(`
|
|
delete from team_repos
|
|
where
|
|
org_id = %d and
|
|
team_id = %d and
|
|
repo_id not in (%s)
|
|
`, orgID, teamID, strings.Join(ids, ", ")))
|
|
|
|
var values = []string{}
|
|
for _, id := range repos {
|
|
values = append(values, fmt.Sprintf("(%d, %d, %d)", orgID, teamID, id))
|
|
}
|
|
tx.MustExec(`
|
|
insert into team_repos (org_id, team_id, repo_id)
|
|
values ` + strings.Join(values, ", ") + `
|
|
on duplicate key update org_id = org_id
|
|
`)
|
|
|
|
if err := tx.Commit(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|