Updated API for selected dog type

Signed-off-by: Pin <wf6DJd8a3xSSCZbn@protonmail.com>

db/dogselect.go

@@ -0,0 +1,39 @@
+/*
+    Simple API the serves pictures of dogs
+    Copyright (C) 2021  Pin
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Affero General Public License as
+    published by the Free Software Foundation, either version 3 of the
+    License, or (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Affero General Public License for more details.
+
+    You should have received a copy of the GNU Affero General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+*/
+
+package db
+
+import (
+	"dogapi/structs"
+	"log"
+)
+
+func DogSelect(dog string) (structs.DogPicRow) {
+	q := `select * from dog_pictures where breed_id=(select breed_id from breed_manifest where breed=$1) offset floor(random()*(select count(*) from dog_pictures where breed_id=(select breed_id from breed_manifest where breed=$1))) limit 1;`
+	selstat, err := conn.Query(q, dog)
+	if err != nil {
+		log.Fatalln(err)
+	}
+	var rowsel structs.DogPicRow
+	for selstat.Next() {
+	        if err := selstat.Scan(&rowsel.Dog_id, &rowsel.Breed_id, &rowsel.Author, &rowsel.Title, &rowsel.Format, &rowsel.Hash, &rowsel.Url); err != nil {
+			log.Fatal(err)
+		}
+	}
+	return rowsel
+}

handlers/apihandler.go

@@ -29,7 +29,15 @@ func APIHandler(w http.ResponseWriter, r *http.Request) {
 	switch r.Method {
 	case "GET":
 		var dogquery structs.DogPicRow
+		keys := r.URL.Query().Get("dog")
+		if len(keys) == 0 {
 			dogquery = db.RandomSelect()
 			json.NewEncoder(w).Encode(dogquery)
+			return
+		} else {
+			dogquery = db.DogSelect(keys)
+			json.NewEncoder(w).Encode(dogquery)
+			return
+		}
 	}
 }