Forráskód Böngészése

Export all fields of the QueryInfo struct

Ben Hood 11 éve
szülő
commit
aa38232411
2 módosított fájl, 21 hozzáadás és 21 törlés
  1. 8 8
      cassandra_test.go
  2. 13 13
      conn.go

+ 8 - 8
cassandra_test.go

@@ -601,9 +601,9 @@ type ClusteredKeyValue struct {
 }
 
 func (kv *ClusteredKeyValue) Bind(q *QueryInfo) ([]interface{}, error) {
-	values := make([]interface{}, len(q.args))
+	values := make([]interface{}, len(q.Args))
 
-	for i, info := range q.args {
+	for i, info := range q.Args {
 		fieldName := upcaseInitial(info.Name)
 		value := reflect.ValueOf(kv)
 		field := reflect.Indirect(value).FieldByName(fieldName)
@@ -728,8 +728,8 @@ func injectInvalidPreparedStatement(t *testing.T, session *Session, table string
 	stmtsLRU.lru.Add(conn.addr+stmt, flight)
 	stmtsLRU.mu.Unlock()
 	flight.info = &QueryInfo{
-		id: []byte{'f', 'o', 'o', 'b', 'a', 'r'},
-		args: []ColumnInfo{ColumnInfo{
+		Id: []byte{'f', 'o', 'o', 'b', 'a', 'r'},
+		Args: []ColumnInfo{ColumnInfo{
 			Keyspace: "gocql_test",
 			Table:    table,
 			Name:     "foo",
@@ -777,13 +777,13 @@ func TestQueryInfo(t *testing.T) {
 		t.Fatalf("Failed to execute query for preparing statement: %v", err)
 	}
 
-	if len(info.args) != 1 {
-		t.Fatalf("Was not expecting meta data for %d query arguments, but got %d\n", 1, len(info.args))
+	if len(info.Args) != 1 {
+		t.Fatalf("Was not expecting meta data for %d query arguments, but got %d\n", 1, len(info.Args))
 	}
 
 	if *flagProto > 1 {
-		if len(info.rval) != 2 {
-			t.Fatalf("Was not expecting meta data for %d result columns, but got %d\n", 2, len(info.rval))
+		if len(info.Rval) != 2 {
+			t.Fatalf("Was not expecting meta data for %d result columns, but got %d\n", 2, len(info.Rval))
 		}
 	}
 }

+ 13 - 13
conn.go

@@ -329,9 +329,9 @@ func (c *Conn) prepareStatement(stmt string, trace Tracer) (*QueryInfo, error) {
 		switch x := resp.(type) {
 		case resultPreparedFrame:
 			flight.info = &QueryInfo{
-				id:   x.PreparedId,
-				args: x.Arguments,
-				rval: x.ReturnValues,
+				Id:   x.PreparedId,
+				Args: x.Arguments,
+				Rval: x.ReturnValues,
 			}
 		case error:
 			flight.err = x
@@ -376,13 +376,13 @@ func (c *Conn) executeQuery(qry *Query) *Iter {
 			}
 		}
 
-		if len(values) != len(info.args) {
+		if len(values) != len(info.Args) {
 			return &Iter{err: ErrQueryArgLength}
 		}
-		op.Prepared = info.id
+		op.Prepared = info.Id
 		op.Values = make([][]byte, len(values))
 		for i := 0; i < len(values); i++ {
-			val, err := Marshal(info.args[i].TypeInfo, values[i])
+			val, err := Marshal(info.Args[i].TypeInfo, values[i])
 			if err != nil {
 				return &Iter{err: err}
 			}
@@ -500,23 +500,23 @@ func (c *Conn) executeBatch(batch *Batch) error {
 				}
 			}
 
-			if len(args) != len(info.args) {
+			if len(args) != len(info.Args) {
 				return ErrQueryArgLength
 			}
 
-			stmts[string(info.id)] = entry.Stmt
+			stmts[string(info.Id)] = entry.Stmt
 			if err != nil {
 				return err
 			}
 			f.writeByte(1)
-			f.writeShortBytes(info.id)
+			f.writeShortBytes(info.Id)
 		} else {
 			f.writeByte(0)
 			f.writeLongString(entry.Stmt)
 		}
 		f.writeShort(uint16(len(args)))
 		for j := 0; j < len(args); j++ {
-			val, err := Marshal(info.args[j].TypeInfo, args[j])
+			val, err := Marshal(info.Args[j].TypeInfo, args[j])
 			if err != nil {
 				return err
 			}
@@ -648,9 +648,9 @@ func (c *Conn) setKeepalive(d time.Duration) error {
 
 // QueryInfo represents the meta data associated with a prepared CQL statement.
 type QueryInfo struct {
-	id   []byte
-	args []ColumnInfo
-	rval []ColumnInfo
+	Id   []byte
+	Args []ColumnInfo
+	Rval []ColumnInfo
 }
 
 type callReq struct {