feature: add interactive-mode SQLExecutor with SQLRTTask

examples/sdk/sqa/with_interactive_sql_executor.go

@@ -0,0 +1,60 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"os"
+
+	"github.com/aliyun/aliyun-odps-go-sdk/odps"
+	"github.com/aliyun/aliyun-odps-go-sdk/odps/account"
+	"github.com/aliyun/aliyun-odps-go-sdk/odps/sqa"
+)
+
+func main() {
+	conf, err := odps.NewConfigFromIni(os.Args[1])
+	if err != nil {
+		log.Fatalf("%+v", err)
+	}
+
+	aliAccount := account.NewAliyunAccount(conf.AccessId, conf.AccessKey)
+	odpsIns := odps.NewOdps(aliAccount, conf.Endpoint)
+	odpsIns.SetDefaultProjectName(conf.ProjectName)
+	sql := `select * from all_types_demo_no_parition;`
+	//
+	paramInfo := sqa.SQLExecutorQueryParam{
+		OdpsIns:        odpsIns,
+		TaskName:       "test_mcqa",
+		ServiceName:    sqa.DEFAULT_SERVICE,
+		RunningCluster: "",
+	}
+	ie := sqa.NewInteractiveSQLExecutor(&paramInfo)
+	err = ie.Run(sql, nil)
+	if err != nil {
+		log.Fatalf("%+v", err)
+	}
+	//
+	records, err := ie.GetResult(0, 10, 10000, true)
+	if err != nil {
+		log.Fatalf("%+v", err)
+	}
+	//
+	for _, record := range records {
+		for i, d := range record {
+			if d == nil {
+				fmt.Printf("null")
+			} else {
+				fmt.Printf("%s", d.Sql())
+			}
+
+			if i < record.Len()-1 {
+				fmt.Printf(", ")
+			} else {
+				fmt.Println()
+			}
+		}
+	}
+	err = ie.Close()
+	if err != nil {
+		log.Fatalf("%+v", err)
+	}
+}

odps/instance.go

@@ -19,13 +19,15 @@ package odps
 import (
 	"encoding/json"
 	"encoding/xml"
-	"github.com/aliyun/aliyun-odps-go-sdk/odps/common"
-	"github.com/pkg/errors"
+	"io"
 	"io/ioutil"
 	"net/http"
 	"net/url"
 	"strings"
 	"time"
+
+	"github.com/aliyun/aliyun-odps-go-sdk/odps/common"
+	"github.com/pkg/errors"
 )
 
 type InstanceStatus int
@@ -270,6 +272,10 @@ func (instance *Instance) Id() string {
 	return instance.id
 }
 
+func (instance *Instance) ResourceUrl() string {
+	return instance.resourceUrl
+}
+
 func (instance *Instance) Owner() string {
 	return instance.owner
 }
@@ -358,6 +364,73 @@ func (instance *Instance) GetResult() ([]TaskResult, error) {
 	return resModel.Tasks, nil
 }
 
+type UpdateInfoResult struct {
+	Result string `json:"result"`
+	Status string `json:"status"`
+}
+
+// UpdateInfo set information to running instance
+func (instance *Instance) UpdateInfo(taskName, infoKey, infoValue string) (UpdateInfoResult, error) {
+	// instance set information
+	queryArgs := make(url.Values, 2)
+	queryArgs.Set("info", "")
+	queryArgs.Set("taskname", taskName)
+	//
+	instanceTaskInfoModel := struct {
+		XMLName xml.Name `xml:"Instance"`
+		Key     string   `xml:"Key"`
+		Value   string   `xml:"Value"`
+	}{
+		Key:   infoKey,
+		Value: infoValue,
+	}
+	//
+	var res UpdateInfoResult
+	client := instance.odpsIns.RestClient()
+	err := client.DoXmlWithParseRes(
+		common.HttpMethod.PutMethod,
+		instance.resourceUrl,
+		queryArgs,
+		instanceTaskInfoModel,
+		func(httpRes *http.Response) error {
+			err := json.NewDecoder(httpRes.Body).Decode(&res)
+			if err != nil {
+				return errors.Wrapf(err, "Parse http response failed, body: %+v", httpRes.Body)
+			}
+			return nil
+		},
+	)
+	//
+	return res, err
+}
+
+// GetTaskInfo get the specific info of a task in the instance
+func (instance *Instance) GetTaskInfo(taskName, infoKey string) (string, error) {
+	queryArgs := make(url.Values, 3)
+	queryArgs.Set("info", "")
+	queryArgs.Set("taskname", taskName)
+	queryArgs.Set("key", infoKey)
+
+	client := instance.odpsIns.RestClient()
+	var bodyStr string
+	err := client.DoXmlWithParseRes(
+		common.HttpMethod.GetMethod,
+		instance.resourceUrl,
+		queryArgs,
+		nil,
+		func(httpRes *http.Response) error {
+			body, err := io.ReadAll(httpRes.Body)
+			if err != nil {
+				return errors.Wrapf(err, "Parse http response failed, body: %+v", httpRes.Body)
+			}
+			bodyStr = string(body)
+			return nil
+		},
+	)
+
+	return bodyStr, err
+}
+
 func InstancesStatusFromStr(s string) InstanceStatus {
 	switch strings.ToLower(s) {
 	case "running":