Added logic for Ticket Repo

Author: Arturo

main.go

@@ -2,20 +2,65 @@ package main
 
 import (
 	"issue-tracker/config"
+	"issue-tracker/model"
 	"issue-tracker/repo"
 	"issue-tracker/utils"
 	"log"
 )
 
 func main() {
 	mongoClient := config.NewMongoConnection()
-
 	ticketrepo := repo.NewMongoTicketRepository(mongoClient.Client)
-	ticket, err := ticketrepo.FindByID("first-id")
+
+	ticket := ticketrepo.FindByID("1234")
+	log.Println(utils.ToString(ticket))
+
+	err := ticketrepo.Delete("1234")
 	if err != nil {
-		log.Fatal("Something went wrong")
+		log.Printf("Something wrong happends: %s\n", err)
 	}
-	log.Println(utils.ToString(ticket))
+	err = ticketrepo.Delete("first-id")
+	if err != nil {
+		log.Printf("Something wrong happends: %s\n", err)
+	}
+
+	newTicket := model.Ticket{
+		ID:          "1234",
+		Title:       "This is a test ticket",
+		Description: "I'm a super Test!!",
+		Status:      model.StatusNew,
+	}
+	err = ticketrepo.Add(newTicket)
+	if err != nil {
+		log.Printf("Something wrong happends: %s\n", err)
+	}
+
+	newTicket.Status = model.StatusReady
+	err = ticketrepo.Update(newTicket)
+	if err != nil {
+		log.Printf("Something wrong happends: %s\n", err)
+	}
+
+	updateNonExistingTicket := model.Ticket{
+		ID:          "567",
+		Title:       "This is a test ticket",
+		Description: "I'm a super Test!!",
+		Status:      model.StatusNew,
+	}
+	err = ticketrepo.Update(updateNonExistingTicket)
+	if err != nil {
+		log.Printf("Something wrong happends: %s\n", err)
+	}
+	err = ticketrepo.Add(updateNonExistingTicket)
+	if err != nil {
+		log.Printf("Something wrong happends: %s\n", err)
+	}
+
+	tickets := ticketrepo.FindAll()
+	log.Println(utils.ToString(tickets))
+
+	tickets = ticketrepo.FindAllByStatus(model.StatusReady)
+	log.Println(utils.ToString(tickets))
 
 	err = mongoClient.CloseMongoConnection()
 	if err != nil {

repo/repository.go

@@ -3,8 +3,10 @@ package repo
 import "issue-tracker/model"
 
 type TicketRepository interface {
-	FindByID(ID string) (model.Ticket, error)
-	FindAllByStatus(status model.Status) ([]model.Ticket, error)
+	FindByID(ID string) model.Ticket
+	FindAll() []model.Ticket
+	FindAllByStatus(status model.Status) []model.Ticket
 	Add(ticket model.Ticket) error
-	Remove(ticket model.Ticket) error
+	Update(ticket model.Ticket) error
+	Delete(ID string) error
 }

repo/ticketrepository.go

@@ -2,6 +2,7 @@ package repo
 
 import (
 	"context"
+	"errors"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"
 	"issue-tracker/model"
@@ -21,24 +22,78 @@ func NewMongoTicketRepository(client *mongo.Client) TicketRepository {
 	}
 }
 
-func (r *MongoTicketRepository) FindByID(ID string) (model.Ticket, error) {
+func (r *MongoTicketRepository) FindByID(ID string) model.Ticket {
 	var result model.Ticket
 	err := r.collection.FindOne(context.TODO(), bson.D{{"id", ID}}).Decode(&result)
 	if err != nil {
 		log.Printf("Unable to retrieve ticket with ID: '%s', error: %s\n", ID, err)
-		return model.Ticket{}, err
+		return model.Ticket{}
 	}
-	return result, nil
+	return result
 }
 
-func (r *MongoTicketRepository) FindAllByStatus(status model.Status) ([]model.Ticket, error) {
-	panic("implement me")
+func (r *MongoTicketRepository) findMany(filter bson.D) []model.Ticket {
+	var results []model.Ticket
+
+	cur, err := r.collection.Find(context.TODO(), filter, nil)
+	if err != nil {
+		log.Printf("Unable to retrieve all tickets, error: %s\n", err)
+		return []model.Ticket{}
+	}
+
+	for cur.Next(context.TODO()) {
+		var elem model.Ticket
+		err := cur.Decode(&elem)
+		if err != nil {
+			log.Printf("Unable to decode ticket, error: %s\n", err)
+		}
+		results = append(results, elem)
+	}
+
+	if err := cur.Err(); err != nil {
+		log.Printf("Something unexpected went wrong while retreaving all tickets, error: %s\n", err)
+	}
+	if err := cur.Close(context.TODO()); err != nil {
+		log.Printf("Unable to close cursor while retreaving all tickets, error: %s\n", err)
+	}
+
+	return results
+}
+
+func (r *MongoTicketRepository) FindAll() []model.Ticket {
+	return r.findMany(bson.D{{}})
+}
+
+func (r *MongoTicketRepository) FindAllByStatus(status model.Status) []model.Ticket {
+	return r.findMany(bson.D{{"status", status}})
 }
 
 func (r *MongoTicketRepository) Add(ticket model.Ticket) error {
-	panic("implement me")
+	_, err := r.collection.InsertOne(context.TODO(), ticket)
+	return err
 }
 
-func (r *MongoTicketRepository) Remove(ticket model.Ticket) error {
-	panic("implement me")
+func (r *MongoTicketRepository) Update(ticket model.Ticket) error {
+	filter := bson.D{{"id", ticket.ID}}
+	update := bson.D{
+		{"$set", bson.D{
+			{"title", ticket.Title},
+			{"description", ticket.Description},
+			{"status", ticket.Status},
+		}},
+	}
+
+	updateResult, err := r.collection.UpdateOne(context.TODO(), filter, update)
+	if updateResult.MatchedCount == 0 || updateResult.ModifiedCount == 0 {
+		return errors.New("Ticket '" + ticket.ID + "' cannot be updated because it has not been created yet")
+	}
+	return err
+}
+
+func (r *MongoTicketRepository) Delete(ID string) error {
+	deleteResult, err := r.collection.DeleteOne(context.TODO(), bson.D{{"id", ID}}, nil)
+	if deleteResult.DeletedCount == 0 {
+		return errors.New("Ticket '" + ID + "' cannot be deleted because it does not exist")
+	}
+	return err
 }