github.com/e154/smart-home@v0.17.2-0.20240311175135-e530a6e5cd45/plugins/notify/notify.go (about)

     1  // This file is part of the Smart Home
     2  // Program complex distribution https://github.com/e154/smart-home
     3  // Copyright (C) 2023, Filippov Alex
     4  //
     5  // This library is free software: you can redistribute it and/or
     6  // modify it under the terms of the GNU Lesser General Public
     7  // License as published by the Free Software Foundation; either
     8  // version 3 of the License, or (at your option) any later version.
     9  //
    10  // This library is distributed in the hope that it will be useful,
    11  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    13  // Library General Public License for more details.
    14  //
    15  // You should have received a copy of the GNU Lesser General Public
    16  // License along with this library.  If not, see
    17  // <https://www.gnu.org/licenses/>.
    18  
    19  package notify
    20  
    21  import (
    22  	"github.com/alitto/pond"
    23  	"github.com/e154/smart-home/adaptors"
    24  	m "github.com/e154/smart-home/models"
    25  	"github.com/e154/smart-home/plugins/notify/common"
    26  )
    27  
    28  const (
    29  	maxWorkers = 5
    30  	minWorkers = 1
    31  )
    32  
    33  type Notify struct {
    34  	adaptors *adaptors.Adaptors
    35  	pool     *pond.WorkerPool
    36  }
    37  
    38  // NewNotify ...
    39  func NewNotify(adaptors *adaptors.Adaptors) *Notify {
    40  	return &Notify{
    41  		adaptors: adaptors,
    42  		pool:     pond.New(maxWorkers, 0, pond.MinWorkers(minWorkers)),
    43  	}
    44  }
    45  
    46  func (n *Notify) Start() {
    47  	n.pool.RunningWorkers()
    48  }
    49  
    50  func (n *Notify) Shutdown() {
    51  	n.pool.StopAndWait()
    52  }
    53  
    54  func (n *Notify) Send(msg *m.MessageDelivery, provider Provider) {
    55  	n.pool.Submit(func() {
    56  		NewWorker(n.adaptors).Send(msg, provider)
    57  	})
    58  }
    59  
    60  func (n *Notify) SaveAndSend(msg common.Message, provider Provider) {
    61  	n.pool.Submit(func() {
    62  		NewWorker(n.adaptors).SaveAndSend(msg, provider)
    63  	})
    64  }