feat: 实现发送质量监控与报备状态消息通知

gateway/internal/upstream/submit.go

@@ -40,7 +40,15 @@ func (p *connectionPool) submit(
 	ctx context.Context,
 	cmd queue.SubmitCommand,
 	onSegment func(queue.SubmitSegmentResult) error,
-) (queue.SubmitResult, error) {
+) (final queue.SubmitResult, finalErr error) {
+	defer func() {
+		for _, segment := range final.Segments {
+			if segment.FirstWireSubmitAt != nil && (final.FirstWireSubmitAt == nil || segment.FirstWireSubmitAt.Before(*final.FirstWireSubmitAt)) {
+				final.FirstWireSubmitAt = segment.FirstWireSubmitAt
+				final.WireTimeSource = "gateway_write_complete"
+			}
+		}
+	}()
 	parts, err := splitSubmitContent(cmd.CMPP.MsgFmt, cmd.Content)
 	if err != nil {
 		result := submitResult(cmd, 0, "", "rejected", "ENCODE_FAILED", err.Error())
@@ -96,7 +104,10 @@ func (p *connectionPool) submit(
 	return result, nil
 }
 
-func (c *connection) submitPart(ctx context.Context, cmd queue.SubmitCommand, part submitPart) (uint32, string, queue.SubmitResult, error) {
+func (c *connection) submitPart(ctx context.Context, cmd queue.SubmitCommand, part submitPart) (finalSequence uint32, finalID string, final queue.SubmitResult, finalErr error) {
+	var wireAt *time.Time
+	wireSource := "not_written"
+	defer func() { final.FirstWireSubmitAt = wireAt; final.WireTimeSource = wireSource }()
 	startedAt := time.Now()
 	defer func() {
 		c.mu.Lock()
@@ -116,7 +127,13 @@ func (c *connection) submitPart(ctx context.Context, cmd queue.SubmitCommand, pa
 		return 0, "", result, err
 	}
 	c.sendMu.Lock()
+	wireSource = "write_uncertain"
 	seq, err := client.SendReqPkt(pkt)
+	if err == nil {
+		at := time.Now().UTC()
+		wireAt = &at
+		wireSource = "gateway_write_complete"
+	}
 	c.sendMu.Unlock()
 	if err != nil {
 		c.emitProtocolLog(protocolLogEvent{
@@ -326,6 +343,8 @@ func submitResult(cmd queue.SubmitCommand, sequenceID uint32, gatewayMessageID s
 		ErrorCode:        code,
 		ErrorMessage:     message,
 		SubmittedAt:      time.Now().UTC(),
+		WireTimeSource:   "not_written",
+		ReceiptRequested: cmd.CMPP.RegisteredDelivery != 0,
 	}
 }
 
@@ -334,14 +353,17 @@ func submitSegmentResult(part submitPart, sequenceID uint32, gatewayMessageID st
 		gatewayMessageID = result.GatewayMessageID
 	}
 	return queue.SubmitSegmentResult{
-		SegmentTotal:     int(part.PkTotal),
-		SegmentIndex:     int(part.PkNumber),
-		SequenceID:       sequenceID,
-		GatewayMessageID: gatewayMessageID,
-		SubmitStatus:     result.SubmitStatus,
-		ErrorCode:        result.ErrorCode,
-		ErrorMessage:     result.ErrorMessage,
-		SubmittedAt:      result.SubmittedAt,
+		FirstWireSubmitAt: result.FirstWireSubmitAt,
+		WireTimeSource:    result.WireTimeSource,
+		ReceiptRequested:  result.ReceiptRequested,
+		SegmentTotal:      int(part.PkTotal),
+		SegmentIndex:      int(part.PkNumber),
+		SequenceID:        sequenceID,
+		GatewayMessageID:  gatewayMessageID,
+		SubmitStatus:      result.SubmitStatus,
+		ErrorCode:         result.ErrorCode,
+		ErrorMessage:      result.ErrorMessage,
+		SubmittedAt:       result.SubmittedAt,
 	}
 }