From 052d32379cf3a19d35591263ec82fae8c56c5555 Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Sat, 5 Jun 2021 15:39:02 +0800 Subject: [PATCH] =?UTF-8?q?=E5=9F=9F=E5=90=8D=E6=9C=8D=E5=8A=A1=E9=9B=86?= =?UTF-8?q?=E7=BE=A4=E5=88=9B=E5=BB=BA=E6=97=B6=E5=8F=AF=E4=BB=A5=E9=80=89?= =?UTF-8?q?=E6=8B=A9=E5=BC=80=E5=90=AF=E8=AE=BF=E9=97=AE=E6=97=A5=E5=BF=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/db/models/nameservers/ns_cluster_dao.go | 7 ++++++- internal/rpc/services/nameservers/service_ns_cluster.go | 2 +- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/internal/db/models/nameservers/ns_cluster_dao.go b/internal/db/models/nameservers/ns_cluster_dao.go index 1a87810b..c3fdf06d 100644 --- a/internal/db/models/nameservers/ns_cluster_dao.go +++ b/internal/db/models/nameservers/ns_cluster_dao.go @@ -73,9 +73,14 @@ func (this *NSClusterDAO) FindEnabledNSClusterName(tx *dbs.Tx, id int64) (string } // CreateCluster 创建集群 -func (this *NSClusterDAO) CreateCluster(tx *dbs.Tx, name string) (int64, error) { +func (this *NSClusterDAO) CreateCluster(tx *dbs.Tx, name string, accessLogRefJSON []byte) (int64, error) { op := NewNSClusterOperator() op.Name = name + + if len(accessLogRefJSON) > 0 { + op.AccessLog = accessLogRefJSON + } + op.IsOn = true op.State = NSClusterStateEnabled return this.SaveInt64(tx, op) diff --git a/internal/rpc/services/nameservers/service_ns_cluster.go b/internal/rpc/services/nameservers/service_ns_cluster.go index ee34818a..1c5db62b 100644 --- a/internal/rpc/services/nameservers/service_ns_cluster.go +++ b/internal/rpc/services/nameservers/service_ns_cluster.go @@ -21,7 +21,7 @@ func (this *NSClusterService) CreateNSCluster(ctx context.Context, req *pb.Creat return nil, err } var tx = this.NullTx() - clusterId, err := nameservers.SharedNSClusterDAO.CreateCluster(tx, req.Name) + clusterId, err := nameservers.SharedNSClusterDAO.CreateCluster(tx, req.Name, req.AccessLogJSON) if err != nil { return nil, err }