Fix duplicate query insert error message

* Return appropriate error message when attempting to insert a query by ID that already exists in the DB.
kolide · Oct 19, 2020 · 102ea1b · 102ea1b
1 parent 6b576ad
commit 102ea1b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/server/datastore/mysql/queries.go b/server/datastore/mysql/queries.go
@@ -99,7 +99,10 @@ func (d *Datastore) NewQuery(query *kolide.Query, opts ...kolide.OptionalArg) (*
 		) VALUES ( ?, ?, ?, ?, ? )
 	`
 	result, err := db.Exec(sqlStatement, query.Name, query.Description, query.Query, query.Saved, query.AuthorID)
-	if err != nil {
+
+	if err != nil && isDuplicate(err) {
+		return nil, alreadyExists("Query", 0)
+	} else if err != nil {
 		return nil, errors.Wrap(err, "creating new Query")
 	}