diff --git a/internal/accesslogs/storage_manager_write.go b/internal/accesslogs/storage_manager_write.go index 9710d0a1..f8b212f2 100644 --- a/internal/accesslogs/storage_manager_write.go +++ b/internal/accesslogs/storage_manager_write.go @@ -1,7 +1,6 @@ // Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved. //go:build !plus -// +build !plus package accesslogs @@ -10,6 +9,6 @@ import ( ) // 写入日志 -func (this *StorageManager) Write(policyId int64, accessLogs []*pb.HTTPAccessLog) error { - return nil +func (this *StorageManager) Write(policyId int64, accessLogs []*pb.HTTPAccessLog) (success bool, failMessage string, err error) { + return false, "only works in plus version", nil } diff --git a/internal/rpc/services/service_http_access_log.go b/internal/rpc/services/service_http_access_log.go index 0627d76e..43ed244c 100644 --- a/internal/rpc/services/service_http_access_log.go +++ b/internal/rpc/services/service_http_access_log.go @@ -43,7 +43,7 @@ func (this *HTTPAccessLogService) CreateHTTPAccessLogs(ctx context.Context, req return nil, err } if policyId > 0 { - err = accesslogs.SharedStorageManager.Write(policyId, req.HttpAccessLogs) + _, _, err = accesslogs.SharedStorageManager.Write(policyId, req.HttpAccessLogs) if err != nil { return nil, err } diff --git a/internal/rpc/services/service_http_access_log_policy.go b/internal/rpc/services/service_http_access_log_policy.go index b9d4b458..7abdca15 100644 --- a/internal/rpc/services/service_http_access_log_policy.go +++ b/internal/rpc/services/service_http_access_log_policy.go @@ -2,6 +2,7 @@ package services import ( "context" + "errors" "github.com/TeaOSLab/EdgeAPI/internal/accesslogs" "github.com/TeaOSLab/EdgeAPI/internal/db/models" "github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb" @@ -153,9 +154,12 @@ func (this *HTTPAccessLogPolicyService) WriteHTTPAccessLogPolicy(ctx context.Con return nil, err } - err = accesslogs.SharedStorageManager.Write(req.HttpAccessLogPolicyId, []*pb.HTTPAccessLog{req.HttpAccessLog}) + success, failMessage, err := accesslogs.SharedStorageManager.Write(req.HttpAccessLogPolicyId, []*pb.HTTPAccessLog{req.HttpAccessLog}) if err != nil { return nil, err } + if !success { + return nil, errors.New("test failed: " + failMessage) + } return this.Success() }