add movies code

Author: mikebellcoder

movies/Dockerfile

@@ -11,7 +11,7 @@ RUN CGO_ENABLED=0 GOOS=linux go build -ldflags="-w -s" -a -installsuffix cgo -o
 
 # build a small image
 FROM alpine:3.17.3
-LABEL language="golang"
+LABEL language "golang"
 LABEL org.opencontainers.image.source https://github.com/mikebellcoder/microservices-docker-go-mongodb
 COPY --from=builder /etc/passwd /etc/passwd
 COPY --from=builder --chown=elf:1000 /go/bin/cinema-movies /cinema

movies/cmd/app/handlers.go

@@ -0,0 +1,92 @@
+package main
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"github.com/gorilla/mux"
+	"github.com/mikebellcoder/microservices-docker-go-mongodb/movies/pkg/models"
+)
+
+func (app *application) all(w http.ResponseWriter, r *http.Request) {
+	// Get all movies stored
+	movies, err := app.movies.All()
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	// Convert movie list into json encoding
+	b, err := json.Marshal(movies)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	app.infoLog.Println("Movies 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 movie by id
+	m, err := app.movies.FindByID(id)
+	if err != nil {
+		if err.Error() == "ErrNoDocuments" { // TODO make this string a constant
+			app.infoLog.Println("Movie not found")
+			return
+		}
+		// Any other error will send an internal server error
+		app.serverError(w, err)
+	}
+
+	// Convert movie to json encoding
+	b, err := json.Marshal(m)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	app.infoLog.Println("Have found a movie")
+
+	// 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 movie model
+	var m models.Movie
+	// Get request information
+	err := json.NewDecoder(r.Body).Decode(&m)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	// Insert new movie
+	insertResult, err := app.movies.Insert(m)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	app.infoLog.Printf("New movie 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 movie by id
+	deleteResult, err := app.movies.Delete(id)
+	if err != nil {
+		app.serverError(w, err)
+	}
+
+	app.infoLog.Printf("%d movie(s) have been deleted\n", deleteResult.DeletedCount)
+}

movies/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.erroLog.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)
+}

movies/cmd/app/main.go

@@ -0,0 +1,85 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"time"
+
+	"github.com/mikebellcoder/microservices-docker-go-mongodb/movies/pkg/models/mongodb"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+type application struct {
+	erroLog *log.Logger
+	infoLog *log.Logger
+	movies  *mongodb.MovieModel
+}
+
+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", "movies", "Database name")
+	enableCredentials := flag.Bool("enableCredentials", false, "Enable the use of for mongodb connection")
+	flag.Parse()
+
+	// Create logger for writing information and error messages
+	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 mongo client config
+	co := options.Client().ApplyURI(*mongoURI)
+	if *enableCredentials {
+		co.Auth = &options.Credential{
+			Username: os.Getenv("MONGODB_USERNAME"),
+			Password: os.Getenv("MONGODB_PASSWORD"),
+		}
+	}
+
+	// Establish database 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)
+		}
+	}()
+
+	infoLog.Printf("Database connection established\n")
+
+	// Initialize a new instance of a app containing the dependencies
+	app := &application{
+		infoLog: infoLog,
+		erroLog: errLog,
+		movies: &mongodb.MovieModel{
+			C: client.Database(*mongoDatabase).Collection("movies"),
+		},
+	}
+
+	// Initialize a new http.Server struct.
+	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)
+}

movies/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/movies/", app.all).Methods("GET")
+	r.HandleFunc("/api/movies/{id}", app.findByID).Methods("GET")
+	r.HandleFunc("/api/movies/", app.insert).Methods("POST")
+	r.HandleFunc("/api/movies/{id}", app.delete).Methods("DELETE")
+
+	return r
+}

movies/go.mod

@@ -4,6 +4,7 @@ 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

movies/go.sum

@@ -1,5 +1,7 @@
 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=