Criado serviço que retorna as posiçoes dos veiculos de uma linha

Author: luizvnasc

router/router.go

@@ -76,5 +76,15 @@ func veiculosRoutes(s store.Storer) chi.Router {
 		}
 		json.NewEncoder(w).Encode(veiculo)
 	})
+	r.Get("/linha/{codigo}", func(w http.ResponseWriter, r *http.Request) {
+		codigo := chi.URLParam(r, "codigo")
+		veiculos, err := s.VeiculosLinha(codigo)
+		if err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+			json.NewEncoder(w).Encode(err)
+			return
+		}
+		json.NewEncoder(w).Encode(veiculos)
+	})
 	return r
 }

store/mongo.go

@@ -71,7 +71,7 @@ func (ms *MongoStore) Veiculos() (veiculos model.Veiculos, err error) {
 	return
 }
 
-// Linhas retorna uma lista de veiculo por codigo
+// Veiculo retorna uma lista de veiculo por codigo
 func (ms *MongoStore) Veiculo(codigo string) (veiculos model.Veiculos, err error) {
 
 	cur, err := ms.db.Collection("veiculos").Find(ms.ctx, bson.M{"cod": codigo})
@@ -89,3 +89,22 @@ func (ms *MongoStore) Veiculo(codigo string) (veiculos model.Veiculos, err error
 	}
 	return
 }
+
+// VeiculosLinha retorna uma lista dos veiculos de uma linha
+func (ms *MongoStore) VeiculosLinha(codigo string) (veiculos model.Veiculos, err error) {
+
+	cur, err := ms.db.Collection("veiculos").Find(ms.ctx, bson.M{"codigolinha": codigo})
+	if err != nil {
+		return
+	}
+	for cur.Next(ms.ctx) {
+		var veiculo model.Veiculo
+		err = cur.Decode(&veiculo)
+		if err != nil {
+			veiculos = model.Veiculos{}
+			return
+		}
+		veiculos = append(veiculos, veiculo)
+	}
+	return
+}

store/store.go

@@ -10,5 +10,5 @@ type Storer interface {
 	Linha(codigo string) (model.Linha, error)
 	Veiculos() (model.Veiculos, error)
 	Veiculo(codigo string) (model.Veiculos, error)
-	// VeiculosLinhas(codigoLinha string) (model.Veiculos, error)
+	VeiculosLinha(codigoLinha string) (model.Veiculos, error)
 }