piclez · May 10, 2016 03:29
diff --git a/gomongohql.go b/gomongohql.go
 package main

 import (
 	"fmt"
 	"html/template"
        "labix.org/v2/mgo"
        "labix.org/v2/mgo/bson"
        "log"
 	"net/http"
 	"os"
 )

 type Person struct {
        Name string
        Email string
 }

 func main() {
 	http.HandleFunc("/", root)
        http.HandleFunc("/display", display)
        fmt.Println("listening...")
        err := http.ListenAndServe(GetPort(), nil)
        if err != nil {
                log.Fatal("ListenAndServe: ", err)
                return
        }
 }

 func root(w http.ResponseWriter, r *http.Request) {
 	fmt.Fprint(w, rootForm)
 }

 const rootForm = `
  <!DOCTYPE html>
    <html>
      <head>
        <meta charset="utf-8">
        <title>Your details</title>
        <link rel="stylesheet" href="http://yui.yahooapis.com/pure/0.4.2/pure-min.css">
      </head>
      <body style="margin: 20px;">
        <h2>A Fun Go App on Heroku to access MongoDB on MongoLab</h2>
        <p>This simple app will fetch the email id of a person, if it's already there in the MongoDB database.</p>
        <p>Please enter a name (example: Stefan Klaste)</p>
        <form action="/display" method="post" accept-charset="utf-8" class="pure-form">
          <input type="text" name="name" placeholder="name" />
          <input type="submit" value=".. and query database!" class="pure-button pure-button-primary"/>
 	</form>
        <div>
          <p><b>&copy; 2015 Go Challenge. All rights reserved.</b></p>
        </div>	
      </body>
    </html>
 `

 var displayTemplate = template.Must(template.New("display").Parse(displayTemplateHTML))

 func display(w http.ResponseWriter, r *http.Request) {
        // In the open command window set the following for Heroku.
        // Remember to use your login/password in the string below
        // heroku config:set MONGOLAB_URL=mongodb://IndianGuru:[email protected]:51523/godata
        uri := os.Getenv("MONGOLAB_URL")
        if uri == "" {
                fmt.Println("no connection string provided")
                os.Exit(1)
        }
 
        sess, err := mgo.Dial(uri)
        if err != nil {
                fmt.Printf("Can't connect to mongo, go error %v\n", err)
                os.Exit(1)
        }
        defer sess.Close()
        
        sess.SetSafe(&mgo.Safe{})
        
        collection := sess.DB("godata").C("user")

        result := Person{}

        collection.Find(bson.M{"name": r.FormValue("name")}).One(&result)

        if result.Email != "" {
                errn := displayTemplate.Execute(w, "The email id you wanted is: " + result.Email)
                if errn != nil {
                        http.Error(w, errn.Error(), http.StatusInternalServerError)
                } 
        } else {
                displayTemplate.Execute(w, "Sorry... The email id you wanted does not exist.")
        }
 }

 const displayTemplateHTML = ` 
 <!DOCTYPE html>
  <html>
    <head>
      <meta charset="utf-8">
      <title>Results</title>
      <link rel="stylesheet" href="http://yui.yahooapis.com/pure/0.4.2/pure-min.css">
    </head>
    <body>
      <h2>A Fun Go App on Heroku to access MongoDB on MongoLab</h2>
      <p><b>{{html .}}</b></p>
      <p><a href="/">Start again!</a></p>
      <div>
        <p><b>&copy; 2015 Go Challenge. All rights reserved.</b></p>
      </div>
    </body>
  </html>
 `

 // Get the Port from the environment so we can run on Heroku
 func GetPort() string {
        var port = os.Getenv("PORT")
 	// Set a default port if there is nothing in the environment
 	if port == "" {
 		port = "4747"
 		fmt.Println("INFO: No PORT environment variable detected, defaulting to " + port)
 	}
 	return ":" + port
 }
	package main

	import (
	"fmt"
	"html/template"
	"labix.org/v2/mgo"
	"labix.org/v2/mgo/bson"
	"log"
	"net/http"
	"os"
	)

	type Person struct {
	Name string
	Email string
	}

	func main() {
	http.HandleFunc("/", root)
	http.HandleFunc("/display", display)
	fmt.Println("listening...")
	err := http.ListenAndServe(GetPort(), nil)
	if err != nil {
	log.Fatal("ListenAndServe: ", err)
	return
	}
	}

	func root(w http.ResponseWriter, r *http.Request) {
	fmt.Fprint(w, rootForm)
	}

	const rootForm = `
	<!DOCTYPE html>
	<html>
	<head>
	<meta charset="utf-8">
	<title>Your details</title>
	<link rel="stylesheet" href="http://yui.yahooapis.com/pure/0.4.2/pure-min.css">
	</head>
	<body style="margin: 20px;">
	<h2>A Fun Go App on Heroku to access MongoDB on MongoLab</h2>
	<p>This simple app will fetch the email id of a person, if it's already there in the MongoDB database.</p>
	<p>Please enter a name (example: Stefan Klaste)</p>
	<form action="/display" method="post" accept-charset="utf-8" class="pure-form">
	<input type="text" name="name" placeholder="name" />
	<input type="submit" value=".. and query database!" class="pure-button pure-button-primary"/>
	</form>
	<div>
	<p><b>© 2015 Go Challenge. All rights reserved.</b></p>
	</div>
	</body>
	</html>
	`

	var displayTemplate = template.Must(template.New("display").Parse(displayTemplateHTML))

	func display(w http.ResponseWriter, r *http.Request) {
	// In the open command window set the following for Heroku.
	// Remember to use your login/password in the string below
	// heroku config:set MONGOLAB_URL=mongodb://IndianGuru:[email protected]:51523/godata
	uri := os.Getenv("MONGOLAB_URL")
	if uri == "" {
	fmt.Println("no connection string provided")
	os.Exit(1)
	}

	sess, err := mgo.Dial(uri)
	if err != nil {
	fmt.Printf("Can't connect to mongo, go error %v\n", err)
	os.Exit(1)
	}
	defer sess.Close()

	sess.SetSafe(&mgo.Safe{})

	collection := sess.DB("godata").C("user")

	result := Person{}

	collection.Find(bson.M{"name": r.FormValue("name")}).One(&result)

	if result.Email != "" {
	errn := displayTemplate.Execute(w, "The email id you wanted is: " + result.Email)
	if errn != nil {
	http.Error(w, errn.Error(), http.StatusInternalServerError)
	}
	} else {
	displayTemplate.Execute(w, "Sorry... The email id you wanted does not exist.")
	}
	}

	const displayTemplateHTML = `
	<!DOCTYPE html>
	<html>
	<head>
	<meta charset="utf-8">
	<title>Results</title>
	<link rel="stylesheet" href="http://yui.yahooapis.com/pure/0.4.2/pure-min.css">
	</head>
	<body>
	<h2>A Fun Go App on Heroku to access MongoDB on MongoLab</h2>
	<p><b>{{html .}}</b></p>
	<p><a href="/">Start again!</a></p>
	<div>
	<p><b>© 2015 Go Challenge. All rights reserved.</b></p>
	</div>
	</body>
	</html>
	`

	// Get the Port from the environment so we can run on Heroku
	func GetPort() string {
	var port = os.Getenv("PORT")
	// Set a default port if there is nothing in the environment
	if port == "" {
	port = "4747"
	fmt.Println("INFO: No PORT environment variable detected, defaulting to " + port)
	}
	return ":" + port
	}