Add kind column to db

Author: alexander-yakushev

res/migrations/20250106-06-kind.up.sql

@@ -0,0 +1,2 @@
+ALTER TABLE profile
+ADD COLUMN kind TEXT NOT NULL DEFAULT 'flamegraph';

src/flamebin/db.clj

@@ -50,7 +50,7 @@
 
 (defn insert-profile [profile]
   (m/assert Profile profile)
-  (let [{:keys [id file_path profile_type sample_count owner upload_ts
+  (let [{:keys [id file_path profile_type kind sample_count owner upload_ts
                 edit_token config is_public]} profile]
     (log/infof "Inserting profile %s from %s" id owner)
     (with-locking (:lock @db)
@@ -59,6 +59,7 @@
                              {:id            id
                               :file_path     file_path
                               :profile_type  (name profile_type)
+                              :kind          (name kind)
                               :upload_ts     (str upload_ts)
                               :sample_count  sample_count
                               :is_public     is_public
@@ -74,21 +75,21 @@
 
 (defn list-profiles []
   (with-locking (:lock @db)
-    (->> (jdbc/execute! @db ["SELECT id, file_path, profile_type, sample_count, owner, config, upload_ts, is_public FROM profile"])
+    (->> (jdbc/execute! @db ["SELECT id, file_path, profile_type, kind, sample_count, owner, config, upload_ts, is_public FROM profile"])
          (mapv #(-> (unqualify-keys %)
                     (assoc :edit_token nil)
                     (coerce Profile))))))
 
 (defn list-public-profiles [n]
   (with-locking (:lock @db)
-    (->> (jdbc/execute! @db ["SELECT id, file_path, profile_type, sample_count, owner, config, upload_ts, is_public, edit_token FROM profile
+    (->> (jdbc/execute! @db ["SELECT id, file_path, profile_type, kind, sample_count, owner, config, upload_ts, is_public, edit_token FROM profile
 WHERE is_public = 1 ORDER BY upload_ts DESC LIMIT ?" n])
          (mapv #(-> (unqualify-keys %)
                     (coerce Profile))))))
 
 (defn get-profile [profile-id]
   (with-locking (:lock @db)
-    (let [q ["SELECT id, file_path, profile_type, sample_count, owner, config, upload_ts, edit_token, is_public FROM profile WHERE id = ?" profile-id]
+    (let [q ["SELECT id, file_path, profile_type, kind, sample_count, owner, config, upload_ts, edit_token, is_public FROM profile WHERE id = ?" profile-id]
           row (some-> (jdbc/execute-one! @db q)
                       unqualify-keys
                       (coerce Profile))]

src/flamebin/dto.clj

@@ -23,6 +23,7 @@
        :id :nano-id
        :file_path [:and {:gen/fmap #(str % ".dpf")} :string]
        :profile_type :keyword
+       :kind [:enum :flamegraph :diffgraph]
        :sample_count [:maybe nat-int?]
        :owner [:maybe :string]
        :edit_token [:maybe :string]

test/flamebin/db_test.clj

@@ -26,19 +26,19 @@
 
 (deftest manual-test
   (with-temp-db-and-state
-    (db/insert-profile (dto/->Profile "QcXAqv" "some-path.dpf" "cpu" 12345
+    (db/insert-profile (dto/->Profile "QcXAqv" "some-path.dpf" "cpu" :flamegraph 12345
                                       nil "alhdslfglksjdfhg" true "HAA" inst1))
     (is (= {:id "QcXAqv", :file_path "some-path.dpf", :profile_type :cpu,
             :upload_ts inst1, :sample_count 12345, :owner nil, :is_public true
-            :config "HAA", :edit_token "alhdslfglksjdfhg"}
+            :config "HAA", :edit_token "alhdslfglksjdfhg", :kind :flamegraph}
            (db/get-profile "QcXAqv")))
     (is (inst? (:upload_ts (db/get-profile "QcXAqv"))))
 
-    (db/insert-profile (dto/->Profile "tX8nuc" "another-path.dpf" "alloc" 54321
-                                      "me" nil false nil inst1))
+    (db/insert-profile (dto/->Profile "tX8nuc" "another-path.dpf" "alloc"
+                                      :diffgraph 54321 "me" nil false nil inst1))
     (is (= {:id "tX8nuc", :file_path "another-path.dpf", :edit_token nil
             :owner "me", :sample_count 54321, :profile_type :alloc, :is_public false
-            :config nil, :upload_ts inst1}
+            :config nil, :upload_ts inst1, :kind :diffgraph}
            (db/get-profile "tX8nuc")))))
 
 ;;;; Generative testing