forked from fl00r/go-tarantool-1.6
-
Notifications
You must be signed in to change notification settings - Fork 60
box: first box and box.info implementation #414
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Merged
Merged
Changes from all commits
Commits
File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,36 @@ | ||
package box | ||
|
||
import ( | ||
"github.com/tarantool/go-tarantool/v2" | ||
) | ||
|
||
// Box is a helper that wraps box.* requests. | ||
// It holds a connection to the Tarantool instance via the Doer interface. | ||
type Box struct { | ||
conn tarantool.Doer // Connection interface for interacting with Tarantool. | ||
} | ||
|
||
// New returns a new instance of the box structure, which implements the Box interface. | ||
func New(conn tarantool.Doer) *Box { | ||
return &Box{ | ||
conn: conn, // Assigns the provided Tarantool connection. | ||
} | ||
} | ||
|
||
// Info retrieves the current information of the Tarantool instance. | ||
// It calls the "box.info" function and parses the result into the Info structure. | ||
func (b *Box) Info() (Info, error) { | ||
var infoResp InfoResponse | ||
|
||
// Call "box.info" to get instance information from Tarantool. | ||
fut := b.conn.Do(NewInfoRequest()) | ||
|
||
// Parse the result into the Info structure. | ||
err := fut.GetTyped(&infoResp) | ||
if err != nil { | ||
return Info{}, err | ||
} | ||
|
||
// Return the parsed info and any potential error. | ||
return infoResp.Info, err | ||
} |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,29 @@ | ||
package box_test | ||
|
||
import ( | ||
"testing" | ||
|
||
"github.com/stretchr/testify/require" | ||
"github.com/tarantool/go-tarantool/v2/box" | ||
) | ||
|
||
func TestNew(t *testing.T) { | ||
// Create a box instance with a nil connection. This should lead to a panic later. | ||
b := box.New(nil) | ||
|
||
// Ensure the box instance is not nil (which it shouldn't be), but this is not meaningful | ||
// since we will panic when we call the Info method with the nil connection. | ||
require.NotNil(t, b) | ||
|
||
// We expect a panic because we are passing a nil connection (nil Doer) to the By function. | ||
// The library does not control this zone, and the nil connection would cause a runtime error | ||
// when we attempt to call methods (like Info) on it. | ||
// This test ensures that such an invalid state is correctly handled by causing a panic, | ||
// as it's outside the library's responsibility. | ||
require.Panics(t, func() { | ||
|
||
// Calling Info on a box with a nil connection will result in a panic, since the underlying | ||
// connection (Doer) cannot perform the requested action (it's nil). | ||
_, _ = b.Info() | ||
}) | ||
} |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,60 @@ | ||
// Run Tarantool Common Edition before example execution: | ||
// | ||
// Terminal 1: | ||
// $ cd box | ||
// $ TEST_TNT_LISTEN=127.0.0.1:3013 tarantool testdata/config.lua | ||
// | ||
// Terminal 2: | ||
// $ go test -v example_test.go | ||
package box_test | ||
|
||
import ( | ||
"context" | ||
"fmt" | ||
"log" | ||
"time" | ||
|
||
"github.com/tarantool/go-tarantool/v2" | ||
"github.com/tarantool/go-tarantool/v2/box" | ||
) | ||
|
||
func Example() { | ||
dialer := tarantool.NetDialer{ | ||
Address: "127.0.0.1:3013", | ||
User: "test", | ||
Password: "test", | ||
} | ||
ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond) | ||
client, err := tarantool.Connect(ctx, dialer, tarantool.Opts{}) | ||
cancel() | ||
if err != nil { | ||
log.Fatalf("Failed to connect: %s", err) | ||
} | ||
|
||
// You can use Info Request type. | ||
|
||
fut := client.Do(box.NewInfoRequest()) | ||
|
||
resp := &box.InfoResponse{} | ||
|
||
err = fut.GetTyped(resp) | ||
if err != nil { | ||
log.Fatalf("Failed get box info: %s", err) | ||
} | ||
|
||
// Or use simple Box implementation. | ||
|
||
b := box.New(client) | ||
|
||
info, err := b.Info() | ||
if err != nil { | ||
log.Fatalf("Failed get box info: %s", err) | ||
} | ||
|
||
if info.UUID != resp.Info.UUID { | ||
log.Fatalf("Box info uuids are not equal") | ||
} | ||
|
||
fmt.Printf("Box info uuids are equal") | ||
fmt.Printf("Current box info: %+v\n", resp.Info) | ||
} |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,76 @@ | ||
package box | ||
|
||
import ( | ||
"fmt" | ||
|
||
"github.com/tarantool/go-tarantool/v2" | ||
"github.com/vmihailenco/msgpack/v5" | ||
) | ||
|
||
var _ tarantool.Request = (*InfoRequest)(nil) | ||
|
||
// Info represents detailed information about the Tarantool instance. | ||
// It includes version, node ID, read-only status, process ID, cluster information, and more. | ||
type Info struct { | ||
// The Version of the Tarantool instance. | ||
Version string `msgpack:"version"` | ||
// The node ID (nullable). | ||
ID *int `msgpack:"id"` | ||
oleg-jukovec marked this conversation as resolved.
Show resolved
Hide resolved
|
||
// Read-only (RO) status of the instance. | ||
RO bool `msgpack:"ro"` | ||
// UUID - Unique identifier of the instance. | ||
UUID string `msgpack:"uuid"` | ||
// Process ID of the instance. | ||
PID int `msgpack:"pid"` | ||
// Status - Current status of the instance (e.g., running, unconfigured). | ||
Status string `msgpack:"status"` | ||
// LSN - Log sequence number of the instance. | ||
LSN uint64 `msgpack:"lsn"` | ||
} | ||
oleg-jukovec marked this conversation as resolved.
Show resolved
Hide resolved
|
||
|
||
// InfoResponse represents the response structure | ||
// that holds the information of the Tarantool instance. | ||
// It contains a single field: Info, which holds the instance details (version, UUID, PID, etc.). | ||
type InfoResponse struct { | ||
Info Info | ||
} | ||
|
||
func (ir *InfoResponse) DecodeMsgpack(d *msgpack.Decoder) error { | ||
arrayLen, err := d.DecodeArrayLen() | ||
if err != nil { | ||
return err | ||
} | ||
|
||
if arrayLen != 1 { | ||
return fmt.Errorf("protocol violation; expected 1 array entry, got %d", arrayLen) | ||
} | ||
oleg-jukovec marked this conversation as resolved.
Show resolved
Hide resolved
|
||
|
||
i := Info{} | ||
err = d.Decode(&i) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
ir.Info = i | ||
|
||
return nil | ||
} | ||
|
||
// InfoRequest represents a request to retrieve information about the Tarantool instance. | ||
// It implements the tarantool.Request interface. | ||
type InfoRequest struct { | ||
baseRequest | ||
} | ||
|
||
// Body method is used to serialize the request's body. | ||
// It is part of the tarantool.Request interface implementation. | ||
func (i InfoRequest) Body(res tarantool.SchemaResolver, enc *msgpack.Encoder) error { | ||
return i.impl.Body(res, enc) | ||
} | ||
|
||
// NewInfoRequest returns a new empty info request. | ||
func NewInfoRequest() InfoRequest { | ||
req := InfoRequest{} | ||
req.impl = newCall("box.info") | ||
return req | ||
} |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,38 @@ | ||
package box | ||
|
||
import ( | ||
"context" | ||
"io" | ||
|
||
"github.com/tarantool/go-iproto" | ||
"github.com/tarantool/go-tarantool/v2" | ||
) | ||
|
||
type baseRequest struct { | ||
impl *tarantool.CallRequest | ||
} | ||
|
||
func newCall(method string) *tarantool.CallRequest { | ||
return tarantool.NewCallRequest(method) | ||
} | ||
|
||
// Type returns IPROTO type for request. | ||
func (req baseRequest) Type() iproto.Type { | ||
return req.impl.Type() | ||
} | ||
|
||
// Ctx returns a context of request. | ||
func (req baseRequest) Ctx() context.Context { | ||
return req.impl.Ctx() | ||
} | ||
|
||
// Async returns request expects a response. | ||
func (req baseRequest) Async() bool { | ||
return req.impl.Async() | ||
} | ||
|
||
// Response creates a response for the baseRequest. | ||
func (req baseRequest) Response(header tarantool.Header, | ||
body io.Reader) (tarantool.Response, error) { | ||
return req.impl.Response(header, body) | ||
} |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,86 @@ | ||
package box_test | ||
|
||
import ( | ||
"context" | ||
"log" | ||
"os" | ||
"testing" | ||
"time" | ||
|
||
"github.com/google/uuid" | ||
"github.com/stretchr/testify/require" | ||
"github.com/tarantool/go-tarantool/v2" | ||
"github.com/tarantool/go-tarantool/v2/box" | ||
"github.com/tarantool/go-tarantool/v2/test_helpers" | ||
) | ||
|
||
var server = "127.0.0.1:3013" | ||
var dialer = tarantool.NetDialer{ | ||
Address: server, | ||
User: "test", | ||
Password: "test", | ||
} | ||
|
||
func validateInfo(t testing.TB, info box.Info) { | ||
var err error | ||
|
||
// Check all fields run correctly. | ||
_, err = uuid.Parse(info.UUID) | ||
require.NoErrorf(t, err, "validate instance uuid is valid") | ||
|
||
require.NotEmpty(t, info.Version) | ||
// Check that pid parsed correctly. | ||
require.NotEqual(t, info.PID, 0) | ||
} | ||
|
||
func TestBox_Sugar_Info(t *testing.T) { | ||
ctx := context.TODO() | ||
|
||
conn, err := tarantool.Connect(ctx, dialer, tarantool.Opts{}) | ||
require.NoError(t, err) | ||
|
||
info, err := box.New(conn).Info() | ||
require.NoError(t, err) | ||
|
||
validateInfo(t, info) | ||
} | ||
|
||
func TestBox_Info(t *testing.T) { | ||
ctx := context.TODO() | ||
|
||
conn, err := tarantool.Connect(ctx, dialer, tarantool.Opts{}) | ||
require.NoError(t, err) | ||
|
||
fut := conn.Do(box.NewInfoRequest()) | ||
require.NotNil(t, fut) | ||
|
||
resp := &box.InfoResponse{} | ||
err = fut.GetTyped(resp) | ||
require.NoError(t, err) | ||
|
||
validateInfo(t, resp.Info) | ||
} | ||
|
||
func runTestMain(m *testing.M) int { | ||
instance, err := test_helpers.StartTarantool(test_helpers.StartOpts{ | ||
Dialer: dialer, | ||
InitScript: "testdata/config.lua", | ||
Listen: server, | ||
WaitStart: 100 * time.Millisecond, | ||
ConnectRetry: 10, | ||
RetryTimeout: 500 * time.Millisecond, | ||
}) | ||
defer test_helpers.StopTarantoolWithCleanup(instance) | ||
|
||
if err != nil { | ||
log.Printf("Failed to prepare test Tarantool: %s", err) | ||
return 1 | ||
} | ||
|
||
return m.Run() | ||
} | ||
|
||
func TestMain(m *testing.M) { | ||
code := runTestMain(m) | ||
os.Exit(code) | ||
} |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,13 @@ | ||
-- Do not set listen for now so connector won't be | ||
-- able to send requests until everything is configured. | ||
box.cfg{ | ||
work_dir = os.getenv("TEST_TNT_WORK_DIR"), | ||
} | ||
|
||
box.schema.user.create('test', { password = 'test' , if_not_exists = true }) | ||
box.schema.user.grant('test', 'execute', 'universe', nil, { if_not_exists = true }) | ||
|
||
-- Set listen only when every other thing is configured. | ||
box.cfg{ | ||
listen = os.getenv("TEST_TNT_LISTEN"), | ||
} |
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.