feat: add automatic application detection and tagging

internal/repository/job_test.go

@@ -16,9 +16,7 @@ func TestFind(t *testing.T) {
 
 	jobId, cluster, startTime := int64(398998), "fritz", int64(1675957496)
 	job, err := r.Find(&jobId, &cluster, &startTime)
-	if err != nil {
-		t.Fatal(err)
-	}
+	noErr(t, err)
 
 	// fmt.Printf("%+v", job)
 
@@ -31,9 +29,7 @@ func TestFindById(t *testing.T) {
 	r := setup(t)
 
 	job, err := r.FindById(5)
-	if err != nil {
-		t.Fatal(err)
-	}
+	noErr(t, err)
 
 	// fmt.Printf("%+v", job)
 
@@ -46,14 +42,26 @@ func TestGetTags(t *testing.T) {
 	r := setup(t)
 
 	tags, counts, err := r.CountTags(nil)
-	if err != nil {
-		t.Fatal(err)
-	}
+	noErr(t, err)
 
 	fmt.Printf("TAGS %+v \n", tags)
 	// fmt.Printf("COUNTS %+v \n", counts)
 
-	if counts["bandwidth"] != 0 {
-		t.Errorf("wrong tag count \ngot: %d \nwant: 0", counts["bandwidth"])
+	if counts["bandwidth"] != 2 {
+		t.Errorf("wrong tag count \ngot: %d \nwant: 2", counts["bandwidth"])
+	}
+}
+
+func TestHasTag(t *testing.T) {
+	r := setup(t)
+
+	if !r.HasTag(5, "util", "bandwidth") {
+		t.Errorf("Expected has tag")
+	}
+	if r.HasTag(4, "patho", "idle") {
+		t.Errorf("Expected has not tag")
+	}
+	if !r.HasTag(5, "patho", "idle") {
+		t.Errorf("Expected has tag")
 	}
 }

internal/repository/tags.go

@@ -134,6 +134,19 @@ func (r *JobRepository) AddTagOrCreate(jobId int64, tagType string, tagName stri
 	return tagId, nil
 }
 
+func (r *JobRepository) HasTag(jobId int64, tagType string, tagName string) bool {
+	var id int64
+	q := sq.Select("id").From("tag").Join("jobtag ON jobtag.tag_id = tag.id").
+		Where("jobtag.job_id = ?", jobId).Where("tag.tag_type = ?", tagType).
+		Where("tag.tag_name = ?", tagName)
+	err := q.RunWith(r.stmtCache).QueryRow().Scan(&id)
+	if err != nil {
+		return false
+	} else {
+		return true
+	}
+}
+
 // TagId returns the database id of the tag with the specified type and name.
 func (r *JobRepository) TagId(tagType string, tagName string) (tagId int64, exists bool) {
 	exists = true