From 4d8d8d2a3acef82d8d5d84eb7589c1f380c13c28 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Mon, 18 Mar 2024 11:24:36 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=BD=91=E5=85=83=E7=89=88=E6=9C=AC?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=E9=80=9A=E8=BF=87=E7=BD=91=E5=85=83=E7=B1=BB?=
 =?UTF-8?q?=E5=9E=8B=E5=92=8C=E7=BD=91=E5=85=83ID=E6=9F=A5=E8=AF=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../network_element/service/ne_version.go      |  3 +++
 .../network_element/service/ne_version.impl.go | 18 +++++++++++++++---
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/src/modules/network_element/service/ne_version.go b/src/modules/network_element/service/ne_version.go
index ed5aa979..d0ebc058 100644
--- a/src/modules/network_element/service/ne_version.go
+++ b/src/modules/network_element/service/ne_version.go
@@ -22,6 +22,9 @@ type INeVersion interface {
 	// DeleteByIds 批量删除信息
 	DeleteByIds(ids []string) (int64, error)
 
+	// SelectByTypeAndID 通过网元类型和网元ID查询
+	SelectByTypeAndID(neType, neId string) model.NeVersion
+
 	// CheckUniqueTypeAndID 校验网元类型和网元ID是否唯一
 	CheckUniqueTypeAndID(neType, neId, id string) bool
 }
diff --git a/src/modules/network_element/service/ne_version.impl.go b/src/modules/network_element/service/ne_version.impl.go
index 7e3b7239..d3f4f08b 100644
--- a/src/modules/network_element/service/ne_version.impl.go
+++ b/src/modules/network_element/service/ne_version.impl.go
@@ -33,9 +33,9 @@ func (r *NeVersionImpl) SelectById(id string) model.NeVersion {
 	if id == "" {
 		return model.NeVersion{}
 	}
-	neHosts := r.neVersionRepository.SelectByIds([]string{id})
-	if len(neHosts) > 0 {
-		return neHosts[0]
+	neVersions := r.neVersionRepository.SelectByIds([]string{id})
+	if len(neVersions) > 0 {
+		return neVersions[0]
 	}
 	return model.NeVersion{}
 }
@@ -66,6 +66,18 @@ func (r *NeVersionImpl) DeleteByIds(ids []string) (int64, error) {
 	return 0, fmt.Errorf("delete fail")
 }
 
+// SelectByTypeAndID 通过网元类型和网元ID查询
+func (r *NeVersionImpl) SelectByTypeAndID(neType, neId string) model.NeVersion {
+	neVersions := r.neVersionRepository.SelectList(model.NeVersion{
+		NeType: neType,
+		NeId:   neId,
+	})
+	if len(neVersions) > 0 {
+		return neVersions[0]
+	}
+	return model.NeVersion{}
+}
+
 // CheckUniqueTypeAndID 校验网元类型和网元ID是否唯一
 func (r *NeVersionImpl) CheckUniqueTypeAndID(neType, neId, id string) bool {
 	uniqueId := r.neVersionRepository.CheckUniqueTypeAndID(model.NeVersion{