post a path will create a document in mongodb

fzxu · fzxu · commit c2796d560092 · 2013-11-25T00:56:53.000-08:00
diff --git a/config.json b/config.json
@@ -1,5 +1,8 @@
 {
   "address": "localhost:9090",
   "read_timeout": 10,
-  "write_timeout": 10
+  "write_timeout": 10,
+  "db_host": "127.0.0.1:27017",
+  "db_name": "howtodo_test",
+  "collection": "images"
 }
diff --git a/document.go b/document.go
@@ -0,0 +1,35 @@
+package main
+
+import (
+	"log"
+
+	"labix.org/v2/mgo"
+	"labix.org/v2/mgo/bson"
+)
+
+type Document struct {
+	Id          bson.ObjectId `bson:"_id,omitempty"`
+	Name        string        `bson:"name"`
+	Path        string        `bson:"path"`
+	ContentType string        `bson:"content_type"`
+	Binary      bson.Binary   `bson:"binary"`
+}
+
+func (d *Document) Collcetion(s *mgo.Session) *mgo.Collection {
+	return s.DB(Configuration.DBName).C(Configuration.Collection)
+}
+
+func (d *Document) Save(s *mgo.Session) error {
+	coll := d.Collcetion(s)
+
+	if !bson.IsObjectIdHex(d.Id.Hex()) {
+		d.Id = bson.NewObjectId()
+	}
+
+	_, err := coll.Upsert(bson.M{"_id": d.Id}, d)
+	if err != nil {
+		log.Fatalln(err)
+		return err
+	}
+	return nil
+}
diff --git a/handler.go b/handler.go
@@ -4,6 +4,7 @@ import (
 	"io"
 	"log"
 	"net/http"
+	"strings"
 )
 
 type ImgHandler struct {
@@ -24,5 +25,31 @@ func (h *ImgHandler) handleGET(w http.ResponseWriter, req *http.Request) {
 }
 
 func (h *ImgHandler) handlePOST(w http.ResponseWriter, req *http.Request) {
+	h.createDocument(req)
 	io.WriteString(w, "Hello POST\n")
 }
+
+func (h *ImgHandler) createDocument(req *http.Request) {
+	s := MgoSession.Copy()
+	defer s.Close()
+
+	name, path := h.convertPath(req.URL.Path)
+	document := &Document{Name: name, Path: path}
+	err := document.Save(s)
+	if err != nil {
+		log.Fatalln(err)
+	}
+}
+
+func (h *ImgHandler) convertPath(urlPath string) (string, string) {
+	var path []string
+	folders := strings.Split(urlPath, "/")
+	for ind, folder := range folders {
+		trimFolder := strings.Trim(folder, " ")
+		if trimFolder != "" && ind != len(folders)-1 {
+			path = append(path, strings.ToLower(folder))
+		}
+	}
+
+	return folders[len(folders)-1], strings.Join(path, ",")
+}
diff --git a/main.go b/main.go
@@ -7,31 +7,48 @@ import (
 	"os"
 	"path/filepath"
 	"time"
+
+	"labix.org/v2/mgo"
+)
+
+var (
+	Configuration Config
+	MgoSession    *mgo.Session
 )
 
 type Config struct {
 	Address      string        `json:"address"`
 	ReadTimeout  time.Duration `json:"read_timeout"`
 	WriteTimeout time.Duration `json:"write_timeout"`
+	DBHost       string        `json:"db_host"`
+	DBName       string        `json:"db_name"`
+	Collection   string        `json:"collection"`
 }
 
 func main() {
 	// read config file
 	configFile, err := os.Open(filepath.Join(
 		os.Getenv("GOPATH"), "src", "github.com", "arkxu", "imgongo", "config.json"))
 	if err != nil {
-		log.Fatal(err)
+		log.Panicln(err)
 	}
 
-	var configuration Config
-	json.NewDecoder(configFile).Decode(&configuration)
+	json.NewDecoder(configFile).Decode(&Configuration)
+
+	// Initialize mongo connection
+	log.Println(Configuration.DBHost)
+	MgoSession, err = mgo.Dial(Configuration.DBHost)
+	if err != nil {
+		log.Panicln(err)
+	}
 
 	// start the server
 	s := &http.Server{
-		Addr:         configuration.Address,
+		Addr:         Configuration.Address,
 		Handler:      new(ImgHandler),
-		ReadTimeout:  configuration.ReadTimeout * time.Second,
-		WriteTimeout: configuration.WriteTimeout * time.Second,
+		ReadTimeout:  Configuration.ReadTimeout * time.Second,
+		WriteTimeout: Configuration.WriteTimeout * time.Second,
 	}
-	log.Fatal(s.ListenAndServe())
+	log.Panicln(s.ListenAndServe())
+
 }

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,8 @@`
`1`	`1`	`{`
`2`	`2`	`"address": "localhost:9090",`
`3`	`3`	`"read_timeout": 10,`
`4`		`- "write_timeout": 10`
	`4`	`+ "write_timeout": 10,`
	`5`	`+ "db_host": "127.0.0.1:27017",`
	`6`	`+ "db_name": "howtodo_test",`
	`7`	`+ "collection": "images"`
`5`	`8`	`}`