💾 Archived View for source.community › Fuwn › space › blob › main › pkg › database › database.go captured on 2024-02-05 at 10:01:57. Gemini links have been rewritten to link to archived content

-=-=-=-=-=-=-

                                                         .
,-. ,-. . . ,-. ,-. ,-.    ,-. ,-. ,-,-. ,-,-. . . ,-. . |- . .
`-. | | | | |   |   |-'    |   | | | | | | | | | | | | | |  | |
`-' `-' `-^ '   `-' `-' :: `-' `-' ' ' ' ' ' ' `-^ ' ' ' `' `-|
                                                             /|
                                                            `-'

Fuwn / space

git clone https://source.community/Fuwn/space.git

Branches

Log

Tree

/pkg/database/database.go (main)

↑ /pkg/database

blob

View raw contents of /pkg/database/database.go (main)

────╮
   1│ // Copyright (C) 2021-2021 Fuwn
   2│ // SPDX-License-Identifier: GPL-3.0-only
   3│ 
   4│ package database
   5│ 
   6│ import (
   7│ 	"github.com/sonyarouje/simdb"
   8│ )
   9│ 
  10│ var driver *simdb.Driver
  11│ 
  12│ func init() {
  13│ 	var err error
  14│ 	driver, err = simdb.New(".space/database")
  15│ 	if err != nil {
  16│ 		panic(err)
  17│ 	}
  18│ }
  19│ 
  20│ type Hit struct {
  21│ 	Path  string `json:"path"`
  22│ 	Count int    `json:"count"`
  23│ }
  24│ 
  25│ func (c Hit) ID() (jsonField string, value interface{}) {
  26│ 	value = c.Path
  27│ 	jsonField = "path"
  28│ 	return
  29│ }
  30│ 
  31│ func Get(path string) int {
  32│ 	var hit Hit
  33│ 
  34│ 	err := driver.Open(Hit{}).Where("path", "=", path).First().AsEntity(&hit)
  35│ 	if err != nil {
  36│ 		return 0
  37│ 	}
  38│ 
  39│ 	return hit.Count + 1
  40│ }
  41│ 
  42│ func Create(path string) {
  43│ 	driver.Insert(Hit{
  44│ 		Path:  path,
  45│ 		Count: 0,
  46│ 	})
  47│ }
  48│ 
  49│ func Increment(path string) {
  50│ 	var hit Hit
  51│ 
  52│ 	err := driver.Open(Hit{}).Where("path", "=", path).First().AsEntity(&hit)
  53│ 	if err != nil {
  54│ 		return
  55│ 	}
  56│ 
  57│ 	hit.Count = hit.Count + 1
  58│ 	driver.Update(hit)
  59│ }
────╯

· · ·

info@source.community