feat: implement GetSelectivityOfSQLColumns method in PluginImplV1 and PluginImplV2, and add corresponding gRPC server handling

Author: iwanghc

sqle/driver/plugin_adapter_v1.go

@@ -353,3 +353,7 @@ func (s *PluginImplV1) GetDatabaseObjectDDL(ctx context.Context, objInfos []*dri
 func (s *PluginImplV1) GetDatabaseDiffModifySQL(ctx context.Context, calibratedDSN *driverV2.DSN, objInfos []*driverV2.DatabasCompareSchemaInfo) ([]*driverV2.DatabaseDiffModifySQLResult, error) {
 	return nil, fmt.Errorf("unimplement this method")
 }
+
+func (p *PluginImplV1) GetSelectivityOfSQLColumns(ctx context.Context, sql string) (map[string]map[string]float32, error) {
+	return nil, fmt.Errorf("unimplement this method")
+}

sqle/driver/plugin_adapter_v2.go

@@ -725,3 +725,25 @@ func (s *PluginImplV2) GetDatabaseDiffModifySQL(ctx context.Context, calibratedD
 	}
 	return dbDiffSQLs, nil
 }
+
+func (s *PluginImplV2) GetSelectivityOfSQLColumns(ctx context.Context, sql string) (map[string]map[string]float32, error) {
+	api := "GetSelectivityOfSQLColumns"
+	s.preLog(api)
+	resp, err := s.client.GetSelectivityOfSQLColumns(ctx, &protoV2.GetSelectivityOfSQLColumnsRequest{
+		Session: s.Session,
+		Sql:     sql,
+	})
+	s.afterLog(api, err)
+	if err != nil {
+		return nil, err
+	}
+	result := make(map[string]map[string]float32, len(resp.Selectivity))
+	for _, v := range resp.Selectivity {
+		colMap := make(map[string]float32, len(v.SelectivityOfColumns))
+		for k, sel := range v.SelectivityOfColumns {
+			colMap[k] = sel
+		}
+		result[v.TableName] = colMap
+	}
+	return result, nil
+}

sqle/driver/v2/driver_grpc_server.go

@@ -674,3 +674,27 @@ func (d *DriverGrpcServer) GetDatabaseDiffModifySQL(ctx context.Context, req *pr
 		SchemaDiffModify: scheamDiff,
 	}, nil
 }
+
+func (d *DriverGrpcServer) GetSelectivityOfSQLColumns(ctx context.Context, req *protoV2.GetSelectivityOfSQLColumnsRequest) (*protoV2.GetSelectivityOfSQLColumnsResponse, error) {
+	driver, err := d.getDriverBySession(req.Session)
+	if err != nil {
+		return &protoV2.GetSelectivityOfSQLColumnsResponse{}, err
+	}
+	selectivity, err := driver.GetSelectivityOfSQLColumns(ctx, req.Sql)
+	if err != nil {
+		return &protoV2.GetSelectivityOfSQLColumnsResponse{}, err
+	}
+	protoSelectivity := make([]*protoV2.SelectivityOfSQLColumns, 0, len(selectivity))
+	for tableName, colMap := range selectivity {
+		// 直接将 map[string]float32 赋值，无需合并操作
+		merged := make(map[string]float32, len(colMap))
+		for col, val := range colMap {
+			merged[col] = val
+		}
+		protoSelectivity = append(protoSelectivity, &protoV2.SelectivityOfSQLColumns{
+			TableName:            tableName,
+			SelectivityOfColumns: merged,
+		})
+	}
+	return &protoV2.GetSelectivityOfSQLColumnsResponse{Selectivity: protoSelectivity}, nil
+}

sqle/pkg/driver/impl.go

@@ -311,3 +311,7 @@ func (p *DriverImpl) GetDatabaseObjectDDL(ctx context.Context, objInfos []*drive
 func (p *DriverImpl) GetDatabaseDiffModifySQL(ctx context.Context, calibratedDSN *driverV2.DSN, objInfos []*driverV2.DatabasCompareSchemaInfo) ([]*driverV2.DatabaseDiffModifySQLResult, error) {
 	return []*driverV2.DatabaseDiffModifySQLResult{}, nil
 }
+
+func (p *DriverImpl) GetSelectivityOfSQLColumns(ctx context.Context, sql string) (map[string]map[string]float32, error) {
+	return nil, nil
+}

sqle/server/sqled_test.go

@@ -126,6 +126,10 @@ func (d *mockDriver) ExplainJSONFormat(ctx context.Context, conf *driverV2.Expla
 	return nil, nil
 }
 
+func (d *mockDriver) GetSelectivityOfSQLColumns(ctx context.Context, sql string) (map[string]map[string]float32, error) {
+	return make(map[string]map[string]float32), nil
+}
+
 func TestAction_validation(t *testing.T) {
 	actions := map[int]*action{
 		ActionTypeAudit:    {typ: ActionTypeAudit},