Branch data     Line data    Source code

       1                 :            : /*
       2                 :            :  * Copyright (C) 2026 by Thun Lu. All rights reserved.
       3                 :            :  * Author: Thun Lu <thun.lu@zohomail.cn>
       4                 :            :  * Repo:   https://github.com/thun-res/vlink
       5                 :            :  *  _    __   __      _           __
       6                 :            :  * | |  / /  / /     (_) ____    / /__
       7                 :            :  * | | / /  / /     / / / __ \  / //_/
       8                 :            :  * | |/ /  / /___  / / / / / / / ,<
       9                 :            :  * |___/  /_____/ /_/ /_/ /_/ /_/|_|
      10                 :            :  *
      11                 :            :  * Licensed under the Apache License, Version 2.0 (the "License");
      12                 :            :  * you may not use this file except in compliance with the License.
      13                 :            :  * You may obtain a copy of the License at
      14                 :            :  *
      15                 :            :  *     http://www.apache.org/licenses/LICENSE-2.0
      16                 :            :  *
      17                 :            :  * Unless required by applicable law or agreed to in writing, software
      18                 :            :  * distributed under the License is distributed on an "AS IS" BASIS,
      19                 :            :  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
      20                 :            :  * See the License for the specific language governing permissions and
      21                 :            :  * limitations under the License.
      22                 :            :  */
      23                 :            : 
      24                 :            : #include "./intra_subscriber_impl.h"
      25                 :            : 
      26                 :            : #include <utility>
      27                 :            : 
      28                 :            : #include "./base/helpers.h"
      29                 :            : 
      30                 :            : namespace vlink {
      31                 :            : 
      32                 :            : // IntraSubscriberImpl
      33         [ +  - ]:        185 : IntraSubscriberImpl::IntraSubscriberImpl(const IntraConf& conf) : conf_(conf) {
      34         [ +  + ]:        185 :   if (conf.type == "direct") {
      35                 :         26 :     type_ = IntraType::kDirect;
      36                 :            :   }
      37                 :        185 : }
      38                 :            : 
      39                 :        182 : void IntraSubscriberImpl::init() {
      40   [ +  +  +  -  :        182 :   static auto& factory = IntraFactory::get();
             +  -  -  - ]
      41                 :            : 
      42                 :        182 :   conf_.hash_code = Helpers::get_hash_code(conf_.event);
      43                 :            : 
      44         [ +  - ]:        182 :   object_ = factory.get_object<Object>({kImplType, conf_.address, conf_.pipeline, type_, conf_.hash_code});
      45                 :        182 :   object_->add_impl(this);
      46                 :        182 : }
      47                 :            : 
      48                 :        182 : void IntraSubscriberImpl::deinit() {
      49                 :        182 :   bool was_listened = is_listened;
      50                 :            : 
      51                 :        182 :   object_->remove_impl(this);
      52                 :        182 :   is_listened = false;
      53                 :            : 
      54   [ +  +  +  +  :        182 :   if (was_listened && object_->msg_map_is_empty() && object_->intra_msg_map_is_empty()) {
             +  +  +  + ]
      55   [ +  -  +  - ]:        326 :     object_->traverse_sub_connect_callback([](NodeImpl*, const auto& callback) { callback(false); });
      56                 :            :   }
      57                 :        182 : }
      58                 :            : 
      59                 :          2 : bool IntraSubscriberImpl::suspend() {
      60                 :          2 :   has_suspend = true;
      61                 :            : 
      62                 :          2 :   return true;
      63                 :            : }
      64                 :            : 
      65                 :          2 : bool IntraSubscriberImpl::resume() {
      66                 :          2 :   has_suspend = false;
      67                 :            : 
      68                 :          2 :   return true;
      69                 :            : }
      70                 :            : 
      71                 :          4 : bool IntraSubscriberImpl::is_suspend() const { return has_suspend; }
      72                 :            : 
      73                 :       1230 : const Conf* IntraSubscriberImpl::get_conf() const { return &conf_; }
      74                 :            : 
      75                 :          1 : const AbstractNode* IntraSubscriberImpl::get_abstract_node() const { return object_.get(); }
      76                 :            : 
      77                 :        166 : bool IntraSubscriberImpl::listen(MsgCallback&& callback) {
      78   [ +  +  +  - ]:        166 :   const bool was_empty = object_->msg_map_is_empty() && object_->intra_msg_map_is_empty();
      79                 :            : 
      80                 :        166 :   object_->register_msg_callback(this, std::move(callback));
      81                 :            : 
      82   [ +  +  +  +  :        166 :   if VLIKELY (was_empty || init_impl_type == kGetter) {
                   +  + ]
      83   [ +  -  +  - ]:        311 :     object_->traverse_sub_connect_callback([](NodeImpl*, const auto& target_callback) { target_callback(true); });
      84                 :            :   }
      85                 :            : 
      86                 :        166 :   return true;
      87                 :            : }
      88                 :            : 
      89                 :         10 : bool IntraSubscriberImpl::listen(IntraMsgCallback&& callback) {
      90   [ +  -  +  + ]:         10 :   const bool was_empty = object_->msg_map_is_empty() && object_->intra_msg_map_is_empty();
      91                 :            : 
      92                 :         10 :   object_->register_intra_msg_callback(this, std::move(callback));
      93                 :            : 
      94         [ +  + ]:         10 :   if VLIKELY (was_empty) {
      95   [ +  -  +  - ]:         17 :     object_->traverse_sub_connect_callback([](NodeImpl*, const auto& target_callback) { target_callback(true); });
      96                 :            :   }
      97                 :            : 
      98                 :         10 :   return true;
      99                 :            : }
     100                 :            : 
     101                 :            : }  // namespace vlink