api_w_page_view.go 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. package route
  2. import (
  3. "database/sql"
  4. "log"
  5. "opennamu/route/tool"
  6. "strconv"
  7. jsoniter "github.com/json-iterator/go"
  8. )
  9. func Api_w_page_view(call_arg []string) string {
  10. var json = jsoniter.ConfigCompatibleWithStandardLibrary
  11. other_set := map[string]string{}
  12. json.Unmarshal([]byte(call_arg[0]), &other_set)
  13. db := tool.DB_connect()
  14. defer db.Close()
  15. stmt, err := db.Prepare(tool.DB_change("select set_data from data_set where doc_name = ? and set_name = 'view_count'"))
  16. if err != nil {
  17. log.Fatal(err)
  18. }
  19. defer stmt.Close()
  20. var view_count string
  21. err = stmt.QueryRow(other_set["doc_name"]).Scan(&view_count)
  22. if err != nil {
  23. if err == sql.ErrNoRows {
  24. view_count = "0"
  25. } else {
  26. log.Fatal(err)
  27. }
  28. }
  29. if view_count == "0" {
  30. stmt, err := db.Prepare(tool.DB_change("insert into data_set (doc_name, doc_rev, set_name, set_data) values (?, '', 'view_count', '1')"))
  31. if err != nil {
  32. log.Fatal(err)
  33. }
  34. defer stmt.Close()
  35. _, err = stmt.Exec(other_set["doc_name"])
  36. if err != nil {
  37. log.Fatal(err)
  38. }
  39. } else {
  40. view_count_int, _ := strconv.Atoi(view_count)
  41. stmt, err := db.Prepare(tool.DB_change("update data_set set set_data = ? where doc_name = ? and set_name = 'view_count'"))
  42. if err != nil {
  43. log.Fatal(err)
  44. }
  45. defer stmt.Close()
  46. _, err = stmt.Exec(view_count_int+1, other_set["doc_name"])
  47. if err != nil {
  48. log.Fatal(err)
  49. }
  50. }
  51. return_data := make(map[string]interface{})
  52. return_data["response"] = "ok"
  53. json_data, _ := json.Marshal(return_data)
  54. return string(json_data)
  55. }