Mercurial > gemma
view pkg/models/intservices.go @ 904:e4b72a199258
New default bottleneck colors
Mainly to make the stroke color one actually selectable in the ui.
In addition the pink does better match the collors used on the ECDIS layer.
author | Sascha Wilde <wilde@intevation.de> |
---|---|
date | Tue, 02 Oct 2018 13:34:59 +0200 |
parents | 0c79068c46b7 |
children | f0b6852c14d1 |
line wrap: on
line source
package models import ( "context" "database/sql" "log" "sync" "gemma.intevation.de/gemma/pkg/auth" "gemma.intevation.de/gemma/pkg/config" ) type IntEntry struct { Name string `json:"name"` Style sql.NullString `json:"-"` // This should be done separately. WMS bool `json:"wms"` WFS bool `json:"wfs"` } type IntServices struct { entries []IntEntry mu sync.Mutex } const selectPublishedServices = `SELECT relname, style, as_wms, as_wfs FROM sys_admin.published_services JOIN pg_class ON name = oid ORDER by relname` var InternalServices = &IntServices{} func (ps *IntServices) Find(name string) (string, bool) { ps.mu.Lock() defer ps.mu.Unlock() if ps.entries == nil { if err := ps.load(); err != nil { log.Printf("error: %v\n", err) return "", false } } if ps.has(name) { return config.GeoServerURL() + "/" + name, true } return "", false } func (ps *IntServices) has(service string) bool { var check func(*IntEntry) bool switch service { case "wms": check = func(e *IntEntry) bool { return e.WMS } case "wfs": check = func(e *IntEntry) bool { return e.WFS } default: return false } for i := range ps.entries { if check(&ps.entries[i]) { return true } } return false } func (ps *IntServices) load() error { // make empty slice to prevent retry if slice is empty. ps.entries = []IntEntry{} return auth.RunAs("sys_admin", context.Background(), func(conn *sql.Conn) error { rows, err := conn.QueryContext( context.Background(), selectPublishedServices) if err != nil { return err } defer rows.Close() for rows.Next() { var entry IntEntry if err := rows.Scan( &entry.Name, &entry.Style, &entry.WMS, &entry.WFS, ); err != nil { return err } ps.entries = append(ps.entries, entry) } return rows.Err() }) } func (ps *IntServices) Invalidate() { ps.mu.Lock() ps.entries = nil ps.mu.Unlock() } func InternalAll(IntEntry) bool { return true } func IntWMS(entry IntEntry) bool { return entry.WMS } func IntWFS(entry IntEntry) bool { return entry.WFS } func (ps *IntServices) Filter(accept func(IntEntry) bool) []IntEntry { ps.mu.Lock() defer ps.mu.Unlock() if ps.entries == nil { if err := ps.load(); err != nil { log.Printf("error: %v\n", err) return nil } } pe := make([]IntEntry, 0, len(ps.entries)) for _, e := range ps.entries { if accept(e) { pe = append(pe, e) } } return pe }