udeServer/internal/logic/deleteservicelogic.go

// udeServer: DeleteService 走事务 —— 同一事务内先删该服务下的 t_directory,
// 再删 t_service。任一步失败全部回滚,不会残留孤儿目录数据。

package logic

import (
	"context"
	"fmt"
	"io"
	"net/http"
	"net/url"
	"path"
	"strings"

	"udeServer/internal/audit"
	"udeServer/internal/model"
	"udeServer/internal/svc"
	"udeServer/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

type DeleteServiceLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewDeleteServiceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteServiceLogic {
	return &DeleteServiceLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *DeleteServiceLogic) DeleteService(req *types.ServiceIdReq) (*types.DeleteResp, error) {
	if err := checkId(req.Id); err != nil {
		return nil, err
	}
	// 先查出服务实体:删库前要拿 ip/port 去通知外挂程序卸载,删了就查不到了
	svc, err := findService(l.ctx, l.svcCtx.ServiceModel, req.Id)
	if err != nil {
		return nil, err
	}

	// P1 联动卸载:先卸载后删库,卸载成功是删除的前置条件(硬阻断)。
	// 任一目录卸载失败(外挂不可达/返回错误)都中止删除,失败原因原样返回给前端,
	// 避免服务记录删了、挂载还挂在外挂机上成为无人管理的孤儿挂载。
	unmountErr := l.notifyFuseUnmount(svc)
	if unmountErr != "" {
		return nil, fmt.Errorf("删除已中止,请先解决挂载卸载问题后重试: %s", unmountErr)
	}

	err = l.svcCtx.DB.Transact(func(session sqlx.Session) error {
		dirModel := l.svcCtx.DirectoryModel.WithSession(session)
		svcModel := l.svcCtx.ServiceModel.WithSession(session)

		if err := dirModel.DeleteByServiceId(l.ctx, req.Id); err != nil {
			return fmt.Errorf("delete t_directory of service %d: %w", req.Id, err)
		}
		if err := svcModel.Delete(l.ctx, req.Id); err != nil {
			return fmt.Errorf("delete t_service %d: %w", req.Id, err)
		}
		return nil
	})
	if err != nil {
		l.Errorf("delete service tx failed: %v", err)
		return nil, err
	}

	return &types.DeleteResp{Ok: true}, nil
}

// notifyFuseUnmount 删库前回调外挂程序(svc.Ip:svc.Port)的 GET /api/fuseUnMount 接口,
func (l *DeleteServiceLogic) notifyFuseUnmount(svc *model.TService) string {
	// MountStatus 是 sql.NullString 类型,.String 取其中的字符串值
	if svc.MountStatus.String != "mounted" {
		return "" // 未挂载/挂载失败,库里明确没有活动挂载,无需通知外挂
	}
	ip, port := svc.Ip.String, svc.Port // port 已是 int64
	if ip == "" || port == 0 {
		return "外挂服务 IP/端口为空,无法联动卸载,目录可能仍处于挂载状态"
	}
	dirs, err := l.svcCtx.DirectoryModel.FindDirsByServiceId(l.ctx, svc.Id)
	if err != nil {
		return fmt.Sprintf("查询保护目录失败,无法联动卸载: %v", err)
	}
	if len(dirs) == 0 {
		return "" // 没配保护目录,无需卸载
	}
	baseURL := fmt.Sprintf("http://%s:%d/api/fuseUnMount", ip, port) // %d:port 是 int64
	var msgs []string
	for _, mount := range dirs {
		// path.Clean(非 filepath!):该路径发给远程 Linux 外挂程序,必须保持 / 分隔符
		cleanMount := path.Clean(mount)
		reqURL, uerr := url.Parse(baseURL)
		if uerr != nil {
			msgs = append(msgs, fmt.Sprintf("解析 URL 失败: %v", uerr))
			continue
		}
		// q.Encode() 自动做 URL 编码(/ → %2F),外挂侧 Query().Get 自动解码还原
		q := reqURL.Query()
		q.Set("mount", cleanMount)
		reqURL.RawQuery = q.Encode()
		l.Infof("fuse unmount forward: GET %s", reqURL.String()) // 打出转发 URL 便于排查
		// DoSimple 记录 type=2 接口日志,父 Span 指向本次入站请求(GET 用 url 描述请求原文)
		if _, callErr := audit.DoSimple(l.svcCtx.SpanRecorder, l.ctx, "FuseUnMount", reqURL.String(),
			audit.BuildRemoteGetRequest(http.MethodGet, reqURL.String()),
			func() (map[string]interface{}, int64, error) {
				// doFuseRequest 注入 X-TOKEN(按 iden 换算的 apiSecret);client.Get 快捷方式带不了自定义头,裸发会被外挂 401 拒绝
				resp, rerr := doFuseRequest(l.ctx, l.svcCtx, fuseMountClient, svc.Iden.String, http.MethodGet, reqURL.String(), nil, "") // 复用挂载逻辑里的 HTTP 客户端(同包变量)
				if rerr != nil {
					return nil, 0, fmt.Errorf("卸载目录 %s 失败(外挂程序未启动?): %w", cleanMount, rerr)
				}
				raw, _ := io.ReadAll(resp.Body) // 读到 EOF 保证 TCP 连接复用 Keep-Alive
				resp.Body.Close()
				status := int64(resp.StatusCode)
				if status >= http.StatusBadRequest {
					return nil, status, fmt.Errorf("卸载目录 %s 失败,外挂程序返回 HTTP %d: %s", cleanMount, status, string(raw))
				}
				return map[string]interface{}{"status": status, "body": string(raw)}, status, nil
			}); callErr != nil {
			msgs = append(msgs, callErr.Error())
			continue
		}
		l.Infof("fuse unmount ok: %s", cleanMount)
	}
	if len(msgs) == 0 {
		return ""
	}
	return "卸载挂载失败: " + strings.Join(msgs, "; ")
}