io.github.robothy.sdwebui.sdk.services.CacheableSystemInfoFetcher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sdwebui-java-sdk Show documentation

Stable Diffusion Web UI Java SDK

The newest version!

package io.github.robothy.sdwebui.sdk.services;

import com.fasterxml.jackson.databind.ObjectMapper;
import io.github.robothy.sdwebui.sdk.SdWebuiBeanContainer;
import io.github.robothy.sdwebui.sdk.SdWebuiBeanFactory;
import io.github.robothy.sdwebui.sdk.SystemInfoFetcher;
import io.github.robothy.sdwebui.sdk.models.SystemInfo;
import org.apache.hc.client5.http.classic.HttpClient;
import org.apache.hc.client5.http.classic.methods.HttpGet;
import org.apache.hc.core5.http.ClassicHttpResponse;

import java.io.IOException;

public class CacheableSystemInfoFetcher implements SystemInfoFetcher, SdWebuiBeanFactory {

  private static final String PATH = "/internal/sysinfo";

  private volatile SystemInfo cachedSystemInfo;

  private final String endpoint;

  private final SdWebuiBeanContainer serviceContainer;

  public CacheableSystemInfoFetcher(String endpoint, SdWebuiBeanContainer serviceContainer) {
    this.endpoint = endpoint;
    this.serviceContainer = serviceContainer;
  }

  @Override
  public SystemInfo systemInfo() {
    fetchSystemInfoIfAbsent();
    return cachedSystemInfo;
  }

  @Override
  public SystemInfo getBean(SdWebuiBeanContainer beanContainer) {
    return systemInfo();
  }

  void fetchSystemInfoIfAbsent() {
    if (cachedSystemInfo == null) {
      synchronized (this) {
        if (cachedSystemInfo == null) {
          cachedSystemInfo = fetchSystemInfo();
        }
      }
    }
  }

  SystemInfo fetchSystemInfo() {
    try {
      return this.serviceContainer.getBean(HttpClient.class)
        .execute(new HttpGet(endpoint + PATH), this::parseSystemInfo);
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }

  SystemInfo parseSystemInfo(ClassicHttpResponse response) {
    try {
      return this.serviceContainer.getBean(ObjectMapper.class).readValue(response.getEntity().getContent(), SystemInfo.class);
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }

}