criado serviço que retorna ultimas posiçoes de um veiculo

Author: luizvnasc

router/router.go

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

store/mongo.go

@@ -70,3 +70,22 @@ func (ms *MongoStore) Veiculos() (veiculos model.Veiculos, err error) {
 	}
 	return
 }
+
+// Linhas 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})
+	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

@@ -9,6 +9,6 @@ type Storer interface {
 	Linhas() (model.Linhas, error)
 	Linha(codigo string) (model.Linha, error)
 	Veiculos() (model.Veiculos, error)
-	// Veiculo(codigo string) (model.Veiculo, error)
+	Veiculo(codigo string) (model.Veiculos, error)
 	// VeiculosLinhas(codigoLinha string) (model.Veiculos, error)
 }