fix: extension cannot start smoothly

Author: LinuxSuRen

console/atest-ui/src/App.vue

@@ -6,7 +6,7 @@ import {
     Share,
     ArrowDown,
     Guide,
-    DataAnalysis, Help, Setting
+    Help, Setting
 } from '@element-plus/icons-vue'
 import * as ElementPlusIcons from '@element-plus/icons-vue'
 import { ref, watch, getCurrentInstance} from 'vue'
@@ -152,16 +152,16 @@ API.GetMenus((menus) => {
           <el-icon><Guide /></el-icon>
           <template #title>{{ t('title.mock' )}}</template>
         </el-menu-item>
-        <el-menu-item index="store">
-          <el-icon><location /></el-icon>
-          <template #title>{{ t('title.stores') }}</template>
-        </el-menu-item>
         <span v-for="menu in extensionMenus" :key="menu.index" :index="menu.index">
             <el-menu-item :index="menu.index">
                 <el-icon><component :is="ElementPlusIcons[menu.icon]" /></el-icon>
                 <template #title>{{ menu.name }}</template>
             </el-menu-item>
         </span>
+        <el-menu-item index="store">
+          <el-icon><location /></el-icon>
+          <template #title>{{ t('title.stores') }}</template>
+        </el-menu-item>
       </el-menu>
     </el-aside>
 

pkg/server/remote_server.go

@@ -28,6 +28,7 @@ import (
 	"path/filepath"
 	reflect "reflect"
 	"regexp"
+	"runtime"
 	"slices"
 	"strconv"
 	"strings"
@@ -1247,10 +1248,7 @@ func (s *server) CreateStore(ctx context.Context, in *Store) (reply *Store, err
 	storeFactory := testing.NewStoreFactory(s.configDir)
 	store := ToNormalStore(in)
 
-	if store.Kind.URL == "" {
-		store.Kind.URL = fmt.Sprintf("unix://%s", home.GetExtensionSocketPath(store.Kind.Name))
-	}
-
+	handleStore(&store)
 	if err = storeFactory.CreateStore(store); err == nil && s.storeExtMgr != nil {
 		err = s.storeExtMgr.Start(store.Kind.Name, store.Kind.URL)
 	}
@@ -1260,6 +1258,7 @@ func (s *server) UpdateStore(ctx context.Context, in *Store) (reply *Store, err
 	reply = &Store{}
 	storeFactory := testing.NewStoreFactory(s.configDir)
 	store := ToNormalStore(in)
+	handleStore(&store)
 	if err = storeFactory.UpdateStore(store); err == nil && s.storeExtMgr != nil {
 		// TODO need to restart extension if config was changed
 		err = s.storeExtMgr.Start(store.Kind.Name, store.Kind.URL)
@@ -1276,14 +1275,21 @@ func (s *server) VerifyStore(ctx context.Context, in *SimpleQuery) (reply *Exten
 	reply = &ExtensionStatus{}
 	var loader testing.Writer
 	if loader, err = s.getLoaderByStoreName(in.Name); err == nil && loader != nil {
-		readOnly, verifyErr := loader.Verify()
+		readOnly, version, verifyErr := loader.Verify()
 		reply.Ready = verifyErr == nil
+		reply.Version = version
 		reply.ReadOnly = readOnly
 		reply.Message = util.OKOrErrorMessage(verifyErr)
 	}
 	return
 }
 
+func handleStore(store *testing.Store) {
+	if store.Kind.URL == "" && runtime.GOOS != "windows" {
+		store.Kind.URL = fmt.Sprintf("unix://%s", home.GetExtensionSocketPath(store.Kind.Name))
+	}
+}
+
 // secret related interfaces
 func (s *server) GetSecrets(ctx context.Context, in *Empty) (reply *Secrets, err error) {
 	return s.secretServer.GetSecrets(ctx, in)

pkg/server/server.pb.go

@@ -616,6 +616,7 @@ message StoreKind {
   repeated StoreKindDependency dependencies = 4;
   string link = 5;
   repeated StoreKindParam params = 6;
+  repeated string categories = 7;
 }
 
 message StoreKindDependency {

pkg/server/server.proto

@@ -98,13 +98,18 @@ func (s *storeExtManager) StartPlugin(storeKind testing.StoreKind) {
 }
 
 func (s *storeExtManager) Start(name, socket string) (err error) {
+	serverLogger.Info("start", "extension", name, "socket", socket)
 	if v, ok := s.extStatusMap[name]; ok && v {
 		return
 	}
+
 	platformBasedName := name
 	if s.execer.OS() == "windows" {
 		platformBasedName += ".exe"
+	} else {
+		socket = fmt.Sprintf("unix://%s", home.GetExtensionSocketPath(name))
 	}
+
 	targetDir := home.GetUserBinDir()
 	targetBinaryFile := filepath.Join(targetDir, platformBasedName)
 

pkg/server/store_ext_manager.go

@@ -24,7 +24,7 @@ type Loader interface {
 	GetCount() int
 	Reset()
 
-	Verify() (readOnly bool, err error)
+	Verify() (readOnly bool, version string, err error)
 	PProf(name string) []byte
 
 	Query(query map[string]string) (result DataResult, err error)

pkg/testing/loader.go

@@ -504,7 +504,7 @@ func (l *fileLoader) RenameTestSuite(oldName, newName string) (err error) {
 	return
 }
 
-func (l *fileLoader) Verify() (readOnly bool, err error) {
+func (l *fileLoader) Verify() (readOnly bool, version string, err error) {
 	// always be okay
 	return
 }

pkg/testing/loader_file.go

@@ -141,7 +141,7 @@ func (l *nonLoader) RenameTestSuite(oldName, newName string) (err error) {
 	return
 }
 
-func (l *nonLoader) Verify() (readOnly bool, err error) {
+func (l *nonLoader) Verify() (readOnly bool, version string, err error) {
 	// always be okay
 	return
 }

pkg/testing/loader_non.go

@@ -285,14 +285,15 @@ func (g *gRPCLoader) RenameTestSuite(oldName, newName string) (err error) {
 	return
 }
 
-func (g *gRPCLoader) Verify() (readOnly bool, err error) {
+func (g *gRPCLoader) Verify() (readOnly bool, version string, err error) {
 	// avoid to long to wait the response
 	ctx, cancel := context.WithTimeout(g.ctx, time.Second*5)
 	defer cancel()
 
 	var result *server.ExtensionStatus
 	if result, err = g.client.Verify(ctx, &server.Empty{}); err == nil {
 		readOnly = result.ReadOnly
+		version = result.Version
 		if !result.Ready {
 			err = errors.New(result.Message)
 		}

pkg/testing/remote/grpc_store.go

@@ -105,6 +105,7 @@ type StoreKind struct {
 	Params       []StoreKindParam
 	Link         string
 	Enabled      bool
+	Categories   []string
 }
 
 type StoreKindDependency struct {