Merge pull request #106 from ydb-platform/split-session-go

split session.go to {transaction,data_query,statement,keep_in_cache}.go

Author: asmyasnikov

internal/table/data_query.go

@@ -0,0 +1,47 @@
+package table
+
+import "github.com/ydb-platform/ydb-go-genproto/protos/Ydb_Table"
+
+type dataQuery struct {
+	query    Ydb_Table.Query
+	queryID  Ydb_Table.Query_Id
+	queryYQL Ydb_Table.Query_YqlText
+}
+
+func (q *dataQuery) String() string {
+	var emptyID Ydb_Table.Query_Id
+	if q.queryID == emptyID {
+		return q.queryYQL.YqlText
+	}
+	return q.queryID.Id
+}
+
+func (q *dataQuery) ID() string {
+	return q.queryID.Id
+}
+
+func (q *dataQuery) YQL() string {
+	return q.queryYQL.YqlText
+}
+
+func (q *dataQuery) initFromText(s string) {
+	q.queryID = Ydb_Table.Query_Id{} // Reset id field.
+	q.queryYQL.YqlText = s
+	q.query.Query = &q.queryYQL
+}
+
+func (q *dataQuery) initPrepared(id string) {
+	q.queryYQL = Ydb_Table.Query_YqlText{} // Reset yql field.
+	q.queryID.Id = id
+	q.query.Query = &q.queryID
+}
+
+func (q *dataQuery) initPreparedText(s, id string) {
+	q.queryYQL = Ydb_Table.Query_YqlText{} // Reset yql field.
+	q.queryYQL.YqlText = s
+
+	q.queryID = Ydb_Table.Query_Id{} // Reset id field.
+	q.queryID.Id = id
+
+	q.query.Query = &q.queryID // Prefer preared query.
+}

internal/table/keep_in_cache.go

@@ -0,0 +1,8 @@
+package table
+
+import "github.com/ydb-platform/ydb-go-genproto/protos/Ydb_Table"
+
+func keepInCache(req *Ydb_Table.ExecuteDataQueryRequest) bool {
+	p := req.QueryCachePolicy
+	return p != nil && p.KeepInCache
+}

internal/table/keep_in_cache_test.go

@@ -0,0 +1,89 @@
+package table
+
+import (
+	"context"
+	"testing"
+
+	"google.golang.org/protobuf/proto"
+
+	"github.com/ydb-platform/ydb-go-genproto/protos/Ydb_Table"
+
+	"github.com/ydb-platform/ydb-go-sdk/v3/table"
+	"github.com/ydb-platform/ydb-go-sdk/v3/table/options"
+	"github.com/ydb-platform/ydb-go-sdk/v3/testutil"
+)
+
+func TestQueryCachePolicyKeepInCache(t *testing.T) {
+	for _, test := range [...]struct {
+		name                   string
+		queryCachePolicyOption []options.QueryCachePolicyOption
+	}{
+		{
+			name: "with server cache",
+			queryCachePolicyOption: []options.QueryCachePolicyOption{
+				options.WithQueryCachePolicyKeepInCache(),
+			},
+		},
+		{
+			name:                   "no server cache",
+			queryCachePolicyOption: []options.QueryCachePolicyOption{},
+		},
+	} {
+		t.Run(test.name, func(t *testing.T) {
+			b := StubBuilder{
+				T: t,
+				cc: testutil.NewDB(
+					testutil.WithInvokeHandlers(
+						testutil.InvokeHandlers{
+							// nolint:unparam
+							testutil.TableExecuteDataQuery: func(request interface{}) (proto.Message, error) {
+								r, ok := request.(*Ydb_Table.ExecuteDataQueryRequest)
+								if !ok {
+									t.Fatalf("cannot cast request '%T' to *Ydb_Table.ExecuteDataQueryRequest", request)
+								}
+								if len(test.queryCachePolicyOption) > 0 {
+									if !r.QueryCachePolicy.GetKeepInCache() {
+										t.Fatalf("keep-in-cache policy must be true, got: %v", r.QueryCachePolicy.GetKeepInCache())
+									}
+								} else {
+									if r.QueryCachePolicy.GetKeepInCache() {
+										t.Fatalf("keep-in-cache policy must be false, got: %v", r.QueryCachePolicy.GetKeepInCache())
+									}
+								}
+								return &Ydb_Table.ExecuteQueryResult{
+									TxMeta: &Ydb_Table.TransactionMeta{
+										Id: "",
+									},
+								}, nil
+							},
+							// nolint:unparam
+							testutil.TableCreateSession: func(interface{}) (proto.Message, error) {
+								return &Ydb_Table.CreateSessionResult{
+									SessionId: testutil.SessionID(),
+								}, nil
+							},
+						},
+					),
+				),
+			}
+			s, err := b.createSession(context.Background())
+			if err != nil {
+				t.Fatal(err)
+			}
+			_, _, err = s.Execute(
+				context.Background(), table.TxControl(
+					table.BeginTx(
+						table.WithOnlineReadOnly(),
+					),
+					table.CommitTx(),
+				),
+				"SELECT 1",
+				table.NewQueryParameters(),
+				options.WithQueryCachePolicy(test.queryCachePolicyOption...),
+			)
+			if err != nil {
+				t.Fatal(err)
+			}
+		})
+	}
+}