You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
1.1 KiB
73 lines
1.1 KiB
package service |
|
|
|
import ( |
|
"context" |
|
"sync" |
|
|
|
"go-common/app/admin/main/creative/conf" |
|
"go-common/app/admin/main/creative/dao" |
|
"go-common/library/log" |
|
|
|
"github.com/jinzhu/gorm" |
|
) |
|
|
|
// Service str |
|
type Service struct { |
|
conf *conf.Config |
|
dao *dao.Dao |
|
DB *gorm.DB |
|
DBArchive *gorm.DB |
|
wg sync.WaitGroup |
|
asynch chan func() |
|
closed bool |
|
} |
|
|
|
// New fn |
|
func New(c *conf.Config) (s *Service) { |
|
s = &Service{ |
|
conf: c, |
|
dao: dao.New(c), |
|
asynch: make(chan func(), 10240), |
|
} |
|
|
|
s.DB = s.dao.DB |
|
s.DBArchive = s.dao.DBArchive |
|
s.wg.Add(1) |
|
go s.asynproc() |
|
return |
|
} |
|
|
|
// Ping fn |
|
func (s *Service) Ping(c context.Context) (err error) { |
|
return s.dao.Ping(c) |
|
} |
|
|
|
func (s *Service) addAsyn(f func()) { |
|
select { |
|
case s.asynch <- f: |
|
default: |
|
log.Warn("asynproc chan full") |
|
} |
|
} |
|
|
|
// cacheproc is a routine for executing closure. |
|
func (s *Service) asynproc() { |
|
defer s.wg.Done() |
|
for { |
|
if s.closed { |
|
return |
|
} |
|
f, ok := <-s.asynch |
|
if !ok { |
|
return |
|
} |
|
f() |
|
} |
|
} |
|
|
|
// Close dao |
|
func (s *Service) Close() { |
|
s.dao.Close() |
|
s.closed = true |
|
s.wg.Wait() |
|
}
|
|
|