Examples with StackGresDistributedLogsContext io.stackgres.operator.conciliation.distributedlogs.StackGresDistributedLogsContext used on opensource projects

Example 1 with StackGresDistributedLogsContext

use of io.stackgres.operator.conciliation.distributedlogs.StackGresDistributedLogsContext in project stackgres by ongres.

the class OwnerReferenceDecorator method decorate.

@Override
public void decorate(StackGresDistributedLogsContext context, Iterable<? extends HasMetadata> resources) {
    List<OwnerReference> ownerReferences = List.of(ResourceUtil.getOwnerReference(context.getSource()));
    Seq.seq(resources).filter(resource -> Objects.equals(resource.getMetadata().getNamespace(), context.getSource().getMetadata().getNamespace())).filter(resource -> resource.getMetadata().getOwnerReferences().isEmpty()).forEach(resource -> {
        resource.getMetadata().setOwnerReferences(ownerReferences);
        if (resource.getKind().equals("StatefulSet")) {
            StatefulSet sts = (StatefulSet) resource;
            sts.getSpec().getVolumeClaimTemplates().forEach(vct -> vct.getMetadata().setOwnerReferences(ownerReferences));
        }
    });
}

Example 2 with StackGresDistributedLogsContext

use of io.stackgres.operator.conciliation.distributedlogs.StackGresDistributedLogsContext in project stackgres by ongres.

the class DistributedLogsStatefulSet method generateResource.

@Override
public Stream<HasMetadata> generateResource(StackGresDistributedLogsContext context) {
    final StackGresDistributedLogs cluster = context.getSource();
    final ObjectMeta metadata = cluster.getMetadata();
    final String name = metadata.getName();
    final String namespace = metadata.getNamespace();
    final StackGresDistributedLogsPersistentVolume persistentVolume = cluster.getSpec().getPersistentVolume();
    StorageConfig dataStorageConfig = ImmutableStorageConfig.builder().size(persistentVolume.getSize()).storageClass(Optional.ofNullable(persistentVolume.getStorageClass()).orElse(null)).build();
    final PersistentVolumeClaimSpecBuilder volumeClaimSpec = new PersistentVolumeClaimSpecBuilder().withAccessModes("ReadWriteOnce").withResources(dataStorageConfig.getResourceRequirements()).withStorageClassName(dataStorageConfig.getStorageClass());
    final Map<String, String> labels = labelFactory.clusterLabels(cluster);
    final Map<String, String> podLabels = labelFactory.statefulSetPodLabels(cluster);
    Map<String, VolumePair> availableVolumesPairs = volumeDiscoverer.discoverVolumes(context);
    Map<String, Volume> availableVolumes = availableVolumesPairs.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, vp -> vp.getValue().getVolume()));
    final PodTemplateResult buildPodTemplate = podTemplateSpecFactory.getPodTemplateSpec(ImmutableDistributedLogsContainerContext.builder().distributedLogsContext(context).availableVolumes(availableVolumes).dataVolumeName(dataName(cluster)).build());
    StatefulSet clusterStatefulSet = new StatefulSetBuilder().withNewMetadata().withNamespace(namespace).withName(name).withLabels(labels).endMetadata().withNewSpec().withReplicas(1).withSelector(new LabelSelectorBuilder().addToMatchLabels(podLabels).build()).withUpdateStrategy(new StatefulSetUpdateStrategyBuilder().withType("OnDelete").build()).withServiceName(name).withTemplate(buildPodTemplate.getSpec()).withVolumeClaimTemplates(Stream.of(Stream.of(new PersistentVolumeClaimBuilder().withNewMetadata().withNamespace(namespace).withName(dataName(cluster)).withLabels(labels).endMetadata().withSpec(volumeClaimSpec.build()).build())).flatMap(s -> s).toArray(PersistentVolumeClaim[]::new)).endSpec().build();
    var volumeDependencies = buildPodTemplate.claimedVolumes().stream().map(availableVolumesPairs::get).filter(Objects::nonNull).map(VolumePair::getSource).filter(Optional::isPresent).map(Optional::get);
    return Stream.concat(Stream.of(clusterStatefulSet), volumeDependencies);
}

Example 3 with StackGresDistributedLogsContext

use of io.stackgres.operator.conciliation.distributedlogs.StackGresDistributedLogsContext in project stackgres by ongres.

the class OwnerReferenceDecorator method decorate.

@Override
public void decorate(StackGresDistributedLogsContext context, Iterable<? extends HasMetadata> resources) {
    List<OwnerReference> ownerReferences = getOwnerReferences(context.getSource());
    resources.forEach(resource -> {
        if (!resource.getMetadata().getOwnerReferences().isEmpty()) {
            resource.getMetadata().getOwnerReferences().stream().filter(ownerReference -> ownerReference.getApiVersion().equals(CommonDefinition.GROUP + "/" + CommonDefinition.VERSION)).forEach(ownerReference -> ownerReference.setApiVersion(CommonDefinition.GROUP + "/v1beta1"));
        } else {
            resource.getMetadata().setOwnerReferences(ownerReferences);
            if (resource.getKind().equals("StatefulSet")) {
                StatefulSet sts = (StatefulSet) resource;
                sts.getSpec().getVolumeClaimTemplates().forEach(vct -> vct.getMetadata().setOwnerReferences(ownerReferences));
            }
        }
    });
}