Mercurial > gemma
view pkg/models/extservices.go @ 1755:d3fe20a13339
Import: Filter fairway availabilities by period and resolve insert confilcts.
author | Raimund Renkert <raimund.renkert@intevation.de> |
---|---|
date | Thu, 10 Jan 2019 16:43:42 +0100 |
parents | cabf4789e02b |
children | 5f47eeea988d |
line wrap: on
line source
// This is Free Software under GNU Affero General Public License v >= 3.0 // without warranty, see README.md and license for details. // // SPDX-License-Identifier: AGPL-3.0-or-later // License-Filename: LICENSES/AGPL-3.0.txt // // Copyright (C) 2018 by via donau // – Österreichische Wasserstraßen-Gesellschaft mbH // Software engineering by Intevation GmbH // // Author(s): // * Sascha L. Teichmann <sascha.teichmann@intevation.de> package models import ( "context" "database/sql" "log" "sort" "sync" "gemma.intevation.de/gemma/pkg/auth" ) type ExtEntry struct { Name string `json:"name"` URL string `json:"url"` WFS bool `json:"wfs"` } type ExtServices struct { mu sync.Mutex entries []ExtEntry } var ExternalServices = &ExtServices{} const selectExternalServices = `SELECT local_name, remote_url, is_wfs FROM sys_admin.external_services ORDER BY local_name` func (es *ExtServices) Find(name string) (string, bool) { es.mu.Lock() defer es.mu.Unlock() if es.entries == nil { if err := es.load(); err != nil { log.Printf("error: %v\n", err) return "", false } } n := sort.Search(len(es.entries), func(i int) bool { return es.entries[i].Name >= name }) if n == len(es.entries) || es.entries[n].Name != name { return "", false } return es.entries[n].URL, true } func (es *ExtServices) load() error { // make empty slice to prevent retry if slice is empty. es.entries = []ExtEntry{} return auth.RunAs(context.Background(), "sys_admin", func(conn *sql.Conn) error { rows, err := conn.QueryContext( context.Background(), selectExternalServices) if err != nil { return err } defer rows.Close() for rows.Next() { var entry ExtEntry if err := rows.Scan( &entry.Name, &entry.URL, &entry.WFS, ); err != nil { return err } es.entries = append(es.entries, entry) } return rows.Err() }) } func (es *ExtServices) Invalidate() { es.mu.Lock() es.entries = nil es.mu.Unlock() } func ExternalAll(ExtEntry) bool { return true } func ExternalWMS(entry ExtEntry) bool { return !entry.WFS } func ExternalWFS(entry ExtEntry) bool { return entry.WFS } func (es *ExtServices) Filter(accept func(ExtEntry) bool) []ExtEntry { es.mu.Lock() defer es.mu.Unlock() if es.entries == nil { if err := es.load(); err != nil { log.Printf("error: %v\n", err) return nil } } ee := make([]ExtEntry, 0, len(es.entries)) for _, e := range es.entries { if accept(e) { ee = append(ee, e) } } return ee }