feat: change ping to verify endpoint (#161)

Author: everpcpc

.github/workflows/ci.yaml

@@ -18,7 +18,7 @@ jobs:
       - name: Setup Go
         uses: actions/setup-go@v5
         with:
-          go-version: "1.21"
+          go-version: "1.24"
 
       - name: Unittest
         run: |
@@ -32,7 +32,7 @@ jobs:
 
   notify:
     if: failure()
-    needs: [ test ]
+    needs: test
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -54,4 +54,4 @@ jobs:
               headers: {'Content-Type': 'application/json'}
             });
             const result = await response.json();
-            console.log(result);
+            console.log(result);

VERSION

@@ -1 +1 @@
-0.7.5
+0.8.0

client.go

@@ -761,10 +761,24 @@ func (c *APIClient) UploadToStageByAPI(ctx context.Context, stage *StageLocation
 	return nil
 }
 
+func (c *APIClient) Verify(ctx context.Context) error {
+	var response VerifyResponse
+	err := c.doRequest(ctx, "POST", "/v1/verify", nil, false, &response, nil)
+	if err != nil {
+		return errors.Wrap(err, "verify failed")
+	}
+	if c.tenant != "" && response.Tenant != c.tenant {
+		return errors.Errorf("verify tenant mismatch, expected: %s, got: %s", c.tenant, response.Tenant)
+	}
+	if c.user != "" && response.User != c.user {
+		return errors.Errorf("verify user mismatch, expected: %s, got: %s", c.user, response.User)
+	}
+	return nil
+}
+
 func (c *APIClient) Logout(ctx context.Context) error {
 	if c.NeedKeepAlive() {
-		req := &struct{}{}
-		return c.doRequest(ctx, "POST", "/v1/session/logout/", req, c.NeedSticky(), nil, nil)
+		return c.doRequest(ctx, "POST", "/v1/session/logout", nil, c.NeedSticky(), nil, nil)
 	}
 	return nil
 }

connection.go

@@ -9,6 +9,7 @@ import (
 	"sync/atomic"
 
 	"github.com/google/uuid"
+	"github.com/pkg/errors"
 )
 
 const (
@@ -98,9 +99,9 @@ func (dc *DatabendConn) cleanup() {
 }
 
 func (dc *DatabendConn) Ping(ctx context.Context) error {
-	_, err := dc.exec(ctx, "SELECT 1")
+	err := dc.rest.Verify(ctx)
 	if err != nil {
-		return err
+		return errors.Wrap(err, "ping failed")
 	}
 	return nil
 }

query.go

@@ -137,3 +137,9 @@ func parseAffectedRows(queryResp *QueryResponse) (int64, error) {
 	}
 	return 0, nil
 }
+
+type VerifyResponse struct {
+	Tenant string   `json:"tenant"`
+	User   string   `json:"user"`
+	Roles  []string `json:"roles"`
+}