Dirty migration tool for gridfs

tools/togridfs/togridfs.go

@@ -0,0 +1,61 @@
+package main
+
+import (
+	"fmt"
+	"git.gitorious.org/go-pkg/epubgo.git"
+	"io"
+	"labix.org/v2/mgo/bson"
+	"os"
+)
+
+func main() {
+	db = initDB()
+	defer db.Close()
+	books, _, _ := db.GetBooks(bson.M{})
+	fs := db.GetFS(FS_BOOKS)
+
+	for _, book := range books {
+		if book.Path == "" {
+			fmt.Println("don't needed -- ", book.Title)
+			continue
+		}
+		fmt.Println(book.Title)
+
+		path := "books/" + book.Path
+		file, err := os.Open(path)
+		if err != nil {
+			fmt.Println("os.Open ================", err)
+			continue
+		}
+		defer file.Close()
+
+		fw, err := fs.Create(book.Title + ".epub")
+		if err != nil {
+			fmt.Println("gridfs.Create ================", err)
+			continue
+		}
+		defer fw.Close()
+
+		_, err = io.Copy(fw, file)
+		if err != nil {
+			fmt.Println("io.Copy ================", err)
+			continue
+		}
+		id, _ := fw.Id().(bson.ObjectId)
+
+		e, err := epubgo.Open(path)
+		if err != nil {
+			fmt.Println("epubgo.Open ================", err)
+			continue
+		}
+		defer e.Close()
+
+		cover, coverSmall := GetCover(e, book.Title)
+		if cover != "" {
+			db.UpdateBook(bson.ObjectIdHex(book.Id), bson.M{"cover": cover, "coversmall": coverSmall, "file": id}, bson.M{"path": 1})
+		} else {
+			fmt.Println("No cover ================", book.Title)
+			db.UpdateBook(bson.ObjectIdHex(book.Id), bson.M{"file": id}, bson.M{"path": 1})
+		}
+	}
+}