Download DynamoDB archive, add tests, fix bug on Windows

.gitignore

@@ -0,0 +1,40 @@
+# Created by .ignore support plugin (hsz.mobi)
+### Go template
+# Compiled Object files, Static and Dynamic libs (Shared Objects)
+*.o
+*.a
+*.so
+
+# Folders
+_obj
+_test
+
+# Architecture specific extensions/prefixes
+*.[568vq]
+[568vq].out
+
+*.cgo1.go
+*.cgo2.c
+_cgo_defun.c
+_cgo_gotypes.go
+_cgo_export.*
+
+_testmain.go
+
+*.exe
+*.test
+*.prof
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# external packages folder
+vendor/
+
+# dynamodb archive chore
+dynamodb_local_latest.tar.gz
+LICENSE.txt
+README.txt
+DynamoDBLocal_lib
+DynamoDBLocal.jar
+third_party_licenses

db.go

@@ -1,16 +1,18 @@
 package dynamodbtest
 
 import (
+	"archive/tar"
 	"bufio"
+	"compress/gzip"
 	"errors"
 	"fmt"
+	"io"
 	"log"
 	"net"
+	"net/http"
 	"os"
 	"os/exec"
-	"strings"
 	"sync/atomic"
-	"syscall"
 	"time"
 )
 
@@ -29,16 +31,97 @@ type DB struct {
 	cmd  *exec.Cmd
 }
 
+func read(mpath string) (*os.File, error) {
+	f, err := os.OpenFile(mpath, os.O_RDONLY, 0444)
+	if err != nil {
+		return f, err
+	}
+	return f, nil
+}
+
+func overwrite(mpath string) (*os.File, error) {
+	f, err := os.OpenFile(mpath, os.O_RDWR|os.O_TRUNC, 0777)
+	if err != nil {
+		f, err = os.Create(mpath)
+		if err != nil {
+			return f, err
+		}
+	}
+	return f, nil
+}
+
+func untarIt(basepath, mpath string) {
+	fr, err := read(mpath)
+	defer fr.Close()
+	if err != nil {
+		panic(err)
+	}
+	gr, err := gzip.NewReader(fr)
+	defer gr.Close()
+	if err != nil {
+		panic(err)
+	}
+	tr := tar.NewReader(gr)
+	for {
+		hdr, err := tr.Next()
+		if err == io.EOF {
+			// end of tar archive
+			break
+		}
+		if err != nil {
+			panic(err)
+		}
+		path := hdr.Name
+		switch hdr.Typeflag {
+		case tar.TypeDir:
+			if err := os.MkdirAll(basepath+path, os.FileMode(hdr.Mode)); err != nil {
+				panic(err)
+			}
+		case tar.TypeReg:
+			ow, err := overwrite(basepath + path)
+			defer ow.Close()
+			if err != nil {
+				panic(err)
+			}
+			if _, err := io.Copy(ow, tr); err != nil {
+				panic(err)
+			}
+		default:
+			fmt.Printf("Can't: %c, %s\n", hdr.Typeflag, path)
+		}
+	}
+}
+
 // New returns a started DynamoDB local instance
 func New() (*DB, error) {
 	port := newPort()
 	addr := fmt.Sprintf("localhost:%d", port)
 	// if $GOPATH is composed of multiple paths, use the first one (fix for godep)
-	gopath := strings.Split(os.Getenv("GOPATH"), ":")[0]
+	gopath := os.Getenv("GOPATH")
 	if gopath == "" {
 		return nil, ErrGopath
 	}
-	path := gopath + "/src/github.com/groupme/dynamodbtest/"
+	path := gopath + "/src/github.com/miltador/dynamodbtest/"
+	archivePath := path + "dynamodb_local_latest.tar.gz"
+	if _, err := os.Stat(archivePath); os.IsNotExist(err) {
+		response, err := http.Get("https://s3-us-west-2.amazonaws.com/dynamodb-local/dynamodb_local_latest.tar.gz")
+		if err != nil {
+			panic(err)
+		}
+		defer response.Body.Close()
+
+		f, err := os.Create(archivePath)
+		if err != nil {
+			panic(err)
+		}
+		defer f.Close()
+
+		io.Copy(f, response.Body)
+	}
+	if _, err := os.Stat(path + "DynamoDbLocal_lib/"); os.IsNotExist(err) {
+		untarIt(path, archivePath)
+	}
+
 	db := &DB{
 		addr: addr,
 		cmd: exec.Command(
@@ -78,7 +161,7 @@ func New() (*DB, error) {
 	go func() {
 		// periodically check if connectable
 		ticker := time.NewTicker(time.Millisecond * 10)
-		for _ = range ticker.C {
+		for range ticker.C {
 			c, err := net.Dial("tcp", db.addr)
 			if c != nil {
 				c.Close()
@@ -99,8 +182,7 @@ func New() (*DB, error) {
 }
 
 func (db *DB) Close() error {
-	db.cmd.Process.Signal(syscall.SIGINT)
-	return db.cmd.Wait()
+	return db.cmd.Process.Kill()
 }
 
 func (db *DB) URL() string {

db_test.go

@@ -0,0 +1,26 @@
+package dynamodbtest
+
+import (
+	"testing"
+)
+
+func TestAll(t *testing.T) {
+	// Log output to aid debugging
+	LogOutput = true
+
+	// Start a new test process
+	db, err := New()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	url := db.URL()
+	if url != "http://localhost:8000" {
+		t.Error("URL is not correct")
+	}
+
+	_err := db.Close()
+	if _err != nil {
+		t.Fatal(_err)
+	}
+}