202 lines
8.7 KiB
Go
202 lines
8.7 KiB
Go
package devstar_devcontainer
|
||
|
||
import (
|
||
"code.gitea.io/gitea/models/db"
|
||
devstar_devcontainer_models "code.gitea.io/gitea/models/devstar_devcontainer"
|
||
"code.gitea.io/gitea/modules/log"
|
||
DevcontainersVO "code.gitea.io/gitea/routers/web/devcontainer/vo"
|
||
"context"
|
||
"fmt"
|
||
"github.com/google/uuid"
|
||
"regexp"
|
||
"strings"
|
||
"xorm.io/builder"
|
||
)
|
||
|
||
// GetRepoDevcontainerDetails 获取仓库对应 DevContainer 信息
|
||
func GetRepoDevcontainerDetails(ctx context.Context, opts *DevcontainersVO.SearchRepoDevcontainerDetailsVoOptions) (DevcontainersVO.RepoDevcontainerDetailVO, error) {
|
||
|
||
// 0. 构造异常返回时候的空数据
|
||
resultRepoDevcontainerDetail := DevcontainersVO.RepoDevcontainerDetailVO{}
|
||
|
||
// 1. 检查参数是否有效
|
||
if opts == nil || len(opts.UserName) == 0 || len(opts.RepoName) == 0 {
|
||
return resultRepoDevcontainerDetail, devstar_devcontainer_models.ErrFailedToOperateDevstarDevcontainerDB{
|
||
Action: "construct query condition for devContainer user list",
|
||
Message: "invalid search condition",
|
||
}
|
||
}
|
||
|
||
// 2. 查询数据库
|
||
/*
|
||
SELECT devstar_devcontainer.id AS devcontainer_id, devstar_devcontainer.name AS devcontainer_name, devstar_devcontainer.devcontainer_host AS devcontainer_host, devstar_devcontainer.devcontainer_port AS devcontainer_port, devstar_devcontainer.devcontainer_username AS devcontainer_username, devstar_devcontainer.devcontainer_password AS devcontainer_password, devstar_devcontainer.devcontainer_work_dir AS devcontainer_work_dir, devstar_devcontainer.repo_id AS repo_id, repository.name AS repo_name, repository.description AS repo_description
|
||
FROM devstar_devcontainer
|
||
INNER JOIN repository on devstar_devcontainer.repo_id = repository.id
|
||
WHERE
|
||
devstar_devcontainer.user_id IN ( SELECT id FROM user WHERE name = '#{opts.UserName}')
|
||
AND
|
||
devstar_devcontainer.repo_id IN ( SELECT id FROM repository WHERE name = '#{opts.RepoName}');
|
||
*/
|
||
_, err := db.GetEngine(ctx).
|
||
Table("devstar_devcontainer").
|
||
Select("devstar_devcontainer.repo_id AS repo_id, repository.name AS repo_name, repository.description AS repo_description, devstar_devcontainer.id AS devcontainer_id, devstar_devcontainer.name AS devcontainer_name, devstar_devcontainer.devcontainer_host AS devcontainer_host, devstar_devcontainer.devcontainer_port AS devcontainer_port, devstar_devcontainer.devcontainer_username AS devcontainer_username, devstar_devcontainer.devcontainer_password AS devcontainer_password, devstar_devcontainer.devcontainer_work_dir AS devcontainer_work_dir").
|
||
Join("INNER", "repository", "devstar_devcontainer.repo_id = repository.id").
|
||
Where("devstar_devcontainer.user_id IN ( SELECT id FROM user WHERE name = ?) AND devstar_devcontainer.repo_id IN ( SELECT id FROM repository WHERE name = ?)", opts.UserName, opts.RepoName).
|
||
Get(&resultRepoDevcontainerDetail)
|
||
|
||
// 3. 返回
|
||
if err != nil {
|
||
return resultRepoDevcontainerDetail, devstar_devcontainer_models.ErrFailedToOperateDevstarDevcontainerDB{
|
||
Action: fmt.Sprintf("query devcontainer with repo '%v' and username '%v'", opts.RepoName, opts.UserName),
|
||
Message: err.Error(),
|
||
}
|
||
}
|
||
return resultRepoDevcontainerDetail, nil
|
||
}
|
||
|
||
// CreateRepoDevcontainer 创建 DevContainer
|
||
/*
|
||
必要假设：前置中间件已完成检查，确保数据有效：
|
||
- 当前用户为已登录用户
|
||
- 当前用户拥有 repo code写入权限
|
||
- 数据库此前不存在 该用户在该repo创建的 Dev Container
|
||
*/
|
||
func CreateRepoDevcontainer(ctx context.Context, opts *DevcontainersVO.CreateRepoDevcontainerOptions) error {
|
||
// TODO: 调用 k8s client 创建
|
||
// 目前只是 mock数据
|
||
username := opts.Actor.Name
|
||
repoName := opts.Repository.Name
|
||
newDevcontainer := &devstar_devcontainer_models.DevstarDevcontainer{
|
||
Name: getSanitizedDevcontainerName(username, repoName),
|
||
DevcontainerHost: "127.0.0.1",
|
||
DevcontainerPort: 10086,
|
||
DevcontainerUsername: "mock-username",
|
||
DevcontainerPassword: "mock-password",
|
||
DevcontainerWorkDir: "~/workspace/",
|
||
RepoId: opts.Repository.ID,
|
||
UserId: opts.Actor.ID,
|
||
}
|
||
|
||
dbTransactionErr := db.WithTx(ctx, func(ctx context.Context) error {
|
||
rowsAffect, err := db.GetEngine(ctx).
|
||
Table("devstar_devcontainer").
|
||
Insert(newDevcontainer)
|
||
|
||
if err != nil {
|
||
return devstar_devcontainer_models.ErrFailedToOperateDevstarDevcontainerDB{
|
||
Action: fmt.Sprintf("insert new DevContainer for user '%s' in repo '%s'", username, repoName),
|
||
Message: err.Error(),
|
||
}
|
||
} else if rowsAffect == 0 {
|
||
return devstar_devcontainer_models.ErrFailedToOperateDevstarDevcontainerDB{
|
||
Action: fmt.Sprintf("insert new DevContainer for user '%s' in repo '%s'", username, repoName),
|
||
Message: "expected 1 row to be inserted, but got 0",
|
||
}
|
||
}
|
||
|
||
err = claimDevcontainerResource(newDevcontainer)
|
||
if err != nil {
|
||
return devstar_devcontainer_models.ErrFailedToOperateDevstarDevcontainerDB{
|
||
Action: fmt.Sprintf("Failed to claim resource for Dev Container devstar_devcontainer.DevstarDevcontainer %v", newDevcontainer),
|
||
Message: err.Error(),
|
||
}
|
||
}
|
||
return nil
|
||
})
|
||
|
||
return dbTransactionErr
|
||
}
|
||
|
||
// DeleteRepoDevcontainer 按照 仓库 和/或 用户信息删除 DevContainer(s)
|
||
func DeleteRepoDevcontainer(ctx context.Context, opts *DevcontainersVO.DeleteRepoDevcontainerOptions) error {
|
||
if ctx == nil || opts == nil || (opts.Actor == nil && opts.Repository == nil) {
|
||
return devstar_devcontainer_models.ErrFailedToOperateDevstarDevcontainerDB{
|
||
Action: "construct query parameters",
|
||
Message: "Invalid parameters",
|
||
}
|
||
}
|
||
|
||
// 1. 构造查询条件
|
||
sqlDevcontainerCondition := builder.NewCond()
|
||
if opts.Actor != nil {
|
||
sqlDevcontainerCondition = sqlDevcontainerCondition.And(builder.Eq{"user_id": opts.Actor.ID})
|
||
}
|
||
if opts.Repository != nil {
|
||
sqlDevcontainerCondition = sqlDevcontainerCondition.And(builder.Eq{"repo_id": opts.Repository.ID})
|
||
}
|
||
var devcontainersList []devstar_devcontainer_models.DevstarDevcontainer
|
||
|
||
// 2. 开启事务：先获取 devcontainer列表，后删除
|
||
dbTransactionErr := db.WithTx(ctx, func(ctx context.Context) error {
|
||
var err error
|
||
// 2.1 条件查询： user_id 和/或 repo_id
|
||
err = db.GetEngine(ctx).
|
||
Table("devstar_devcontainer").
|
||
Where(sqlDevcontainerCondition).
|
||
Find(&devcontainersList)
|
||
if err != nil {
|
||
return devstar_devcontainer_models.ErrFailedToOperateDevstarDevcontainerDB{
|
||
Action: fmt.Sprintf("find devcontainer(s) with condition '%v'", sqlDevcontainerCondition),
|
||
Message: err.Error(),
|
||
}
|
||
}
|
||
|
||
// 2.2 条件删除： user_id 和/或 repo_id
|
||
_, err = db.GetEngine(ctx).
|
||
Table("devstar_devcontainer").
|
||
Where(sqlDevcontainerCondition).
|
||
Delete()
|
||
if err != nil {
|
||
return devstar_devcontainer_models.ErrFailedToOperateDevstarDevcontainerDB{
|
||
Action: fmt.Sprintf("MARK devcontainer(s) as DELETED with condition '%v'", sqlDevcontainerCondition),
|
||
Message: err.Error(),
|
||
}
|
||
}
|
||
|
||
return nil
|
||
})
|
||
|
||
if dbTransactionErr != nil {
|
||
return dbTransactionErr
|
||
}
|
||
|
||
// 3. 后台启动一个goroutine慢慢回收 Dev Container 资源 (如果回收失败，将会产生孤儿 Dev Container，只能管理员手动识别、删除)
|
||
go func() {
|
||
_ = purgeDevcontainersResource(devcontainersList)
|
||
}()
|
||
|
||
return dbTransactionErr
|
||
}
|
||
|
||
// getSanitizedDevcontainerName 辅助获取当前用户在当前仓库创建的 devContainer名称
|
||
// DevContainer命名规则： 用户名、仓库名各由小于15位的小写字母和数字组成，中间使用'-'分隔，后面使用'-'分隔符拼接32位UUID小写字母数字字符串
|
||
func getSanitizedDevcontainerName(username, repoName string) string {
|
||
regexpNonAlphaNum := regexp.MustCompile(`[^a-zA-Z0-9]`)
|
||
sanitizedUsername := regexpNonAlphaNum.ReplaceAllString(username, "")
|
||
sanitizedRepoName := regexpNonAlphaNum.ReplaceAllString(repoName, "")
|
||
if len(sanitizedUsername) > 15 {
|
||
sanitizedUsername = strings.ToLower(sanitizedUsername[:15])
|
||
}
|
||
if len(sanitizedRepoName) > 15 {
|
||
sanitizedRepoName = strings.ToLower(sanitizedRepoName[:15])
|
||
}
|
||
newUUID, _ := uuid.NewUUID()
|
||
uuidStr := newUUID.String()
|
||
uuidStr = regexpNonAlphaNum.ReplaceAllString(uuidStr, "")
|
||
return fmt.Sprintf("%s-%s-%s", sanitizedUsername, sanitizedRepoName, uuidStr)
|
||
}
|
||
|
||
// purgeDevcontainersResource 辅助函数，用于goroutine后台执行，回收DevContainer资源 // TODO
|
||
func purgeDevcontainersResource(devcontainersList []devstar_devcontainer_models.DevstarDevcontainer) error {
|
||
for _, devContainer := range devcontainersList {
|
||
log.Info("[Goroutine purgeDevcontainersResource]: 正在装模做样地回收资源： devstar_devcontainer_models.DevstarDevcontainer %v", devContainer)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// claimDevcontainerResource
|
||
func claimDevcontainerResource(newDevContainer *devstar_devcontainer_models.DevstarDevcontainer) error {
|
||
log.Info("[claimDevcontainerResource]: 正在装模做样地分配资源： devstar_devcontainer_models.DevstarDevcontainer %v", newDevContainer)
|
||
return nil
|
||
}
|