gemma: pkg/geoserver/boot.go comparison

comparison pkg/geoserver/boot.go @ 4936:21a48e2d2260 fairway-marks-import

Create group layers in GeoServer via REST calls.

author	Sascha L. Teichmann <sascha.teichmann@intevation.de>
date	Mon, 17 Feb 2020 00:45:21 +0100
parents	c64dba002726
children	40da1b8aba01

comparison

equal deleted inserted replaced

-:c64dba002726
+:21a48e2d2260
 	return err
 }
 func ensureFeatures() error {
+	tables := models.InternalServices.Filter(models.IntWFS)
+	if len(tables) == 0 {
+		log.Println("info: no tables to publish")
+		return nil
+	}
+	log.Printf("info: number of tables to publish %d\n", len(tables))
 	var (
 		geoURL   = config.GeoServerURL()
 		user     = config.GeoServerUser()
 		password = config.GeoServerPassword()
 		auth     = basicAuth(user, password)
 	)
 	datastoreURL := geoURL + "/rest/workspaces/" + workspaceName +
 		"/datastores/" + datastoreName
-	tables := models.InternalServices.Filter(models.IntWFS)
-	if len(tables) == 0 {
-		log.Println("info: no tables to publish")
-		return nil
-	}
-	log.Printf("info: number of tables to publish %d\n", len(tables))
 	var features struct {
 		FeatureTypes struct {
 			FeatureType []struct {
 				Name string `json:"name"`
 		log.Println("info: no groups layers to publish")
 		return nil
 	}
 	log.Printf("info: number of layer groups to publish %d\n", len(groups))
-	// TODO: Implement me!
+	var (
-	/*
+		geoURL   = config.GeoServerURL()
-		for i := range groups {
+		user     = config.GeoServerUser()
-			log.Printf("info: layer group #%d: %s\n", i+1, groups[i].Name)
+		password = config.GeoServerPassword()
-			for _, layer := range groups[i].Layers {
+		auth     = basicAuth(user, password)
-				log.Printf("info: layer: %s\n", layer)
+	)
+	type layerGroups struct {
+		LayerGroups struct {
+			LayerGroup []struct {
+				Name string `json:"name"`
+			} `json:"layerGroup"`
+		} `json:"layerGroups"`
+	}
+	var lg layerGroups
+	hasLayerGroup := func(name string) bool {
+		for i := range lg.LayerGroups.LayerGroup {
+			if lg.LayerGroups.LayerGroup[i].Name == name {
+				return true
 			}
 		}
-	*/
+		return false
+	}
+	layerGroupsURL := geoURL + "/rest/workspaces/" + workspaceName +
+		"/layergroups"
+	// Fetch all layer groups.
+	req, err := http.NewRequest(
+		http.MethodGet,
+		layerGroupsURL+".json",
+		nil)
+	if err != nil {
+		return err
+	}
+	auth(req)
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return err
+	}
+	err = json.NewDecoder(resp.Body).Decode(&lg)
+	resp.Body.Close()
+	if err != nil {
+		// XXX: ignore this error.
+	}
+	var already []string
+	defer func() {
+		if len(already) > 0 {
+			log.Printf("info: already having layer groups: %s\n",
+				strings.Join(already, ", "))
+		}
+	}()
+	for i := range groups {
+		if hasLayerGroup(groups[i].Name) {
+			already = append(already, groups[i].Name)
+			continue
+		}
+		type (
+			Layers struct {
+				Layer []string `xml:"layer"`
+			}
+			Styles struct {
+				Style []string `xml:"style"`
+			}
+			LayerGroup struct {
+				XMLName xml.Name `xml:"layerGroup"`
+				Name    string   `xml:"name"`
+				Title   string   `xml:"title"`
+				Layers  Layers   `xml:"layers"`
+				Styles  Styles   `xml:"styles"`
+			}
+		)
+		lgr := LayerGroup{
+			Name:  groups[i].Name,
+			Title: groups[i].Name,
+			Layers: Layers{
+				Layer: groups[i].Layers,
+			},
+			Styles: Styles{
+				Style: groups[i].Layers,
+			},
+		}
+		req, err = http.NewRequest(
+			http.MethodPost,
+			layerGroupsURL+".xml",
+			toXMLStream(&lgr))
+		if err != nil {
+			return err
+		}
+		asContentType(req, "text/xml")
+		auth(req)
+		resp, err := http.DefaultClient.Do(req)
+		if err != nil {
+			return err
+		}
+		if resp.StatusCode != http.StatusCreated {
+			return fmt.Errorf("status code '%s' (%d)",
+				http.StatusText(resp.StatusCode),
+				resp.StatusCode)
+		}
+	}
 	return nil
 }
 func deleteWorkspace() error {

Mercurial > gemma

comparison pkg/geoserver/boot.go @ 4936:21a48e2d2260 fairway-marks-import