add batch write item

Author: tzzh

README.md

@@ -19,6 +19,11 @@ Compile the pod by running `go build`, then:
                     {"AmazingTable" {:Keys [{:some-property {:S "SomeValue"} 
                                              :something-else {:S "SomethingSomething"}}]}}})
 
+(d/batch-write-item {:RequestItems
+                    {"AmazingTable" [{:PutRequest {:Item {:some-property {:S "abxdggje"}
+                                                          :something-else {:S "zxcmbnj"}
+                                                          :another-thing {:S "asdasdsa"}}}}]}})
+
 (d/get-item {:Key {:lalala {:S "zzzzzzzz"}
                    :bbbbbb {:S "abxbxbxx"}}
              :TableName "SomeTable"})

dynamo.go

@@ -23,6 +23,23 @@ func BatchGetItem(inputMessage *Message) (*dynamodb.BatchGetItemOutput, error) {
 	return res, err
 }
 
+func BatchWriteItem(inputMessage *Message) (*dynamodb.BatchWriteItemOutput, error) {
+
+	svc := dynamodb.New(session.New())
+	input := &dynamodb.BatchWriteItemInput{}
+	inputList := []dynamodb.BatchWriteItemInput{}
+	err := json.Unmarshal([]byte(inputMessage.Args), &inputList)
+	if err != nil {
+		return nil, err
+	}
+	if len(inputList) > 0 {
+		input = &inputList[0]
+	}
+
+	res, err := svc.BatchWriteItem(input)
+	return res, err
+}
+
 func DescribeTable(inputMessage *Message) (*dynamodb.DescribeTableOutput, error) {
 
 	svc := dynamodb.New(session.New())

main.go

@@ -22,6 +22,7 @@ func main() {
 					{Name: "pod.dynamodb",
 						Vars: []Var{
 							{Name: "batch-get-item"},
+							{Name: "batch-write-item"},
 							{Name: "describe-table"},
 							{Name: "get-item"},
 							{Name: "list-tables"},
@@ -41,6 +42,13 @@ func main() {
 				} else {
 					WriteInvokeResponse(message, res)
 				}
+			case "pod.dynamodb/batch-write-item":
+				res, err := BatchWriteItem(message)
+				if err != nil {
+					WriteErrorResponse(message, err)
+				} else {
+					WriteInvokeResponse(message, res)
+				}
 			case "pod.dynamodb/describe-table":
 				res, err := DescribeTable(message)
 				if err != nil {