Merge pull request #2930 from actiontech/2222

knowledge base support license

Author: iwanghc

sqle/config/config.go

@@ -41,7 +41,6 @@ type SqleOptions struct {
 	Service                   SeviceOpts         `yaml:"service"`
 	OptimizationConfig        OptimizationConfig `yaml:"optimization_config"`
 	SQLRewritingConfig        SQLRewritingConfig `yaml:"sql_rewriting_config"`
-	KnowledgeBaseTempLicense  string             `yaml:"knowledge_base_temp_license"`
 }
 
 type SeviceOpts struct {

sqle/dms/license.go

@@ -0,0 +1,40 @@
+package dms
+
+import (
+	"context"
+	"fmt"
+	pkgHttp "github.com/actiontech/dms/pkg/dms-common/pkg/http"
+)
+
+type GetLicenseReply struct {
+	Code    int    `json:"code"`
+	Message string `json:"message"`
+	Content string `json:"content"` // 只需要Content因此只解析Content
+}
+
+// GET /v1/dms/configurations/license Configuration GetLicense
+func GetLicense(ctx context.Context, dmsAddr string) (*GetLicenseReply, error) {
+	// Prepare the header with authorization token (assuming it's similar to the other request)
+	header := map[string]string{
+		"Authorization": pkgHttp.DefaultDMSToken, // Replace with your token if needed
+	}
+
+	// Define the response structure to hold the data
+	reply := &GetLicenseReply{}
+
+	// Construct the URL for the GET request
+	url := fmt.Sprintf("%v/v1/dms/configurations/license", dmsAddr)
+
+	// Perform the GET request
+	if err := pkgHttp.Get(ctx, url, header, nil, reply); err != nil {
+		return nil, fmt.Errorf("failed to get license from %v: %v", url, err)
+	}
+
+	// Handle the response code (if the reply structure contains such validation)
+	if reply.Code != 0 { // Assuming a Code field in the reply, modify accordingly
+		return nil, fmt.Errorf("http reply code(%v) error: %v", reply.Code, reply.Message)
+	}
+
+	// Return the retrieved license data
+	return reply, nil
+}

sqle/license/temp_knowledge_license_ce.go

@@ -3,8 +3,16 @@
 
 package knowledge_base
 
-import "github.com/actiontech/sqle/sqle/model"
+import (
+	"fmt"
+
+	"github.com/actiontech/sqle/sqle/model"
+)
 
 func LoadKnowledge(rulesMap map[string][]*model.Rule) error {
 	return nil
 }
+
+func CheckKnowledgeBaseLicense() error {
+	return fmt.Errorf("license not support knowledge base")
+}

sqle/server/knowledge_base/knowledge_base_ce.go

@@ -3,10 +3,9 @@ package server
 import (
 	"fmt"
 
-	"github.com/actiontech/sqle/sqle/config"
 	"github.com/actiontech/sqle/sqle/driver"
 	driverV2 "github.com/actiontech/sqle/sqle/driver/v2"
-	"github.com/actiontech/sqle/sqle/license"
+	svcKnowledgeBase "github.com/actiontech/sqle/sqle/server/knowledge_base"
 	optimization "github.com/actiontech/sqle/sqle/server/optimization/rule"
 )
 
@@ -65,7 +64,7 @@ type knowledgeBaseChecker struct {
 }
 
 func (s knowledgeBaseChecker) CheckIsSupport() bool {
-	if license.CheckKnowledgeBaseLicense(config.GetOptions().SqleOptions.KnowledgeBaseTempLicense) != nil {
+	if svcKnowledgeBase.CheckKnowledgeBaseLicense() != nil {
 		return false
 	}
 	return true