add hanlder to update tree name

common/json.go

@@ -6,11 +6,6 @@ type UserJson struct {
 	Trees    []string `bson:"trees" omitempty`
 }
 
-type MissionJson struct {
-	Mission string `json:"mission"`
-	TreeId  string `json:"treeId,omitempty"`
-}
-
 type OrganizationJson struct {
 	Organization string `json:"organization"`
 	UserId       string `json:"userId" bson:"_id,omitempty"`

handlers/mission_update.go

@@ -1,15 +1,19 @@
 package handlers
 
 import (
-	"github.com/dmonay/okra/common"
 	"github.com/gin-gonic/gin"
 	"gopkg.in/mgo.v2/bson"
 )
 
 func (dw *DoWorkResource) UpdateMission(c *gin.Context) {
 
+	type MissionJson struct {
+		Mission string `json:"mission"`
+		TreeId  string `json:"treeId,omitempty"`
+	}
+
 	org := c.Params.ByName("organization")
-	var reqBody common.MissionJson
+	var reqBody MissionJson
 
 	c.Bind(&reqBody)
 

handlers/start.go

@@ -58,6 +58,7 @@ func Run(cfg common.Config) error {
 
 	// trees
 	r.POST("/create/tree/:organization", doWorkResource.CreateTree)
+	r.POST("/update/tree/name/:organization/:treeid", doWorkResource.UpdateTreeName)
 	r.GET("/get/trees/:organization/:treeid", doWorkResource.GetTree)
 	r.GET("/get/trees/:organization", doWorkResource.GetAllTrees)
 
@@ -84,7 +85,6 @@ func InitMongo() (*mgo.Database, error) {
 	CheckErr(err, "mongo connection failed")
 
 	db := session.DB("testing")
-	CheckErr(err, "mongo opening database failed")
 
 	return db, nil
 }

handlers/tree_update_name.go

@@ -0,0 +1,27 @@
+package handlers
+
+import (
+	"github.com/gin-gonic/gin"
+	"gopkg.in/mgo.v2/bson"
+)
+
+func (dw *DoWorkResource) UpdateTreeName(c *gin.Context) {
+	org := c.Params.ByName("organization")
+	treeId := c.Params.ByName("treeid")
+	id := bson.ObjectIdHex(treeId)
+
+	type TreeNameJson struct {
+		TreeName string `json:"treename"`
+	}
+
+	var reqBody TreeNameJson
+	c.Bind(&reqBody)
+	newName := reqBody.TreeName
+
+	querier := bson.M{"_id": id}
+	updateName := bson.M{"$set": bson.M{"treename": newName}}
+	err := dw.mongo.C(org).Update(querier, updateName)
+	CheckErr(err, "Mongo failed to update tree name")
+
+	c.JSON(201, "Success")
+}