environment limit and relax actions (#276)

controller/limits/environmentRelaxAction.go

@@ -4,6 +4,8 @@ import (
 	"github.com/jmoiron/sqlx"
 	"github.com/openziti/edge/rest_management_api_client"
 	"github.com/openziti/zrok/controller/store"
+	"github.com/openziti/zrok/controller/zrokEdgeSdk"
+	"github.com/pkg/errors"
 	"github.com/sirupsen/logrus"
 )
 
@@ -16,7 +18,35 @@ func newEnvironmentRelaxAction(str *store.Store, edge *rest_management_api_clien
 	return &environmentRelaxAction{str, edge}
 }
 
-func (a *environmentRelaxAction) HandleEnvironment(e *store.Environment, rxBytes, txBytes int64, limit *BandwidthPerPeriod, trx *sqlx.Tx) error {
-	logrus.Infof("relaxing '%v'", e.ZId)
+func (a *environmentRelaxAction) HandleEnvironment(env *store.Environment, rxBytes, txBytes int64, limit *BandwidthPerPeriod, trx *sqlx.Tx) error {
+	logrus.Infof("relaxing '%v'", env.ZId)
+
+	shrs, err := a.str.FindSharesForEnvironment(env.Id, trx)
+	if err != nil {
+		return errors.Wrapf(err, "error finding shares for environment '%v'", env.ZId)
+	}
+
+	for _, shr := range shrs {
+		if shr.ShareMode == "public" {
+			env, err := a.str.GetEnvironment(shr.EnvironmentId, trx)
+			if err != nil {
+				return errors.Wrap(err, "error finding environment")
+			}
+
+			fe, err := a.str.FindFrontendPubliclyNamed(*shr.FrontendSelection, trx)
+			if err != nil {
+				return errors.Wrapf(err, "error finding frontend name '%v' for '%v'", *shr.FrontendSelection, shr.Token)
+			}
+
+			if err := zrokEdgeSdk.CreateServicePolicyDial(env.ZId+"-"+shr.ZId+"-dial", shr.ZId, []string{fe.ZId}, zrokEdgeSdk.ZrokShareTags(shr.Token).SubTags, a.edge); err != nil {
+				return errors.Wrapf(err, "error creating dial service policy for '%v'", shr.Token)
+			}
+			logrus.Infof("added dial service policy for '%v'", shr.Token)
+
+		} else if shr.ShareMode == "private" {
+			return errors.New("share relax for private share not implemented")
+		}
+	}
+
 	return nil
 }