some util methods

Author: erh

bsonutil.go

@@ -18,6 +18,14 @@ func SimpleBSONConvert(d interface{}) (SimpleBSON, error) {
 	return SimpleBSON{int32(len(raw)), raw}, nil
 }
 
+func SimpleBSONConvertOrPanic(d interface{}) SimpleBSON {
+	raw, err := bson.Marshal(d)
+	if err != nil {
+		panic(err)
+	}
+	return SimpleBSON{int32(len(raw)), raw}
+}
+
 func (sb SimpleBSON) ToBSOND() (bson.D, error) {
 	t := bson.D{}
 	err := bson.Unmarshal(sb.BSON, &t)

wire_insert.go

@@ -61,3 +61,18 @@ func parseInsertMessage(header MessageHeader, buf []byte) (Message, error) {
 
 	return m, nil
 }
+
+func NewInsertMessage(namespace string, docs ...SimpleBSON) *InsertMessage {
+	im := &InsertMessage{}
+
+	im.header.RequestID = 17 // TODO
+	im.header.ResponseTo = 0
+	im.header.OpCode = OP_INSERT
+
+	im.Flags = 0
+	im.Namespace = namespace
+
+	im.Docs = docs
+
+	return im
+}

wire_query.go

@@ -73,3 +73,16 @@ func parseQueryMessage(header MessageHeader, buf []byte) (Message, error) {
 
 	return qm, nil
 }
+
+func NewQueryMessage(ns string, flags int32, skip int32, toReturn int32, query SimpleBSON, project SimpleBSON) *QueryMessage {
+	qm := &QueryMessage{}
+	qm.header.RequestID = 17 // TODO
+	qm.header.OpCode = OP_QUERY
+	qm.Flags = flags
+	qm.Namespace = ns
+	qm.Skip = skip
+	qm.NReturn = toReturn
+	qm.Query = query
+	qm.Project = project
+	return qm
+}