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) { ...@@ -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() { func main() {
slog.SetDefault(slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{ slog.SetDefault(slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{
...@@ -438,6 +441,7 @@ func main() { ...@@ -438,6 +441,7 @@ func main() {
var mysqlPort int var mysqlPort int
var mysqlDbName, mysqlUser, mysqlPassWd string var mysqlDbName, mysqlUser, mysqlPassWd string
//var redisPassWd string //var redisPassWd string
//flag.StringVar(&questAddr, "questAddr", "192.168.1.10:9009", "questDbAddr") //flag.StringVar(&questAddr, "questAddr", "192.168.1.10:9009", "questDbAddr")
...@@ -457,17 +461,23 @@ func main() { ...@@ -457,17 +461,23 @@ func main() {
flag.StringVar(&aigcProduceTopic, "aigcProduceTopic", "pbaigc", "produce topic, default value is: pbaigc") flag.StringVar(&aigcProduceTopic, "aigcProduceTopic", "pbaigc", "produce topic, default value is: pbaigc")
flag.StringVar(&aigcConsumerTopic, "aigcConsumerTopic", "taskreceipt", "consumer topic, default value is: taskreceipt") flag.StringVar(&aigcConsumerTopic, "aigcConsumerTopic", "taskreceipt", "consumer topic, default value is: taskreceipt")
flag.BoolVar(&withBillDb, "withbilldb", true, "enable with bill db, or simulate")
flag.Parse() flag.Parse()
slog.Warn("start param", "quest", slog.GroupValue(slog.String("Addr", questAddr))) 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.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", "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", "redis", slog.GroupValue(slog.String("Addr", redisAddr), slog.String("PassWd", redisPassWd)))
slog.Warn("start param", "mysql", slog.GroupValue(slog.String("Addr", mysqlAddr), 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.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} kafkaBrokers = []string{kafkaBroker}
callbackAddr = callbackAddrP callbackAddr = callbackAddrP
...@@ -548,8 +558,6 @@ func main() { ...@@ -548,8 +558,6 @@ func main() {
return c.SendStatus(200) 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{} task := model.TaskType{}
if err := json.Unmarshal([]byte(taskJsonStr), &task); err != nil { if err := json.Unmarshal([]byte(taskJsonStr), &task); err != nil {
...@@ -592,15 +600,16 @@ func ApiAndJWT(c *fiber.Ctx) error { ...@@ -592,15 +600,16 @@ func ApiAndJWT(c *fiber.Ctx) error {
var task *model.TaskType var task *model.TaskType
if true { if withBillDb {
task = replanceQueryTask
} else {
var err error var err error
task, err = cache.Query(pathInDB, int64(uidAsInt)) task, err = cache.Query(pathInDB, int64(uidAsInt))
if err != nil { if err != nil {
return c.SendString(fmt.Sprintf("cache.Query %v", err.Error())) return c.SendString(fmt.Sprintf("cache.Query %v", err.Error()))
} }
} else {
task = replanceQueryTask
} }
//var cmd string //var cmd string
...@@ -645,6 +654,10 @@ func ApiAndJWT(c *fiber.Ctx) error { ...@@ -645,6 +654,10 @@ func ApiAndJWT(c *fiber.Ctx) error {
asyncMode := false asyncMode := false
// for k, v := range reqHeaders {
// fmt.Println("k-------", k, "v", v)
// }
if headerSync, ok := reqHeaders["Async"]; ok { if headerSync, ok := reqHeaders["Async"]; ok {
//fmt.Println("sync-----------------sync", headerSync) //fmt.Println("sync-----------------sync", headerSync)
for _, syncAsString := range 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