package main import ( "context" "flag" "fmt" "fusenapi/constants" "fusenapi/server/render/consumer" "net/http" "time" "fusenapi/utils/auth" "fusenapi/server/render/internal/config" "fusenapi/server/render/internal/handler" "fusenapi/server/render/internal/svc" "github.com/zeromicro/go-zero/core/conf" "github.com/zeromicro/go-zero/rest" ) var configFile = flag.String("f", "etc/render.yaml", "the config file") func main() { flag.Parse() var c config.Config conf.MustLoad(*configFile, &c) c.Timeout = int64(time.Second * 15) server := rest.MustNewServer(c.RestConf, rest.WithCustomCors(auth.FsCors, func(w http.ResponseWriter) { })) defer server.Stop() ctx := svc.NewServiceContext(c) handler.RegisterHandlers(server, ctx) //消费渲染前组装数据队列 ctx1 := context.Background() ctx2, cancel := context.WithCancel(ctx1) defer cancel() go ctx.RabbitMq.Consume(ctx2, constants.RABBIT_MQ_ASSEMBLE_RENDER_DATA, &consumer.MqConsumerRenderAssemble{}) fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port) server.Start() }