Entre em alta: usando Docker + Go com MongoDB

1
go mod init

1
module github.com/mongodb-developer/docker-golang-example
2
go 1.15
3
require go.mongodb.org/mongo-driver v1.7.0
4
require github.com/gorilla/mux v1.8.0

1
package main
2
3
import (
4
	"context"
5
	"encoding/json"
6
	"fmt"
7
	"net/http"
8
	"os"
9
	"time"
10
11
	"github.com/gorilla/mux"
12
	"go.mongodb.org/mongo-driver/bson"
13
	"go.mongodb.org/mongo-driver/mongo"
14
	"go.mongodb.org/mongo-driver/mongo/options"
15
)
16
17
var client *mongo.Client
18
var collection *mongo.Collection
19
20
type Tweet struct {
21
	ID       int64  `json:"_id,omitempty" bson:"_id,omitempty"`
22
	FullText string `json:"full_text,omitempty" bson:"full_text,omitempty"`
23
	User     struct {
24
		ScreenName string `json:"screen_name" bson:"screen_name"`
25
	} `json:"user,omitempty" bson:"user,omitempty"`
26
}
27
28
func GetTweetsEndpoint(response http.ResponseWriter, request *http.Request) {}
29
func SearchTweetsEndpoint(response http.ResponseWriter, request *http.Request) {}
30
31
func main() {
32
	fmt.Println("Starting the application...")
33
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
34
	defer cancel()
35
	client, err := mongo.Connect(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URI")))
36
	defer func() {
37
		if err = client.Disconnect(ctx); err != nil {
38
			panic(err)
39
		}
40
	}()
41
collection = client.Database("synonyms").Collection("tweets")
42
	router := mux.NewRouter()
43
	router.HandleFunc("/tweets", GetTweetsEndpoint).Methods("GET")
44
	router.HandleFunc("/search", SearchTweetsEndpoint).Methods("GET")
45
	http.ListenAndServe(":12345", router)
46
}

1
type Tweet struct {
2
	ID       int64  `json:"_id,omitempty" bson:"_id,omitempty"`
3
	FullText string `json:"full_text,omitempty" bson:"full_text,omitempty"`
4
	User     struct {
5
		ScreenName string `json:"screen_name" bson:"screen_name"`
6
	} `json:"user,omitempty" bson:"user,omitempty"`
7
}

1
func GetTweetsEndpoint(response http.ResponseWriter, request *http.Request) {}
2
func SearchTweetsEndpoint(response http.ResponseWriter, request *http.Request) {}

1
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
2
defer cancel()
3
client, err := mongo.Connect(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URI")))
4
defer func() {
5
	if err = client.Disconnect(ctx); err != nil {
6
		panic(err)
7
	}
8
}()
9
collection = client.Database("synonyms").Collection("tweets")

1
router := mux.NewRouter()
2
router.HandleFunc("/tweets", GetTweetsEndpoint).Methods("GET")
3
router.HandleFunc("/search", SearchTweetsEndpoint).Methods("GET")
4
http.ListenAndServe(":12345", router)

1
func GetTweetsEndpoint(response http.ResponseWriter, request *http.Request) {
2
	response.Header().Set("content-type", "application/json")
3
	var tweets []Tweet
4
	ctx, _ := context.WithTimeout(context.Background(), 30*time.Second)
5
	cursor, err := collection.Find(ctx, bson.M{})
6
	if err != nil {
7
		response.WriteHeader(http.StatusInternalServerError)
8
		response.Write([]byte(`{ "message": "` + err.Error() + `" }`))
9
		return
10
	}
11
	if err = cursor.All(ctx, &tweets); err != nil {
12
		response.WriteHeader(http.StatusInternalServerError)
13
		response.Write([]byte(`{ "message": "` + err.Error() + `" }`))
14
		return
15
	}
16
	json.NewEncoder(response).Encode(tweets)
17
}

1
func SearchTweetsEndpoint(response http.ResponseWriter, request *http.Request) {
2
	response.Header().Set("content-type", "application/json")
3
	queryParams := request.URL.Query()
4
	var tweets []Tweet
5
	ctx, _ := context.WithTimeout(context.Background(), 30*time.Second)
6
	searchStage := bson.D{
7
		{"$search", bson.D{
8
			{"index", "synsearch"},
9
			{"text", bson.D{
10
				{"query", queryParams.Get("q")},
11
				{"path", "full_text"},
12
				{"synonyms", "slang"},
13
			}},
14
		}},
15
	}
16
	cursor, err := collection.Aggregate(ctx, mongo.Pipeline{searchStage})
17
	if err != nil {
18
		response.WriteHeader(http.StatusInternalServerError)
19
		response.Write([]byte(`{ "message": "` + err.Error() + `" }`))
20
		return
21
	}
22
	if err = cursor.All(ctx, &tweets); err != nil {
23
		response.WriteHeader(http.StatusInternalServerError)
24
		response.Write([]byte(`{ "message": "` + err.Error() + `" }`))
25
		return
26
	}
27
	json.NewEncoder(response).Encode(tweets)
28
}

1
​​queryParams := request.URL.Query()

1
searchStage := bson.D{
2
	{"$search", bson.D{
3
		{"index", "synsearch"},
4
		{"text", bson.D{
5
			{"query", queryParams.Get("q")},
6
			{"path", "full_text"},
7
			{"synonyms", "slang"},
8
		}},
9
	}},
10
}

1
#get a base image
2
FROM golang:1.16-buster
3
4
MAINTAINER anaiya raisinghani <anaiya.raisinghani@mongodb.com>
5
6
WORKDIR /go/src/app
7
COPY ./src .
8
9
RUN go get -d -v
10
RUN go build -v
11
12
CMD ["./docker-golang-example"]

1
docker build -t docker-syn-image .
2
docker run -d -p 12345:12345 -e “MONGODB_URI=YOUR_URI_HERE” docker-syn-image

1
version: "3.9" 
2
services:
3
   web:
4
       build: .
5
       ports:
6
           - "12345:12345"
7
       environment:
8
           MONGODB_URI: your_URI_here