main.go 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. package main
  2. import (
  3. "container/list"
  4. "embed"
  5. be "spider_creator/backend"
  6. bdb "spider_creator/backend/db"
  7. "spider_creator/backend/script"
  8. bvm "spider_creator/backend/vm"
  9. bws "spider_creator/backend/webservice"
  10. qu "jygit.jydev.jianyu360.cn/data_processing/common_utils"
  11. "github.com/wailsapp/wails/v2"
  12. "github.com/wailsapp/wails/v2/pkg/options"
  13. "github.com/wailsapp/wails/v2/pkg/options/assetserver"
  14. )
  15. var (
  16. //go:embed all:frontend/dist
  17. assets embed.FS
  18. app *App
  19. db *bdb.SpiderDb
  20. exitCh chan bool
  21. baseDir, attachesDir string = ".", ""
  22. qlmDir string = ""
  23. currentSpiderConfig *be.SpiderConfig = new(be.SpiderConfig)
  24. currentResults = list.New() //b.ResultItems = make(b.ResultItems, 0)
  25. vm *bvm.VM
  26. glvm *script.GLVm
  27. ws *bws.WebService
  28. //重点网站和正式环境
  29. isOnly4MainSite string = "false"
  30. browserDisableLoadResources = "ws://;wss://;.tof;.woff;.ico;.mp4;.zip;.rar;.exe;"
  31. serverAddress = "http://visualizeld.spdata.jianyu360.com/%s" //正式环境
  32. //serverAddress = "http://127.0.0.1:8091/%s" //正式环境
  33. )
  34. //build
  35. // wails build -ldflags="-X 'main.isOnly4MainSite=false'" -o="剑鱼可视化爬虫开发工具_正式.exe"
  36. func init() {
  37. //be.LoadConfig("./config.yaml")
  38. be.Cfg.DisableLoadResource = browserDisableLoadResources
  39. be.Cfg.IsOnly4MainSite = isOnly4MainSite == "true"
  40. if be.Cfg.IsOnly4MainSite {
  41. serverAddress = "http://visualize.spdata.jianyu360.com/%s" //重点网站
  42. }
  43. qu.Debug("重点网站:", be.Cfg.IsOnly4MainSite, serverAddress)
  44. }
  45. func main() {
  46. go updateData() //数据保存
  47. // Create an instance of the app structure
  48. app = NewApp()
  49. // Create application with options
  50. err := wails.Run(&options.App{
  51. Title: "剑鱼-爬虫开发平台 v1.0",
  52. Width: 1224,
  53. Height: 668,
  54. AssetServer: &assetserver.Options{
  55. Assets: assets,
  56. },
  57. BackgroundColour: &options.RGBA{R: 255, G: 255, B: 255, A: 1},
  58. OnStartup: app.startup,
  59. OnShutdown: app.destory,
  60. Bind: []interface{}{
  61. app,
  62. },
  63. })
  64. if err != nil {
  65. println("Error:", err.Error())
  66. }
  67. }