Skip to content

A
ai-api-mgr
Commit 2b13f8d6 authored by Your Name's avatar Your Name
Browse files
Options

add with bill db param

parent 076a0de5
Hide whitespace changes
Inline Side-by-side
Showing with 19 additions and 6 deletions
main.go
View file @ 2b13f8d6
......@@ -425,6 +425,9 @@ func checkErr(err error) {
}
}
var taskJsonStr = `{"id":10,"desc":"测试新增","price":1000,"complexity":1,"hardware_require":"{ disk_size: 100, memory_size: 100 }","image_id":"13221312","image_url":"dasdasdasd","cmd":"{\"image_name\":\"demianhjw/aigic:0129\",\"docker_cmd\":{\"container_port\":\"5001\"},\"api_url\":\"http://127.0.0.1:%d/aigic\"}","workload":100,"api_path":"/demianhjw/aigic/0129","image_name":"demianhjw/aigic:0129","sign_url":"dsdsdsds","username":"sdsds","password":"sdsdsd","created_time":"2024-02-02T03:13:33+08:00","updated_time":"2024-02-02T03:13:33+08:00","deleted":0}`
var withBillDb bool
func main() {
slog.SetDefault(slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{
......@@ -438,6 +441,7 @@ func main() {
var mysqlPort int
var mysqlDbName, mysqlUser, mysqlPassWd string
//var redisPassWd string
//flag.StringVar(&questAddr, "questAddr", "192.168.1.10:9009", "questDbAddr")
......@@ -457,17 +461,23 @@ func main() {
flag.StringVar(&aigcProduceTopic, "aigcProduceTopic", "pbaigc", "produce topic, default value is: pbaigc")
flag.StringVar(&aigcConsumerTopic, "aigcConsumerTopic", "taskreceipt", "consumer topic, default value is: taskreceipt")
flag.BoolVar(&withBillDb, "withbilldb", true, "enable with bill db, or simulate")
flag.Parse()
slog.Warn("start param", "quest", slog.GroupValue(slog.String("Addr", questAddr)))
slog.Warn("start param", "kafka", slog.GroupValue(slog.String("Addr", kafkaBroker)), slog.Group("topic", "aigcProduceTopic", aigcProduceTopic, "aigcConsumerTopic", aigcConsumerTopic))
slog.Info("start param", "callback", slog.GroupValue(slog.String("Addr", callbackAddrP)))
slog.Warn("start param", "callback", slog.GroupValue(slog.String("Addr", callbackAddrP)))
slog.Warn("start param", "listenIpPort", slog.GroupValue(slog.String("Addr", listenIpPort)))
slog.Warn("start param", "redis", slog.GroupValue(slog.String("Addr", redisAddr), slog.String("PassWd", redisPassWd)))
slog.Warn("start param", "mysql", slog.GroupValue(slog.String("Addr", mysqlAddr),
slog.Int("port", mysqlPort), slog.String("DbName", mysqlDbName), slog.String("User", mysqlUser), slog.String("PassWd", mysqlPassWd))) //, slog.GroupValue(slog.Int("port",mysqlPort),slog.String("mysqlDbName",mysqlDbName),slog.GroupValue(slog.String("mysqlUser",mysqlUser)))))
slog.Warn("start param", "!!!!!!!!!!!!!!!!!!!!**********With Bill DB**********!!!!!!!!!!!!!!!!!!!!", withBillDb)
slog.Warn("start param", "!!!!!!!!!!!!!!!!!!!!**********With Bill DB**********!!!!!!!!!!!!!!!!!!!!", withBillDb)
slog.Warn("start param", "!!!!!!!!!!!!!!!!!!!!**********With Bill DB**********!!!!!!!!!!!!!!!!!!!!", withBillDb)
kafkaBrokers = []string{kafkaBroker}
callbackAddr = callbackAddrP
......@@ -548,8 +558,6 @@ func main() {
return c.SendStatus(200)
})
var taskJsonStr = `{"id":10,"desc":"测试新增","price":1000,"complexity":1,"hardware_require":"{ disk_size: 100, memory_size: 100 }","image_id":"13221312","image_url":"dasdasdasd","cmd":"{\"image_name\":\"demianhjw/aigic:0129\",\"docker_cmd\":{\"container_port\":\"5001\"},\"api_url\":\"http://127.0.0.1:%d/aigic\"}","workload":100,"api_path":"/demianhjw/aigic/0129","image_name":"demianhjw/aigic:0129","sign_url":"dsdsdsds","username":"sdsds","password":"sdsdsd","created_time":"2024-02-02T03:13:33+08:00","updated_time":"2024-02-02T03:13:33+08:00","deleted":0}`
task := model.TaskType{}
if err := json.Unmarshal([]byte(taskJsonStr), &task); err != nil {
......@@ -592,15 +600,16 @@ func ApiAndJWT(c *fiber.Ctx) error {
var task *model.TaskType
if true {
task = replanceQueryTask
} else {
if withBillDb {
var err error
task, err = cache.Query(pathInDB, int64(uidAsInt))
if err != nil {
return c.SendString(fmt.Sprintf("cache.Query %v", err.Error()))
}
} else {
task = replanceQueryTask
}
//var cmd string
......@@ -645,6 +654,10 @@ func ApiAndJWT(c *fiber.Ctx) error {
asyncMode := false
// for k, v := range reqHeaders {
// fmt.Println("k-------", k, "v", v)
// }
if headerSync, ok := reqHeaders["Async"]; ok {
//fmt.Println("sync-----------------sync", headerSync)
for _, syncAsString := range headerSync {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment