api_w_page_view.go 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  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. pv_continue := tool.Get_setting(db, "not_use_view_count", "")
  16. if len(pv_continue) == 0 || pv_continue[0][0] == "" {
  17. stmt, err := db.Prepare(tool.DB_change("select set_data from data_set where doc_name = ? and set_name = 'view_count'"))
  18. if err != nil {
  19. log.Fatal(err)
  20. }
  21. defer stmt.Close()
  22. var view_count string
  23. err = stmt.QueryRow(other_set["doc_name"]).Scan(&view_count)
  24. if err != nil {
  25. if err == sql.ErrNoRows {
  26. view_count = "0"
  27. } else {
  28. log.Fatal(err)
  29. }
  30. }
  31. if view_count == "0" {
  32. stmt, err := db.Prepare(tool.DB_change("insert into data_set (doc_name, doc_rev, set_name, set_data) values (?, '', 'view_count', '1')"))
  33. if err != nil {
  34. log.Fatal(err)
  35. }
  36. defer stmt.Close()
  37. _, err = stmt.Exec(other_set["doc_name"])
  38. if err != nil {
  39. log.Fatal(err)
  40. }
  41. } else {
  42. view_count_int, _ := strconv.Atoi(view_count)
  43. stmt, err := db.Prepare(tool.DB_change("update data_set set set_data = ? where doc_name = ? and set_name = 'view_count'"))
  44. if err != nil {
  45. log.Fatal(err)
  46. }
  47. defer stmt.Close()
  48. _, err = stmt.Exec(view_count_int+1, other_set["doc_name"])
  49. if err != nil {
  50. log.Fatal(err)
  51. }
  52. }
  53. }
  54. return_data := make(map[string]interface{})
  55. return_data["response"] = "ok"
  56. json_data, _ := json.Marshal(return_data)
  57. return string(json_data)
  58. }