create bookings service

Author: mikebellcoder

bookings/cmd/app/handlers.go

@@ -0,0 +1,93 @@
+package main
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"github.com/gorilla/mux"
+	"github.com/mikebellcoder/microservices-docker-go-mongodb/bookings/pkg/models"
+)
+
+func (app *application) all(w http.ResponseWriter, r *http.Request) {
+	// Get all bookings stored
+	bookings, err := app.bookings.All()
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	// Convert booking list into json encoding
+	b, err := json.Marshal(bookings)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	app.infoLog.Println("Bookings have been listed")
+
+	// Send response back
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(http.StatusOK)
+	w.Write(b)
+}
+
+func (app *application) findByID(w http.ResponseWriter, r *http.Request) {
+	// Get id from incoming url
+	vars := mux.Vars(r)
+	id := vars["id"]
+
+	// Find booking by id
+	m, err := app.bookings.FindByID(id)
+	if err != nil {
+		if err.Error() == "ErrNoDocuments" {
+			app.infoLog.Println("Booking not found")
+			return
+		}
+
+		app.serverError(w, err)
+	}
+
+	// Convert booking to json encoding
+	b, err := json.Marshal(m)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	app.infoLog.Println("Found a booking")
+
+	// Send response back
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(http.StatusOK)
+	w.Write(b)
+}
+
+func (app *application) insert(w http.ResponseWriter, r *http.Request) {
+	// Define booking model
+	var m models.Booking
+
+	// Get request information
+	err := json.NewDecoder(r.Body).Decode(&m)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	// Insert new booking
+	insertResult, err := app.bookings.Insert(m)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	app.infoLog.Printf("New booking has been created, id=%s\n", insertResult.InsertedID)
+}
+
+func (app *application) delete(w http.ResponseWriter, r *http.Request) {
+	// Get id from incoming url
+	vars := mux.Vars(r)
+	id := vars["id"]
+
+	// Delete booking by id
+	deleteResult, err := app.bookings.Delete(id)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	app.infoLog.Printf("Deleted %d booking(s)\n", deleteResult.DeletedCount)
+}

bookings/cmd/app/helpers.go

@@ -0,0 +1,18 @@
+package main
+
+import (
+	"fmt"
+	"net/http"
+	"runtime/debug"
+)
+
+func (app *application) serverError(w http.ResponseWriter, err error) {
+	trace := fmt.Sprintf("%s\n%s", err.Error(), debug.Stack())
+	app.errorLog.Output(2, trace)
+
+	http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
+}
+
+func (app *application) clientError(w http.ResponseWriter, status int) {
+	http.Error(w, http.StatusText(status), status)
+}

bookings/cmd/app/main.go

@@ -0,0 +1,83 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"time"
+
+	"github.com/mikebellcoder/microservices-docker-go-mongodb/bookings/pkg/models/mongodb"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+type application struct {
+	errorLog *log.Logger
+	infoLog  *log.Logger
+	bookings *mongodb.BookingModel
+}
+
+func main() {
+	// Define command-line flags
+	serverAddr := flag.String("serverAddr", "", "HTTP server network address")
+	serverPort := flag.Int("serverPort", 4000, "HTTP server network port")
+	mongoURI := flag.String("mongoURI", "mongodb://localhost:27017", "Database hostname url")
+	mongoDatabase := flag.String("mongoDatabase", "bookings", "Database name")
+	enableCredentials := flag.Bool("enableCredentials", false, "Enable the use of credentials for mongo connection")
+	flag.Parse()
+
+	// Create loggers
+	infoLog := log.New(os.Stdout, "INFO\t", log.Ldate|log.Ltime)
+	errLog := log.New(os.Stderr, "ERROR\t", log.Ldate|log.Ltime|log.Lshortfile)
+
+	// Create client config
+	co := options.Client().ApplyURI(*mongoURI)
+	if *enableCredentials {
+		co.Auth = &options.Credential{
+			Username: os.Getenv("MONGODB_USERNAME"),
+			Password: os.Getenv("MONGODB_PASSWORD"),
+		}
+	}
+
+	// Establish DB connection
+	ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
+	defer cancel()
+
+	client, err := mongo.Connect(ctx, co)
+	if err != nil {
+		errLog.Fatal(err)
+	}
+
+	defer func() {
+		if err = client.Disconnect(ctx); err != nil {
+			panic(err)
+		}
+	}()
+
+	// Initialize new app instance
+	app := &application{
+		infoLog:  infoLog,
+		errorLog: errLog,
+		bookings: &mongodb.BookingModel{
+			C: client.Database(*mongoDatabase).Collection("bookings"),
+		},
+	}
+
+	// Initialize an http.Server
+	serverURI := fmt.Sprintf("%s:%d", *serverAddr, *serverPort)
+	srv := &http.Server{
+		Addr:         serverURI,
+		ErrorLog:     errLog,
+		Handler:      app.routes(),
+		IdleTimeout:  time.Minute,
+		ReadTimeout:  5 * time.Second,
+		WriteTimeout: 10 * time.Second,
+	}
+
+	infoLog.Printf("Starting server on %s\n", serverURI)
+	err = srv.ListenAndServe()
+	errLog.Fatal(err)
+}

bookings/cmd/app/routes.go

@@ -0,0 +1,14 @@
+package main
+
+import "github.com/gorilla/mux"
+
+func (app *application) routes() *mux.Router {
+	// Register handler functions.
+	r := mux.NewRouter()
+	r.HandleFunc("/api/bookings/", app.all).Methods("GET")
+	r.HandleFunc("/api/bookings/{id}", app.findByID).Methods("GET")
+	r.HandleFunc("/api/bookings/", app.insert).Methods("POST")
+	r.HandleFunc("/api/bookings/{id}", app.delete).Methods("DELETE")
+
+	return r
+}

bookings/go.mod

@@ -0,0 +1,18 @@
+module github.com/mikebellcoder/microservices-docker-go-mongodb/bookings
+
+go 1.22.5
+
+require (
+	github.com/golang/snappy v0.0.4 // indirect
+	github.com/gorilla/mux v1.8.1 // indirect
+	github.com/klauspost/compress v1.16.7 // indirect
+	github.com/montanaflynn/stats v0.7.1 // indirect
+	github.com/xdg-go/pbkdf2 v1.0.0 // indirect
+	github.com/xdg-go/scram v1.1.2 // indirect
+	github.com/xdg-go/stringprep v1.0.4 // indirect
+	github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 // indirect
+	go.mongodb.org/mongo-driver v1.17.2 // indirect
+	golang.org/x/crypto v0.26.0 // indirect
+	golang.org/x/sync v0.8.0 // indirect
+	golang.org/x/text v0.17.0 // indirect
+)

bookings/go.sum

@@ -0,0 +1,48 @@
+github.com/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM=
+github.com/golang/snappy v0.0.4/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
+github.com/gorilla/mux v1.8.1 h1:TuBL49tXwgrFYWhqrNgrUNEY92u81SPhu7sTdzQEiWY=
+github.com/gorilla/mux v1.8.1/go.mod h1:AKf9I4AEqPTmMytcMc0KkNouC66V3BtZ4qD5fmWSiMQ=
+github.com/klauspost/compress v1.16.7 h1:2mk3MPGNzKyxErAw8YaohYh69+pa4sIQSC0fPGCFR9I=
+github.com/klauspost/compress v1.16.7/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
+github.com/montanaflynn/stats v0.7.1 h1:etflOAAHORrCC44V+aR6Ftzort912ZU+YLiSTuV8eaE=
+github.com/montanaflynn/stats v0.7.1/go.mod h1:etXPPgVO6n31NxCd9KQUMvCM+ve0ruNzt6R8Bnaayow=
+github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c=
+github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
+github.com/xdg-go/scram v1.1.2 h1:FHX5I5B4i4hKRVRBCFRxq1iQRej7WO3hhBuJf+UUySY=
+github.com/xdg-go/scram v1.1.2/go.mod h1:RT/sEzTbU5y00aCK8UOx6R7YryM0iF1N2MOmC3kKLN4=
+github.com/xdg-go/stringprep v1.0.4 h1:XLI/Ng3O1Atzq0oBs3TWm+5ZVgkq2aqdlvP9JtoZ6c8=
+github.com/xdg-go/stringprep v1.0.4/go.mod h1:mPGuuIYwz7CmR2bT9j4GbQqutWS1zV24gijq1dTyGkM=
+github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 h1:ilQV1hzziu+LLM3zUTJ0trRztfwgjqKnBWNtSRkbmwM=
+github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78/go.mod h1:aL8wCCfTfSfmXjznFBSZNN13rSJjlIOI1fUNAtF7rmI=
+github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
+go.mongodb.org/mongo-driver v1.17.2 h1:gvZyk8352qSfzyZ2UMWcpDpMSGEr1eqE4T793SqyhzM=
+go.mongodb.org/mongo-driver v1.17.2/go.mod h1:Hy04i7O2kC4RS06ZrhPRqj/u4DTYkFDAAccj+rVKqgQ=
+golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
+golang.org/x/crypto v0.26.0 h1:RrRspgV4mU+YwB4FYnuBoKsUapNIL5cohGAmSH3azsw=
+golang.org/x/crypto v0.26.0/go.mod h1:GY7jblb9wI+FOo5y8/S2oY4zWP07AkOJ4+jxCqdqn54=
+golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
+golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
+golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
+golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
+golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.8.0 h1:3NFvSEYkUoMifnESzZl15y791HH1qU2xm6eCJU5ZPXQ=
+golang.org/x/sync v0.8.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
+golang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=
+golang.org/x/text v0.17.0 h1:XtiM5bkSOt+ewxlOE/aE/AKEHibwj/6gvWMl9Rsh0Qc=
+golang.org/x/text v0.17.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
+golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
+golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=

bookings/pkg/models/models.go

@@ -0,0 +1,11 @@
+package models
+
+import "go.mongodb.org/mongo-driver/bson/primitive"
+
+// Booking is used to represent booking profile data
+type Booking struct {
+	ID         primitive.ObjectID `bson:"_id,omitempty"`
+	UserID     string             `bson:"userid"`
+	ShowtimeID string             `bson:"showtimeid"`
+	Movies     []string           `bson:"movies"`
+}

bookings/pkg/models/mongodb/bookings.go

@@ -0,0 +1,69 @@
+package mongodb
+
+import (
+	"context"
+	"errors"
+
+	"github.com/mikebellcoder/microservices-docker-go-mongodb/bookings/pkg/models"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+)
+
+// BookingModel represents a mgo database session with a booking data model
+type BookingModel struct {
+	C *mongo.Collection
+}
+
+// All method will be used to get all records from bookings collection
+func (m *BookingModel) All() ([]models.Booking, error) {
+	// Define variables
+	ctx := context.TODO()
+	b := []models.Booking{}
+
+	// Find all bookings
+	bookingCursor, err := m.C.Find(ctx, bson.M{})
+	if err != nil {
+		return nil, err
+	}
+	err = bookingCursor.All(ctx, &b)
+	if err != nil {
+		return nil, err
+	}
+
+	return b, err
+}
+
+// FindByID will be used to find a booking registry by id
+func (m *BookingModel) FindByID(id string) (*models.Booking, error) {
+	p, err := primitive.ObjectIDFromHex(id)
+	if err != nil {
+		return nil, err
+	}
+
+	var booking = models.Booking{}
+	err = m.C.FindOne(context.TODO(), bson.M{"_id": p}).Decode(&booking)
+	if err != nil {
+		// Check if the booking was not found
+		if err == mongo.ErrNoDocuments {
+			return nil, errors.New("ErrNoDocuments")
+		}
+		return nil, err
+	}
+
+	return &booking, nil
+}
+
+// Insert will be used to insert a new booking registry
+func (m *BookingModel) Insert(booking models.Booking) (*mongo.InsertOneResult, error) {
+	return m.C.InsertOne(context.TODO(), booking)
+}
+
+// Delete will be used to delete a booking registry
+func (m *BookingModel) Delete(id string) (*mongo.DeleteResult, error) {
+	p, err := primitive.ObjectIDFromHex(id)
+	if err != nil {
+		return nil, err
+	}
+	return m.C.DeleteOne(context.TODO(), bson.M{"_id": p})
+}