/// <summary>
/// DB のテナント・レジストリ・ノード情報を Cluster(k8s)・ObjectStorage へ同期させるメソッドです。
/// </summary>
protected override void DoWork(object state, int doWorkCount)
{
LogInfo($"DB のテナント・レジストリ・ノード情報を Cluster(k8s) へ同期させます。");
try
{
// DB のテナント情報に対応する名前空間、ロール、クォータを Cluster(k8s) へ同期
var tenants = tenantRepository.GetAllTenants();
foreach (Tenant tenant in tenants)
{
// 名前空間とロールを同期
bool ret = clusterManagementService.RegistTenantAsync(tenant.Name).Result;
if (ret)
{
LogDebug($"DB のテナント \"{tenant.Name}\" に対応する名前空間とロールを Cluster(k8s) へ同期させました。");
}
else
{
LogError($"DB のテナント \"{tenant.Name}\" に対応する名前空間とロールを Cluster(k8s) へ同期させる処理に失敗しました。");
}
// クォータを同期
int cpu = tenant.LimitCpu == null ? 0 : tenant.LimitCpu.Value;
int memory = tenant.LimitMemory == null ? 0 : tenant.LimitMemory.Value;
int gpu = tenant.LimitGpu == null ? 0 : tenant.LimitGpu.Value;
ret = clusterManagementService.SetQuotaAsync(tenant.Name, cpu, memory, gpu).Result;
string quotaInfo = $"cpu={cpu} memory={memory} gpu={gpu}";
if (ret)
{
LogDebug($"DB のテナント \"{tenant.Name}\" に対応するクォータ [{quotaInfo}] を Cluster(k8s) へ同期させました。");
}
if (!ret)
{
LogError($"DB のテナント \"{tenant.Name}\" に対応するクォータ [{quotaInfo}] を Cluster(k8s) へ同期させる処理に失敗しました。");
}
// テナントの古い ClusterToken を削除
tenantRepository.DeleteClusterToken(tenant.Id);
// ObjectStorage への同期処理を行う
if (tenant.Storage != null)
{
// ObjectStorage に バケットを作成する
storageLogic.CreateBucketAsync(tenant, tenant.Storage);
}
}
// テナントの古い ClusterToken 削除を確定する
unitOfWork.Commit();
// DB のレジストリ情報(UserTenantRegistryMap)に対応するシークレットを Cluster(k8s) へ同期
var userTenantRegistryMaps = registryRepository.GetUserTenantRegistryMapAll();
foreach (UserTenantRegistryMap userTenantRegistryMap in userTenantRegistryMaps)
{
// ログ情報
string registryPasswd = string.IsNullOrEmpty(userTenantRegistryMap.RegistryPassword) ? "無し" : "有り";
string mapInfo = $"UserTenantRegistryMapId={userTenantRegistryMap.Id}, UserId={userTenantRegistryMap.UserId}, " +
$"RegistryPassword={registryPasswd}, TenantRegistryMapId={userTenantRegistryMap.TenantRegistryMapId}, " +
$"TernantId={userTenantRegistryMap.TenantRegistryMap.TenantId}, " +
$"RegistryId={userTenantRegistryMap.TenantRegistryMap.RegistryId}";
Registry registry = userTenantRegistryMap.Registry;
if (registry == null)
{
// Registry が null というのはあり得ないが、取り敢えずはチェック
LogDebug($"DB のレジストリ情報 [{mapInfo}] に対応する Registry が存在しません。");
continue;
}
mapInfo += $", RegistryServiceType={registry.ServiceType}";
// テナントの取得
Tenant tenant = tenantRepository.Get(userTenantRegistryMap.TenantRegistryMap.TenantId);
if (tenant == null)
{
// テナントが null というのはあり得ないが、取り敢えずはチェック
LogError($"DB のレジストリ情報 [{ mapInfo}] に対応するテナントが存在しません。");
continue;
}
// RegistryService の取得
IRegistryService registryService = GetRegistryService(registry);
if (registryService == null)
{
LogError($"DB のレジストリ情報 [{mapInfo}] に対応する RegistryService を取得できませんでした。");
continue;
}
// パスワードが空なら同期させない
if (string.IsNullOrEmpty(userTenantRegistryMap.RegistryPassword))
{
LogDebug($"DB のレジストリ情報 [{mapInfo}] のレジストリ・パスワードが空なので Cluster(k8s) への同期は行いません。");
continue;
}
// Docker コンフィグの取得
string dockerCfg = registryService.GetDockerCfgAuthString(userTenantRegistryMap);
if (dockerCfg == null)
{
LogError($"DB のレジストリ情報 [{ mapInfo}] に同期させる Docker コンフィグを取得できませんでした。");
continue;
}
// シークレット情報の生成
var inModel = new RegistRegistryTokenInputModel()
{
TenantName = tenant.Name,
RegistryTokenKey = userTenantRegistryMap.RegistryTokenKey,
DockerCfgAuthString = dockerCfg,
Url = userTenantRegistryMap.Registry.RegistryUrl
};
// Cluster(k8s) にシークレットを同期
bool ret = clusterManagementService.RegistRegistryTokenyAsync(inModel).Result;
if (ret)
{
LogDebug($"DB のレジストリ情報 [{mapInfo}] に対応するシークレットを Cluster(k8s) へ同期させました。");
}
else
{
LogError($"DB のレジストリ情報 [{mapInfo}] に対応するシークレットを Cluster(k8s) へ同期させる処理に失敗しました。");
}
}
// DB のノード情報に対応するパーティションを Cluster(k8s) へ同期
var nodes = nodeRepository.GetAll();
foreach (Node node in nodes)
{
bool ret = clusterManagementLogic.UpdatePartitionLabelAsync(node.Name, node.Partition).Result;
if (!ret)
{
LogError($"DB のノード情報 [{node.Name}] に対応するパーティションを Cluster(k8s) へ同期させる処理に失敗しました。");
continue;
}
ret = clusterManagementLogic.UpdateTensorBoardEnabledLabelAsync(node.Name, node.TensorBoardEnabled).Result;
if (ret)
{
LogDebug($"DB のノード情報 [{node.Name}] に対応する TensorBoard 可否設定を Cluster(k8s) へ同期させました。");
}
else
{
LogError($"DB のノード情報 [{node.Name}] に対応する TensorBoard 可否設定を Cluster(k8s) へ同期させる処理に失敗しました。");
}
ret = clusterManagementLogic.UpdateNotebookEnabledLabelAsync(node.Name, node.NotebookEnabled).Result;
if (ret)
{
LogDebug($"DB のノード情報 [{node.Name}] に対応する Notebook 可否設定を Cluster(k8s) へ同期させました。");
}
else
{
LogError($"DB のノード情報 [{node.Name}] に対応する Notebook 可否設定を Cluster(k8s) へ同期させる処理に失敗しました。");
}
// KQI管理者用名前空間の実行可否設定を Cluster(k8s) へ同期する
ret = SyncKqiAdminEnabledLabel(node).Result;
if (ret)
{
LogDebug($"DB のノード [{node.Name}] に対応する KQI管理者用名前空間 [{containerManageOptions.KqiAdminNamespace}] のアクセス可否設定を Cluster(k8s) へ同期させました。");
}
else
{
LogError($"DB のノード [{node.Name}] に対応する KQI管理者用名前空間 [{containerManageOptions.KqiAdminNamespace}] のアクセス可否設定を Cluster(k8s) へ同期させる処理に失敗しました。");
}
// テナントの実行可否設定を Cluster(k8s) へ同期する
int failedCount = SyncTenantEnabledLabel(node).Result;
if (failedCount == 0)
{
LogInfo($"DB のノード [{node.Name}] に対するテナントの実行可否設定情報を Cluster(k8s) へ同期させる処理は終了しました。");
}
else
{
LogError($"DB のノード [{node.Name}] に対するテナントの実行可否設定情報 {failedCount} 件の Cluster(k8s) へ同期させる処理に失敗しました。");
}
}
LogInfo("DB のテナント・レジストリ・ノード情報を Cluster(k8s) へ同期させる処理は終了しました。");
}
catch (Exception e)
{
//例外をキャッチしたが ERROR ログを出力して処理を継続
LogError($"DB のテナント・レジストリ・ノード情報を Cluster(k8s) へ同期させる時に例外をキャッチしましたが web-api は継続処理します。 例外メッセージ=\"{e.Message}\"");
}
}
public async Task <IActionResult> Edit(long?id, [FromBody] EditInputModel model)
{
//データの入力チェック
if (!ModelState.IsValid || !id.HasValue)
{
return(JsonBadRequest("Invalid inputs."));
}
//データの存在チェック
var tenant = await tenantRepository.GetTenantWithStorageForUpdateAsync(id.Value);
if (tenant == null)
{
return(JsonNotFound($"Tenant ID {id.Value} is not found."));
}
if (model.DefaultGitId != null && model.GitIds.Contains(model.DefaultGitId.Value) == false)
{
//デフォルトGitがGit一覧の中になかったらエラー
return(JsonConflict($"Default Git ID {model.DefaultGitId.Value} does NOT exist in selected gits."));
}
if (model.DefaultRegistryId != null && model.RegistryIds.Contains(model.DefaultRegistryId.Value) == false)
{
//デフォルトレジストリがレジストリ一覧の中になかったらエラー
return(JsonConflict($"Default Registry ID {model.DefaultRegistryId.Value} does NOT exist in selected registries."));
}
if (model.StorageId != null)
{
//データの存在チェック
var storage = tenantRepository.GetStorage(model.StorageId.Value);
if (storage == null)
{
return(JsonNotFound($"The selected storage ID {model.StorageId.Value} is not found."));
}
//バケットを作成する
await storageLogic.CreateBucketAsync(tenant, storage);
}
tenant.DisplayName = model.DisplayName;
tenant.StorageId = model.StorageId;
//コンテナ管理サービス作業
//テナントを登録
var tenantResult = await clusterManagementLogic.RegistTenantAsync(tenant.Name);
if (tenantResult == false)
{
return(JsonError(HttpStatusCode.ServiceUnavailable, "Couldn't create cluster master namespace. Please check the configuration to the connect cluster manager service."));
}
//テナントとGitを紐づけ
//まずは現状のGitを取得して、そこから増減を判断する
var currentGits = gitRepository.GetGitAll(tenant.Id).ToList();
if (model.GitIds != null && model.GitIds.Count() > 0)
{
//デフォルトGitの設定（無ければ一個目）
tenant.DefaultGitId = model.DefaultGitId == null?
model.GitIds.ElementAt(0) : model.DefaultGitId.Value;
foreach (long gitId in model.GitIds)
{
Git currentGit = currentGits.FirstOrDefault(r => r.Id == gitId);
if (currentGit != null)
{
//以前も紐づいていたので、無視。
currentGits.Remove(currentGit);
continue;
}
//データの存在チェック
Git git = await gitRepository.GetByIdAsync(gitId);
if (git == null)
{
return(JsonNotFound($"The selected git ID {gitId} is not found."));
}
await gitRepository.AttachGitToTenantAsync(tenant, git, false);
}
}
//残っているのは削除された紐づけなので、消す
foreach (var removedGit in currentGits)
{
gitRepository.DetachGitFromTenant(tenant, removedGit);
}
//テナントとレジストリを紐づけ
//まずは現状のレジストリを取得して、そこから増減を判断する
var currentRegistries = registryRepository.GetRegistryAll(tenant.Id).ToList();
if (model.RegistryIds != null && model.RegistryIds.Count() > 0)
{
//デフォルトレジストリの設定（無ければ一個目）
tenant.DefaultRegistryId = model.DefaultRegistryId == null?
model.RegistryIds.ElementAt(0) : model.DefaultRegistryId.Value;
foreach (long registryId in model.RegistryIds)
{
Registry currentRegistry = currentRegistries.FirstOrDefault(r => r.Id == registryId);
if (currentRegistry != null)
{
//以前も紐づいていたので、無視。
currentRegistries.Remove(currentRegistry);
continue;
}
//データの存在チェック
Registry registry = await registryRepository.GetByIdAsync(registryId);
if (registry == null)
{
return(JsonNotFound($"The selected registry ID {registryId} is not found."));
}
var maps = await registryRepository.AttachRegistryToTenantAsync(tenant, registry, false);
if (maps != null)
{
foreach (var map in maps)
{
//レジストリを登録
var registryResult = await clusterManagementLogic.RegistRegistryToTenantAsync(tenant.Name, map);
if (registryResult == false)
{
return(JsonError(HttpStatusCode.ServiceUnavailable, "Couldn't map the tenant and the registry in a cluster management service. Please check the configuration to the connect cluster manager service."));
}
}
}
}
}
//残っているのは削除された紐づけなので、消す
foreach (var removedRegistry in currentRegistries)
{
registryRepository.DetachRegistryFromTenant(tenant, removedRegistry);
}
// 関連するクラスタトークンをリセット
tenantRepository.DeleteClusterToken(tenant.Id);
tenantRepository.Update(tenant, unitOfWork);
return(JsonOK(new IndexOutputModel(tenant)));
}