1
0

service.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. // Copyright 2017 fatedier, fatedier@gmail.com
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package client
  15. import (
  16. "github.com/fatedier/frp/models/config"
  17. "github.com/fatedier/frp/utils/log"
  18. )
  19. type Service struct {
  20. // manager control connection with server
  21. ctl *Control
  22. closedCh chan int
  23. }
  24. func NewService(pxyCfgs map[string]config.ProxyConf, visitorCfgs map[string]config.ProxyConf) (svr *Service) {
  25. svr = &Service{
  26. closedCh: make(chan int),
  27. }
  28. ctl := NewControl(svr, pxyCfgs, visitorCfgs)
  29. svr.ctl = ctl
  30. return
  31. }
  32. func (svr *Service) Run() error {
  33. err := svr.ctl.Run()
  34. if err != nil {
  35. return err
  36. }
  37. if config.ClientCommonCfg.AdminPort != 0 {
  38. err = svr.RunAdminServer(config.ClientCommonCfg.AdminAddr, config.ClientCommonCfg.AdminPort)
  39. if err != nil {
  40. log.Warn("run admin server error: %v", err)
  41. }
  42. log.Info("admin server listen on %s:%d", config.ClientCommonCfg.AdminAddr, config.ClientCommonCfg.AdminPort)
  43. }
  44. <-svr.closedCh
  45. return nil
  46. }
  47. func (svr *Service) Close() {
  48. svr.ctl.Close()
  49. }