feat: add /chats/stats to display usage estimate

l3s-learnweb · Aug 12, 2024 · 68147bd · 68147bd
1 parent 6fd2c93
commit 68147bd
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 3 deletions.
diff --git a/interweb-server/src/main/java/de/l3s/interweb/server/InterwebApplication.java b/interweb-server/src/main/java/de/l3s/interweb/server/InterwebApplication.java
@@ -24,7 +24,7 @@
         securitySchemes = {
             @SecurityScheme(
                 securitySchemeName = "JWT",
-                description = "Use /users/login to obtain a JWT token.",
+                description = "Use /login to obtain a JWT token.",
                 type = SecuritySchemeType.HTTP,
                 scheme = "bearer",
                 bearerFormat = "JWT"

diff --git a/interweb-server/src/main/java/de/l3s/interweb/server/features/chat/Chat.java b/interweb-server/src/main/java/de/l3s/interweb/server/features/chat/Chat.java
@@ -49,11 +49,11 @@ public class Chat extends PanacheEntityBase {
     public String title;
 
     @NotNull
-    @ColumnDefault("0")
+    @Column(name = "used_tokens")
     public Integer usedTokens = 0;
 
     @NotNull
-    @ColumnDefault("0")
+    @Column(name = "estimated_cost")
     public Double estimatedCost = 0d;
 
     @CreationTimestamp

diff --git a/interweb-server/src/main/java/de/l3s/interweb/server/features/chat/ChatsResource.java b/interweb-server/src/main/java/de/l3s/interweb/server/features/chat/ChatsResource.java
@@ -87,4 +87,12 @@ public Uni<Void> delete(@PathParam("uuid") UUID id) {
 
         return Chat.findById(apikey, id).call(PanacheEntityBase::delete).replaceWithVoid();
     }
+
+    @GET
+    @Path("/stats")
+    public Uni<ChatsStats> chat() {
+        ApiKey apikey = securityIdentity.getCredential(ApiKey.class);
+
+        return ChatsStats.findByApikey(apikey);
+    }
 }
diff --git a/interweb-server/src/main/java/de/l3s/interweb/server/features/chat/ChatsStats.java b/interweb-server/src/main/java/de/l3s/interweb/server/features/chat/ChatsStats.java
@@ -0,0 +1,28 @@
+package de.l3s.interweb.server.features.chat;
+
+import io.quarkus.runtime.annotations.RegisterForReflection;
+import io.smallrye.mutiny.Uni;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+import de.l3s.interweb.server.features.user.ApiKey;
+
+@RegisterForReflection
+public class ChatsStats {
+    @JsonProperty("used_tokens")
+    public Long usedTokens;
+    @JsonProperty("estimated_cost")
+    public Double estimatedCost;
+    @JsonProperty("total_chats")
+    public Long totalChats;
+
+    public ChatsStats(Long usedTokens, Double estimatedCost, Long totalChats) {
+        this.usedTokens = usedTokens;
+        this.estimatedCost = estimatedCost;
+        this.totalChats = totalChats;
+    }
+
+    public static Uni<ChatsStats> findByApikey(ApiKey apikey) {
+        return Chat.find("select sum(usedTokens) as usedTokens, sum(estimatedCost) as estimatedCost, count(*) as totalChats from Chat where apikey.id = ?1", apikey.id).project(ChatsStats.class).singleResult();
+    }
+}