(ctx context.Context, repoType, namespace, repoName string)
| 395 | } |
| 396 | |
| 397 | func (c *GitCallbackComponent) setPrivate(ctx context.Context, repoType, namespace, repoName string) error { |
| 398 | var err error |
| 399 | var dataset *database.Dataset |
| 400 | var model *database.Model |
| 401 | if repoType == DatasetRepoType { |
| 402 | dataset, err = c.ds.FindByPath(ctx, namespace, repoName) |
| 403 | if err != nil { |
| 404 | slog.Error("Failed to find dataset", slog.String("namespace", namespace), slog.String("name", repoName)) |
| 405 | return fmt.Errorf("failed to find dataset, error: %w", err) |
| 406 | } |
| 407 | _, err = c.gs.UpdateRepo(ctx, gitserver.UpdateRepoReq{ |
| 408 | Name: dataset.Repository.Name, |
| 409 | Description: dataset.Repository.Description, |
| 410 | Private: true, |
| 411 | DefaultBranch: dataset.Repository.DefaultBranch, |
| 412 | }) |
| 413 | if err != nil { |
| 414 | return fmt.Errorf("failed to update git server dataset to private, error: %w", err) |
| 415 | } |
| 416 | err = c.ds.Update(ctx, *dataset) |
| 417 | if err != nil { |
| 418 | return fmt.Errorf("failed to update database dataset to private, error: %w", err) |
| 419 | } |
| 420 | } else { |
| 421 | model, err = c.ms.FindByPath(ctx, namespace, repoName) |
| 422 | if err != nil { |
| 423 | return fmt.Errorf("failed to find model by path, error: %w", err) |
| 424 | } |
| 425 | _, err = c.gs.UpdateRepo(ctx, gitserver.UpdateRepoReq{ |
| 426 | Name: model.Repository.Name, |
| 427 | Description: model.Repository.Description, |
| 428 | Private: true, |
| 429 | DefaultBranch: model.Repository.DefaultBranch, |
| 430 | }) |
| 431 | if err != nil { |
| 432 | return fmt.Errorf("failed to update git server model to private, error: %w", err) |
| 433 | } |
| 434 | _, err = c.ms.Update(ctx, *model) |
| 435 | if err != nil { |
| 436 | return fmt.Errorf("failed to update database model to private, error: %w", err) |
| 437 | } |
| 438 | } |
| 439 | slog.Info("set repository to private successed.", slog.String("repoType", repoType), slog.String("namespace", namespace), |
| 440 | slog.String("repo", repoName)) |
| 441 | |
| 442 | return err |
| 443 | } |
| 444 | |
| 445 | func (c *GitCallbackComponent) updateModelRuntimeFrameworks(ctx context.Context, repoType, namespace, repoName, ref, fileName string, deleteAction bool) { |
| 446 | slog.Debug("update model relation for git callback", slog.Any("namespace", namespace), slog.Any("repoName", repoName), slog.Any("repoType", repoType), slog.Any("fileName", fileName), slog.Any("branch", ref)) |
no test coverage detected